def test_image_name_run_value_is_used(self, monkeypatch):
task = CreateContainer()
api = MagicMock()
monkeypatch.setattr("docker.APIClient", api)
task.run(image_name="test")
assert api.return_value.create_container.call_args[1]["image"] == "test"
def test_extra_kwargs(self, monkeypatch, caplog):
task = CreateContainer(extra_docker_kwargs={"network": "test-network"})
api = MagicMock()
monkeypatch.setattr("docker.APIClient", api)
task.run(image_name="test")
assert (api.return_value.create_container.call_args[1]["network"] ==
"test-network")
def test_image_name_is_replaced(self, monkeypatch):
task = CreateContainer(image_name="original")
api = MagicMock()
monkeypatch.setattr("prefect.tasks.docker.containers.docker.APIClient",
api)
task.run(image_name="test")
assert api.return_value.create_container.call_args[1][
"image"] == "test"
def test_invalid_image_name_raises_error(self):
task = CreateContainer()
with pytest.raises(ValueError):
task.run(image_name=None)
def test_empty_image_name_raises_error(self):
task = CreateContainer()
with pytest.raises(ValueError):
task.run()