diff options
Diffstat (limited to 'testing/marionette/harness/marionette_harness/tests/harness_unit/test_marionette_arguments.py')
-rw-r--r-- | testing/marionette/harness/marionette_harness/tests/harness_unit/test_marionette_arguments.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/testing/marionette/harness/marionette_harness/tests/harness_unit/test_marionette_arguments.py b/testing/marionette/harness/marionette_harness/tests/harness_unit/test_marionette_arguments.py new file mode 100644 index 000000000..1a0687028 --- /dev/null +++ b/testing/marionette/harness/marionette_harness/tests/harness_unit/test_marionette_arguments.py @@ -0,0 +1,32 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. +import pytest + +from marionette_harness.runtests import MarionetteArguments + + +@pytest.mark.parametrize("socket_timeout", ['A', '10', '1B-', '1C2', '44.35']) +def test_parse_arg_socket_timeout(socket_timeout): + argv = ['marionette', '--socket-timeout', socket_timeout] + parser = MarionetteArguments() + + def _is_float_convertible(value): + try: + float(value) + return True + except: + return False + + if not _is_float_convertible(socket_timeout): + with pytest.raises(SystemExit) as ex: + parser.parse_args(args=argv) + assert ex.value.code == 2 + else: + args = parser.parse_args(args=argv) + assert hasattr(args, 'socket_timeout') and args.socket_timeout == float(socket_timeout) + + +if __name__ == '__main__': + import sys + sys.exit(pytest.main(['--verbose', __file__])) |