def test_remove_image_all_does_remove(self):
    web = Service('web', image='example', client=self.mock_client)
    assert web.remove_image(ImageType.all)
    self.mock_client.remove_image.assert_called_once_with(web.image_name)
def test_remove_image_all_does_remove(self):
    web = Service('web', image='example', client=self.mock_client)
    assert web.remove_image(ImageType.all)
    self.mock_client.remove_image.assert_called_once_with(web.image_name)