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`.

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)):