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)
def test_parse_arg_socket_timeout_with_multiple_values(sock_timeout_value):
argv = ['marionette', '--socket-timeout', sock_timeout_value]
parser = MarionetteArguments()
def _is_float_convertible(value):
try:
float(value)
return True
except:
return False
if not _is_float_convertible(sock_timeout_value):
# should raising exception, since sock_timeout must be convertible to float.
with pytest.raises(SystemExit) as ex:
parser.parse_args(args=argv)
assert ex.value.code == 2
else:
# should pass without raising exception.
args = parser.parse_args(args=argv)
assert hasattr(args, 'socket_timeout') and args.socket_timeout == float(sock_timeout_value)
def test_parse_arg_socket_timeout_with_multiple_values(sock_timeout_value):
argv = ['marionette', '--socket-timeout', sock_timeout_value]
parser = MarionetteArguments()
def _is_float_convertible(value):
try:
float(value)
return True
except:
return False
if not _is_float_convertible(sock_timeout_value):
# should raising exception, since sock_timeout must be convertible to float.
with pytest.raises(SystemExit) as ex:
parser.parse_args(args=argv)
assert ex.value.code == 2
else:
# should pass without raising exception.
args = parser.parse_args(args=argv)
assert hasattr(args, 'socket_timeout'
) and args.socket_timeout == float(sock_timeout_value)
def setup_marionette_argument_parser():
from marionette.runtests import MarionetteArguments
from mozlog.structured import commandline
parser = MarionetteArguments()
commandline.add_logging_group(parser)
return parser
