def test_execute_in_container_reuses_exec():
fake_container_id = 'fake_container_id'
fake_execid = 'fake_execid'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {
'ExecIDs': [fake_execid]
}
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', fake_command],
}
}
assert execute_in_container(mock_docker_client, fake_container_id,
fake_command,
1) == (fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid,
stream=False)
def test_execute_in_container_reuses_only_valid_exec():
fake_container_id = 'fake_container_id'
fake_execid = 'fake_execid'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
bad_exec = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', 'some_other_command'],
}
}
good_exec = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', fake_command],
}
}
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {'ExecIDs': ['fake_other_exec', fake_execid, 'fake_other_exec']}
mock_docker_client.exec_start.return_value = fake_output
# the last side effect is used to check the exit code of the command
mock_docker_client.exec_inspect.side_effect = [bad_exec, good_exec, bad_exec, good_exec]
assert execute_in_container(mock_docker_client, fake_container_id, fake_command, 1) == (
fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid, stream=False)
def test_execute_in_container_reuses_exec():
fake_container_id = "fake_container_id"
fake_execid = "fake_execid"
fake_return_code = 0
fake_output = "fake_output"
fake_command = "fake_cmd"
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {
"ExecIDs": [fake_execid]
}
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {
"ExitCode": fake_return_code,
"ProcessConfig": {
"entrypoint": "/bin/sh",
"arguments": ["-c", fake_command]
},
}
assert execute_in_container(mock_docker_client, fake_container_id,
fake_command,
1) == (fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid,
stream=False)
def test_execute_in_container_reuses_only_valid_exec():
fake_container_id = 'fake_container_id'
fake_execid = 'fake_execid'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
bad_exec = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', 'some_other_command'],
}
}
good_exec = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', fake_command],
}
}
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {'ExecIDs': ['fake_other_exec', fake_execid, 'fake_other_exec']}
mock_docker_client.exec_start.return_value = fake_output
# the last side effect is used to check the exit code of the command
mock_docker_client.exec_inspect.side_effect = [bad_exec, good_exec, bad_exec, good_exec]
assert execute_in_container(mock_docker_client, fake_container_id, fake_command, 1) == (
fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid, stream=False)
def test_execute_in_container():
fake_container_id = 'fake_container_id'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {'ExitCode': fake_return_code}
assert execute_in_container(mock_docker_client, fake_container_id, fake_command, 1) == (
fake_output, fake_return_code)
expected_cmd = ['/bin/sh', '-c', fake_command]
mock_docker_client.exec_create.assert_called_once_with(fake_container_id, expected_cmd)
def test_execute_in_container():
fake_container_id = 'fake_container_id'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {'ExitCode': fake_return_code}
assert execute_in_container(mock_docker_client, fake_container_id, fake_command, 1) == (
fake_output, fake_return_code)
expected_cmd = ['/bin/sh', '-c', fake_command]
mock_docker_client.exec_create.assert_called_once_with(fake_container_id, expected_cmd)
def perform_cmd_healthcheck(docker_client, container_id, command, timeout):
"""Returns true if return code of command is 0 when executed inside container, false otherwise
:param docker_client: Docker client object
:param container_id: Docker container id
:param command: command to execute
:param timeout: timeout in seconds
:returns: True if command exits with return code 0, false otherwise
"""
(output, return_code) = execute_in_container(docker_client, container_id, command, timeout)
if return_code == 0:
return (True, output)
else:
return (False, output)
def perform_cmd_healthcheck(docker_client, container_id, command, timeout):
"""Returns true if return code of command is 0 when executed inside container, false otherwise
:param docker_client: Docker client object
:param container_id: Docker container id
:param command: command to execute
:param timeout: timeout in seconds
:returns: True if command exits with return code 0, false otherwise
"""
(output, return_code) = execute_in_container(docker_client, container_id, command, timeout)
if return_code == 0:
return (True, output)
else:
return (False, output)
def test_execute_in_container():
fake_container_id = "fake_container_id"
fake_return_code = 0
fake_output = "fake_output"
fake_command = "fake_cmd"
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {
"ExitCode": fake_return_code
}
assert execute_in_container(mock_docker_client, fake_container_id,
fake_command,
1) == (fake_output, fake_return_code)
expected_cmd = ["/bin/sh", "-c", fake_command]
mock_docker_client.exec_create.assert_called_once_with(
fake_container_id, expected_cmd)
def test_execute_in_container_reuses_exec():
fake_container_id = 'fake_container_id'
fake_execid = 'fake_execid'
fake_return_code = 0
fake_output = 'fake_output'
fake_command = 'fake_cmd'
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {'ExecIDs': [fake_execid]}
mock_docker_client.exec_start.return_value = fake_output
mock_docker_client.exec_inspect.return_value = {
'ExitCode': fake_return_code,
'ProcessConfig': {
'entrypoint': '/bin/sh',
'arguments': ['-c', fake_command],
}
}
assert execute_in_container(mock_docker_client, fake_container_id, fake_command, 1) == (
fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid, stream=False)
def test_execute_in_container_reuses_only_valid_exec():
fake_container_id = "fake_container_id"
fake_execid = "fake_execid"
fake_return_code = 0
fake_output = "fake_output"
fake_command = "fake_cmd"
bad_exec = {
"ExitCode": fake_return_code,
"ProcessConfig": {
"entrypoint": "/bin/sh",
"arguments": ["-c", "some_other_command"],
},
}
good_exec = {
"ExitCode": fake_return_code,
"ProcessConfig": {
"entrypoint": "/bin/sh",
"arguments": ["-c", fake_command]
},
}
mock_docker_client = mock.MagicMock(spec_set=docker.Client)
mock_docker_client.inspect_container.return_value = {
"ExecIDs": ["fake_other_exec", fake_execid, "fake_other_exec"]
}
mock_docker_client.exec_start.return_value = fake_output
# the last side effect is used to check the exit code of the command
mock_docker_client.exec_inspect.side_effect = [
bad_exec,
good_exec,
bad_exec,
good_exec,
]
assert execute_in_container(mock_docker_client, fake_container_id,
fake_command,
1) == (fake_output, fake_return_code)
assert mock_docker_client.exec_create.call_count == 0
mock_docker_client.exec_start.assert_called_once_with(fake_execid,
stream=False)