Show only version on http-prompt --version

httpie · May 30, 2016 · ac92e27 · ac92e27
1 parent accf14e
commit ac92e27
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/http_prompt/cli.py b/http_prompt/cli.py
@@ -30,7 +30,7 @@ def fix_incomplete_url(url):
 ))
 @click.argument('url', default='http://localhost')
 @click.argument('http_options', nargs=-1, type=click.UNPROCESSED)
-@click.version_option()
+@click.version_option(message='%(version)s')
 def cli(url, http_options):
     click.echo('Version: %s' % __version__)
 

diff --git a/tests/test_installation.py b/tests/test_installation.py
@@ -26,7 +26,7 @@ def test_help():
 @pytest.mark.slow
 def test_version():
     out, err = run_http_prompt(['--version'])
-    expected = 'http-prompt, version {0}\n'.format(__version__)
-    if hasattr(expected, 'encode'):
-        expected = expected.encode('ascii')
-    assert out == expected
+    version = __version__
+    if hasattr(version, 'encode'):
+        version = version.encode('ascii')
+    assert out.rstrip() == version