Skip to content

Commit 2716796

Browse files
committed
Fix test_run_cmd_requires_args
1 parent 1725ac6 commit 2716796

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

tests/test_cli.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,15 @@
1313
from covimerage.cli import get_version_message
1414

1515

16+
def format_click_arg(name):
17+
if click.__version__ < '7.0':
18+
return '"{}"'.format(name)
19+
elif click.__version__ < '7.1':
20+
return '"{}..."'.format(name.upper())
21+
else:
22+
return "'{}...'".format(name.upper())
23+
24+
1625
def test_dunder_main_run(capfd):
1726
assert call([sys.executable, '-m', 'covimerage']) == 0
1827
out, err = capfd.readouterr()
@@ -795,9 +804,10 @@ def test_run_forwards_sighup(devnull):
795804

796805
def test_run_cmd_requires_args(runner):
797806
result = runner.invoke(cli.run, [])
798-
assert 'Error: Missing argument "%s".' % (
799-
'args' if click.__version__ < '7.0' else 'ARGS...',
800-
) in result.output.splitlines()
807+
assert (
808+
'Error: Missing argument {}.'.format(format_click_arg('args'))
809+
in result.output.splitlines()
810+
)
801811
assert result.exit_code == 2
802812

803813

0 commit comments

Comments
 (0)