def test_get_args_pass():
tests = [
[['updown_service', 'myservice.name', 'up'], 'up', None, 300, False,
None],
[['updown_service', 'myservice.name', 'down'], 'down', None, 300,
False, None],
[['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42,
42, False, None],
[['updown_service', 'myservice.name', 'down', '-x'], 'down', None, 300,
True, None],
[['updown_service', 'myservice.name:1234', 'down', '-x'], 'down', None,
300, True, 1234],
]
for test in tests:
(argv, expected_state, expected_args_timeout, expected_timeout,
expected_wait_only, expected_port) = test
with mock.patch('sys.argv', argv):
args = updown_service.get_args()
timeout = updown_service._get_timeout_s(args.service, args.timeout)
assert args.service == 'myservice.name'
assert args.state == expected_state
assert args.timeout == expected_args_timeout
assert timeout == expected_timeout
assert args.wait_only == expected_wait_only
assert args.port == expected_port
def test_timeout_s():
arg_timeout_s = 30
new_timeout_s = 50
mconfig_path = 'nerve_tools.updown_service.load_service_namespace_config'
mconfig = mock.Mock(return_value={})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main', arg_timeout_s) == arg_timeout_s
mconfig = mock.Mock(return_value={})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main', None) == updown_service.DEFAULT_TIMEOUT_S
mconfig = mock.Mock(return_value={'updown_timeout_s': new_timeout_s})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main', arg_timeout_s) == arg_timeout_s
mconfig = mock.Mock(return_value={'updown_timeout_s': new_timeout_s})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main', None) == new_timeout_s
def test_timeout_s():
arg_timeout_s = 30
new_timeout_s = 50
mconfig_path = 'nerve_tools.updown_service.load_service_namespace_config'
mconfig = mock.Mock(return_value={})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main',
arg_timeout_s) == arg_timeout_s
mconfig = mock.Mock(return_value={})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s(
'test.main', None) == updown_service.DEFAULT_TIMEOUT_S
mconfig = mock.Mock(return_value={'updown_timeout_s': new_timeout_s})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main',
arg_timeout_s) == arg_timeout_s
mconfig = mock.Mock(return_value={'updown_timeout_s': new_timeout_s})
with mock.patch(mconfig_path, new=mconfig):
assert updown_service._get_timeout_s('test.main',
None) == new_timeout_s
def test_get_args_pass():
tests = [
[['updown_service', 'myservice.name', 'up'], 'up', None, 300],
[['updown_service', 'myservice.name', 'down'], 'down', None, 300],
[['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42, 42],
]
for test in tests:
argv, expected_state, expected_args_timeout, expected_timeout = test
with mock.patch('sys.argv', argv):
args = updown_service.get_args()
timeout = updown_service._get_timeout_s(args.service, args.timeout)
assert args.service == 'myservice.name'
assert args.state == expected_state
assert args.timeout == expected_args_timeout
assert timeout == expected_timeout
def test_get_args_pass():
tests = [
[['updown_service', 'myservice.name', 'up'], 'up', None, 300],
[['updown_service', 'myservice.name', 'down'], 'down', None, 300],
[['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42,
42],
]
for test in tests:
argv, expected_state, expected_args_timeout, expected_timeout = test
with mock.patch('sys.argv', argv):
args = updown_service.get_args()
timeout = updown_service._get_timeout_s(args.service, args.timeout)
assert args.service == 'myservice.name'
assert args.state == expected_state
assert args.timeout == expected_args_timeout
assert timeout == expected_timeout