def test_image_is_replaced(self, monkeypatch): task = RemoveImage(image="original") api = MagicMock() monkeypatch.setattr("docker.APIClient", api) task.run(image="test") assert api.return_value.remove_image.call_args[1]["image"] == "test"
def test_image_run_value_is_used(self, monkeypatch): task = RemoveImage() api = MagicMock() monkeypatch.setattr("prefect.tasks.docker.containers.docker.APIClient", api) task.run(image="test") assert api.return_value.remove_image.call_args[1]["image"] == "test"
def test_invalid_image_raises_error(self): task = RemoveImage() with pytest.raises(ValueError): task.run(image=None)
def test_empty_image_raises_error(self): task = RemoveImage() with pytest.raises(ValueError): task.run()