From 052a788792f542e13a5710411706e631aa6a4dd3 Mon Sep 17 00:00:00 2001
From: Daniel Hahler <git@thequod.de>
Date: Sun, 16 May 2021 15:00:47 +0200
Subject: [PATCH] Fix test_run_cmd_requires_args

---
 tests/test_cli.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/tests/test_cli.py b/tests/test_cli.py
index 93de3271..aa6d967e 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -13,6 +13,15 @@
 from covimerage.cli import get_version_message
 
 
+def format_click_arg(name):
+    if click.__version__ < '7.0':
+        return '"{}"'.format(name)
+    elif click.__version__ < '7.1':
+        return '"{}..."'.format(name.upper())
+    else:
+        return "'{}...'".format(name.upper())
+
+
 def test_dunder_main_run(capfd):
     assert call([sys.executable, '-m', 'covimerage']) == 0
     out, err = capfd.readouterr()
@@ -795,9 +804,10 @@ def test_run_forwards_sighup(devnull):
 
 def test_run_cmd_requires_args(runner):
     result = runner.invoke(cli.run, [])
-    assert 'Error: Missing argument "%s".' % (
-        'args' if click.__version__ < '7.0' else 'ARGS...',
-    ) in result.output.splitlines()
+    assert (
+        'Error: Missing argument {}.'.format(format_click_arg('args'))
+        in result.output.splitlines()
+    )
     assert result.exit_code == 2