From 32b5035c36e67926b2eb790b6df4d337baa8dc17 Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Sat, 24 Dec 2016 22:53:16 +0000 Subject: Add function `get_args` * gd_diff.py (get_args): New function. * tests/tests_gd_diff.py (test_get_args_gd_diff_version) (test_get_args): Tests for function `get_args`. --- gd_diff.py | 10 ++++++++++ tests/test_gd_diff.py | 19 +++++++++++++++++++ 2 files changed, 29 insertions(+) diff --git a/gd_diff.py b/gd_diff.py index c6b3b36..c89729b 100644 --- a/gd_diff.py +++ b/gd_diff.py @@ -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)): -- cgit v1.2.3