Useful helpers for writing tests for your Python CLI program.
Project description
cli-test-helpers
Useful helpers for writing tests for your Python CLI program.
Writing tests for a command line interface (CLI) tool may not seem strictly straight-forward when you think in terms of unit tests. Especially, when you use the argparse module or the click package, control of the application entry point is a bit taken away from you.
But it’s not all that bad. This package is here to help. The examples give you some guidance on how to get started, and the helpers allow you to deal with common cases, such as mocking CLI arguments and environment variable values.
Installation
pip install cli-test-helpers
Preferrably, though, you add cli-test-helpers as a dependency to your Tox environment (see example).
Usage
Let’s assume you use pytest for running your tests, which is certainly a good idea. Your CLI program is called foobar. You have prepared a setup.py with a CLI entrypoint. For the tests you have prepared a tests/ folder (outside of foobar/, because you don’t want your tests to be packaged up with your application code). Then your directory layout looks somewhat like our example.
Functional tests
Start with a simple set of functional tests:
Is the entrypoint script installed? (tests the configuration in your setup.py)
Can this package be run as a Python module? (i.e. without having to be installed)
Is command XYZ available? etc. Cover your entire CLI usage here!
This is almost a stupid exercise: Run the command as a shell command and inspect the status code of the exiting process (see example). The trick is that you run a non-destructive command, e.g. by using the usual --help option of every command. This should cover your entire CLI user interface definition.
Unit tests
Then you’re ready to take advantage of our helpers.
ArgvContext allows you to mimic the use of specific CLI arguments:
def test_cli_command(mock_command):
"""Is the correct code called when invoked via the CLI?"""
with ArgvContext('foobar', 'baz'):
foobar.command.baz()
assert mock_command.call_count == 1
EnvironContext allows you to mimic the presence of environment values:
def test_fail_without_secret():
"""Must fail without a ``SECRET`` environment variable specified"""
message = "Environment value SECRET not set."
with EnvironContext(SECRET=None):
with pytest.raises(SystemExit, match=message):
foobar.command.baz()
pytest.fail("CLI doesn't abort with missing SECRET")
See example.
TDD
Remember to stick to the test-driven mantra:
Write one line of test code. Make the test fail.
Write one line of application code. Make the test pass.
Goto 1.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for cli_test_helpers-1.0.0-py2-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ed1bbae7031f66339510076450ad4ccff918ffadf565f7dfa884f148cb674d00 |
|
MD5 | f5858821f887c8f31db82e3476b6a43a |
|
BLAKE2b-256 | 19692df351e8e099b562ff17bda3a8e9d9ba09d283165119d0a826335670d604 |