diff options
-rw-r--r-- | gd_diff.py | 10 | ||||
-rw-r--r-- | tests/test_gd_diff.py | 19 |
2 files changed, 29 insertions, 0 deletions
@@ -395,3 +395,13 @@ def write_wiki_page(release, content): wd_release = wiki_page_dir(release) wp_file = os.path.join(wd_release, 'last.rev') write_file(wp_file, content) + + +def get_args(): + parser = argparse.ArgumentParser() + parser.add_argument('--version', action='version', version=__version__) + parser.add_argument('release', help='gNewSense release name') + parser.add_argument('version', help='gNewSense version number', + type=int) + + return parser.parse_args() 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)): |