Merge pull request #6 from lfyzjck/master

update arg_parser.py to fix `--port` argument type
ClearcodeHQ · Apr 23, 2015 · ca1f27d · ca1f27d
2 parents 0f3d1e1 + d7abdf0
commit ca1f27d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/ianitor/args_parser.py b/src/ianitor/args_parser.py
@@ -143,6 +143,7 @@ def get_parser():
 
     parser.add_argument(
         "--port",
+        type=int,
         help="set service port",
     )
 

diff --git a/tests/test_args_parser.py b/tests/test_args_parser.py
@@ -45,6 +45,12 @@ def test_parse_args():
     args, invocation = args_parser.parse_args()
     assert invocation == ['tail', '-f', 'something']
 
+
+@patch('sys.argv', ["ianitor", "tailf", '--port', '1234', '--', 'tail', '-f', 'something'])  # noqa
+def test_parse_port():
+    args, invocation = args_parser.parse_args()
+    assert args.port == 1234
+
 TEST_TTL = 100