Python RemoveImage.run示例

示例#1

    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"

示例#2

文件： test_images.py 项目： zorrock/prefect

    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"

示例#3

 def test_invalid_image_raises_error(self):
     task = RemoveImage()
     with pytest.raises(ValueError):
         task.run(image=None)

示例#4

 def test_empty_image_raises_error(self):
     task = RemoveImage()
     with pytest.raises(ValueError):
         task.run()