From 73489e3b4b3fea7501d3734c6f0c9343cf758cae Mon Sep 17 00:00:00 2001 From: ulif <uli@gnufix.de> Date: Fri, 27 Mar 2015 11:43:06 +0100 Subject: [PATCH] Make sure we get version infos from main. --- tests/test_diceware.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/test_diceware.py b/tests/test_diceware.py index e63264f..7217591 100644 --- a/tests/test_diceware.py +++ b/tests/test_diceware.py @@ -6,7 +6,7 @@ from io import StringIO from diceware import ( SRC_DIR, RE_LANG_CODE, SPECIAL_CHARS, get_wordlist, get_wordlist_path, insert_special_char, get_passphrase, - handle_options, main, + handle_options, main, __version__, ) @@ -195,6 +195,15 @@ class TestDicewareModule(object): expected_output = fd.read() assert out == expected_output + def test_main_version(self, argv_handler, capsys): + # we can get version infos. + sys.argv = ['diceware', '--version'] + with pytest.raises(SystemExit) as exc_info: + main() + assert exc_info.value.code == 0 + out, err = capsys.readouterr() + assert __version__ in out + def test_main_argv(self, argv_handler): # main() handles sys.argv if nothing is provided sys.argv = ['diceware', '--help'] -- GitLab