Testing Click Applications¶
Click provides the click.testing module to help you invoke command line applications and check their behavior.
These tools should only be used for testing since they change the entire interpreter state for simplicity. They are not thread-safe!
The examples use pytest style tests.
Basic Example¶
The key pieces are:
CliRunner- used to invoke commands as command line scripts.Result- returned fromCliRunner.invoke(). Captures output data, exit code, optional exception, and captures the output as bytes and binary data.
import click
@click.command()
@click.argument('name')
def hello(name):
click.echo(f'Hello {name}!')
from click.testing import CliRunner
from hello import hello
def test_hello_world():
runner = CliRunner()
result = runner.invoke(hello, ['Peter'])
assert result.exit_code == 0
assert result.output == 'Hello Peter!\n'
Subcommands¶
A subcommand name must be specified in the args parameter
CliRunner.invoke():
import click
@click.group()
@click.option('--debug/--no-debug', default=False)
def cli(debug):
click.echo(f"Debug mode is {'on' if debug else 'off'}")
@cli.command()
def sync():
click.echo('Syncing')
from click.testing import CliRunner
from sync import cli
def test_sync():
runner = CliRunner()
result = runner.invoke(cli, ['--debug', 'sync'])
assert result.exit_code == 0
assert 'Debug mode is on' in result.output
assert 'Syncing' in result.output
Context Settings¶
Additional keyword arguments passed to CliRunner.invoke() will be used to
construct the initial Context object.
For example, setting a fixed terminal width equal to 60:
import click
@click.group()
def cli():
pass
@cli.command()
def sync():
click.echo('Syncing')
from click.testing import CliRunner
from sync import cli
def test_sync():
runner = CliRunner()
result = runner.invoke(cli, ['sync'], terminal_width=60)
assert result.exit_code == 0
assert 'Debug mode is on' in result.output
assert 'Syncing' in result.output
File System Isolation¶
The CliRunner.isolated_filesystem() context manager sets the current
working directory to a new, empty folder.
import click
@click.command()
@click.argument('f', type=click.File())
def cat(f):
click.echo(f.read())
from click.testing import CliRunner
from cat import cat
def test_cat():
runner = CliRunner()
with runner.isolated_filesystem():
with open('hello.txt', 'w') as f:
f.write('Hello World!')
result = runner.invoke(cat, ['hello.txt'])
assert result.exit_code == 0
assert result.output == 'Hello World!\n'
Pass in a path to control where the temporary directory is created. In this case, the directory will not be removed by Click. Its useful to integrate with a framework like Pytest that manages temporary files.
from click.testing import CliRunner
from cat import cat
def test_cat_with_path_specified():
runner = CliRunner()
with runner.isolated_filesystem('~/test_folder'):
with open('hello.txt', 'w') as f:
f.write('Hello World!')
result = runner.invoke(cat, ['hello.txt'])
assert result.exit_code == 0
assert result.output == 'Hello World!\n'
Input Streams¶
The test wrapper can provide input data for the input stream (stdin). This is very useful for testing prompts.
import click
@click.command()
@click.option('--foo', prompt=True)
def prompt(foo):
click.echo(f"foo={foo}")
from click.testing import CliRunner
from prompt import prompt
def test_prompts():
runner = CliRunner()
result = runner.invoke(prompt, input='wau wau\n')
assert not result.exception
assert result.output == 'Foo: wau wau\nfoo=wau wau\n'
Prompts will be emulated so they write the input data to the output stream as well. If hidden input is expected then this does not happen.
Capture modes¶
CliRunner captures output by replacing sys.stdout and sys.stderr
with in-memory wrappers. The capture parameter controls which strategy is
used.
capture="sys" (default)¶
Captures Python-level writes (print(), click.echo(), sys.stdout.write()).
It is fast and sufficient for most Click applications.
Code that holds a reference to the original sys.stdout (like a library that
does from sys import stdout at import time) bypasses the capture and its
output is lost.
In this mode sys.stdout.fileno() and sys.stderr.fileno() raise
io.UnsupportedOperation, matching the pre-8.3.3 behavior. C-level
consumers (faulthandler, subprocess, C extensions) that expect a
real file descriptor must opt into the capture="fd" mode.
capture="fd"¶
Redirects OS file descriptors 1 and 2 to a temporary file via
os.dup2(), inspired by Pytest’s
capfd.
This catches output that bypasses sys.stdout, including:
Stale references to the original
sys.stdoutandsys.stderr.Logging frameworks that cache the original stream (like
structlogor the stdlib’sloggingmodule).C extensions and subprocesses that write directly to
fd 1orfd 2.
from click.testing import CliRunner
from myapp import cli
def test_captures_everything():
runner = CliRunner(capture="fd")
result = runner.invoke(cli)
# result.stdout contains both Python-level and fd-level output
assert "expected output" in result.stdout
In this mode sys.stdout.fileno() returns the saved (pre-redirection) fd, so
faulthandler and similar consumers keep working. Writes to fd 1 and
fd 2 land in the capture tmpfile, so os.dup2() calls inside the CLI no
longer leak into the host runner’s stdout.
Note
capture="fd" is not available on Windows.
Changed in version 8.4.0: Added the capture parameter. The default sys mode no longer exposes the
original fd through fileno(), reverting the change introduced in 8.3.3
that broke Pytest’s fd-level capture teardown. Use capture="fd" to restore
that behavior with proper isolation.