def setUp(self):
        super(TestImageSet, self).setUp()

        # This is the return value for utils.find_resource()
        self.images_mock.get.return_value = self._image
        self.images_mock.update.return_value = self._image

        # Get the command object to test
        self.cmd = image.SetImage(self.app, None)
    def setUp(self):
        super(TestImageSet, self).setUp()

        # This is the return value for utils.find_resource()
        self.client.find_image = mock.Mock(return_value=self._image)
        self.client.update_image = mock.Mock(return_value=self._image)

        # Get the command object to test
        self.cmd = image.SetImage(self.app, None)
    def setUp(self):
        super(TestImageSet, self).setUp()

        # This is the return value for utils.find_resource()
        self.images_mock.get.return_value = fakes.FakeResource(
            None,
            copy.deepcopy(image_fakes.IMAGE),
            loaded=True,
        )
        self.images_mock.update.return_value = fakes.FakeResource(
            None,
            copy.deepcopy(image_fakes.IMAGE),
            loaded=True,
        )

        # Get the command object to test
        self.cmd = image.SetImage(self.app, None)