def test_platform_missing(mock_run_cmd):
mock_run_cmd.return_value = ""
with pytest.raises(exceptions.OIIInspectorError):
container_manager.ContainerManager("test").start_container()
assert mock_run_cmd.call_count == 2
assert call("command -v podman") in mock_run_cmd.call_args_list
assert call("command -v docker") in mock_run_cmd.call_args_list
def test_get_address_before_running(mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with container_manager.ContainerManager("test") as image_manager_instance:
with pytest.raises(exceptions.OIIInspectorError):
_ = image_manager_instance.local_address_of_image
# This _image_is_running variable is set to True only if container is running
image_manager_instance._container_running = True
assert image_manager_instance.local_address_of_image == "localhost:50051"
def test_already_running_container_will_not_start(mock_run_cmd,
mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with container_manager.ContainerManager("test") as image_manager_instance:
image_manager_instance._container_running = True
image_manager_instance.start_container()
# set value back to actual state, because of close method called by context manager
image_manager_instance._container_running = False
assert mock_run_cmd.call_count == 0
def test_exception_unknown(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.side_effect = ["terminated", None]
# retry is applied
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert mock_subprocess.Popen.return_value.poll.call_count == 2
assert image_manager_instance._port == image_manager_instance._grpc_start_port
assert mock_run_cmd.call_count == 6
def test_exception_name_in_use(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.side_effect = ["terminated", None]
mock_subprocess.Popen.return_value.stderr.read.return_value = (
"the container name test_ is already in use by test_")
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert mock_subprocess.Popen.return_value.poll.call_count == 2
assert mock_run_cmd.call_count == 6
def test_retry_called_twice(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with pytest.raises(exceptions.OIIInspectorError):
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
# other ports are not used, as exception will get trough try block in _serve_index_registry
assert mock_subprocess.Popen.return_value.kill.call_count == 2
assert mock_run_cmd.call_count > 5
def test_platform_check(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with container_manager.ContainerManager("test") as image_manager_instance:
image_manager_instance.start_container()
assert call("command -v podman") in mock_run_cmd.call_args_list
assert call("command -v docker") in mock_run_cmd.call_args_list
assert image_manager_instance._container_platform == "podman"
with container_manager.ContainerManager("test") as image_manager_instance:
image_manager_instance.start_container()
assert call("command -v podman") in mock_run_cmd.call_args_list
assert call("command -v docker") in mock_run_cmd.call_args_list
assert image_manager_instance._container_platform == "docker"
mock_run_cmd.reset_mock()
with container_manager.ContainerManager("test") as image_manager_instance:
image_manager_instance._container_platform = "podman"
image_manager_instance.start_container()
assert call("command -v podman") not in mock_run_cmd.call_args_list
assert call("command -v docker") not in mock_run_cmd.call_args_list
assert call("podman pull test") in mock_run_cmd.call_args_list
def test_address_already_in_use_and_next_port_used(mock_run_cmd,
mock_subprocess):
# only two return values are needed, as the retry decorator will not be applied due to thrown exception
mock_subprocess.Popen.return_value.poll.side_effect = ["terminated", None]
mock_subprocess.Popen.return_value.stderr.read.return_value = (
"address already in use")
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert (image_manager_instance._port ==
image_manager_instance._grpc_start_port + 1)
assert mock_run_cmd.call_count == 6
def test_exception_no_free_port(mock_run_cmd, mock_subprocess, mock_time,
mock_sleep):
mock_subprocess.Popen.return_value.poll.return_value = "terminated"
mock_subprocess.Popen.return_value.stderr.read.return_value = (
"address already in use")
# retry is applied
with pytest.raises(exceptions.NoFreePortFound):
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert mock_sleep.call_count == 100
assert mock_time.call_count == 200
assert mock_run_cmd.call_count == 4
def test_time_out(mock_run_cmd, mock_subprocess, mock_time):
mock_subprocess.Popen.return_value.poll.return_value = None
mock_time.time.side_effect = [5, 11]
# suspending retry functioning, also suspends catching the OIIInspectorError exception
with mock.patch.object(retry.api, "__retry_internal",
lambda f, *args: f()):
with pytest.raises(exceptions.OIIInspectorError):
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
mock_subprocess.Popen.return_value.kill.assert_called_once()
assert mock_run_cmd.call_count == 4
def test_container_commands(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
mock_subprocess.PIPE = "test_PIPE"
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert call("podman pull test_address") in mock_run_cmd.call_args_list
mock_subprocess.Popen.assert_called_once_with(
shlex.split(
"podman run --name=OIIInspector_running_container_50051 -p=50051:50051 test_address"
),
stdout="test_PIPE",
stderr="test_PIPE",
universal_newlines=True,
)
assert (call(
f"podman stop {image_manager_instance._base_container_name}_{image_manager_instance._port}",
tolerate_err=False,
) in mock_run_cmd.call_args_list)
assert (call(
f"podman rm {image_manager_instance._base_container_name}_{image_manager_instance._port}",
tolerate_err=False,
) in mock_run_cmd.call_args_list)
def test_address_setup(mock_run_cmd):
mock_run_cmd.return_value = "/usr/bin/podman"
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
assert image_manager_instance._image_address == "test_address"
def test_close_manager_without_pull(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with container_manager.ContainerManager("test") as image_manager_instance:
image_manager_instance._container_pulled = False
assert mock_run_cmd.call_count == 0
def test_grpcurl_query_failed_recovery(mock_run_cmd, mock_subprocess):
mock_subprocess.Popen.return_value.poll.return_value = None
with container_manager.ContainerManager(
"test_address") as image_manager_instance:
image_manager_instance.start_container()
assert mock_run_cmd.call_count == 7
