def test_container_id_is_replaced(self, monkeypatch):
task = GetContainerLogs(container_id="original")
api = MagicMock()
monkeypatch.setattr("docker.APIClient", api)
task.run(container_id="test")
assert api.return_value.logs.call_args[1]["container"] == "test"
def test_container_id_run_value_is_used(self, monkeypatch):
task = GetContainerLogs()
api = MagicMock()
monkeypatch.setattr("prefect.tasks.docker.containers.docker.APIClient",
api)
task.run(container_id="test")
assert api.return_value.logs.call_args[1]["container"] == "test"
def test_invalid_container_id_raises_error(self):
task = GetContainerLogs()
with pytest.raises(ValueError):
task.run(container_id=None)
def test_empty_container_id_raises_error(self):
task = GetContainerLogs()
with pytest.raises(ValueError):
task.run()