diff options
Diffstat (limited to 'tests/test_gd_diff.py')
-rw-r--r-- | tests/test_gd_diff.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tests/test_gd_diff.py b/tests/test_gd_diff.py index 9f29ef2..b93536c 100644 --- a/tests/test_gd_diff.py +++ b/tests/test_gd_diff.py @@ -22,6 +22,7 @@ from unittest import mock from nose.tools import * from gd_diff import * +from gns_deb_diff._version import __version__ class TestGdDiff(object): @@ -461,6 +462,24 @@ class TestGdDiff(object): assert wiki_page == gns_wiki_header() + '\n' + wiki_table + def test_get_args_gd_diff_version(self): + mock_sys_argv = ['gd-diff', '--version'] + with mock.patch('sys.stdout', new=StringIO()) as output, \ + mock.patch('sys.argv', new=mock_sys_argv): + try: + args = get_args() + except SystemExit as se: + assert_equal(__version__, output.getvalue().rstrip('\n')) + + + def test_get_args(self): + mock_sys_argv = ['gd-diff', 'parkes', '3'] + with mock.patch('sys.argv', new=mock_sys_argv): + args = get_args() + assert args.release == 'parkes' + assert args.version == 3 + + def teardown(self): """Teardown method for this class.""" if(path.exists(self.gns_pkgs_dir)): |