def test_create_images(self):
self.fake_client.images.create_image.return_value = \
self.fake_object['body']
self.useFixture(
mockpatch.PatchObject(javelin,
"client_for_user",
return_value=self.fake_client))
self.useFixture(
mockpatch.PatchObject(javelin,
"_get_image_by_name",
return_value=[]))
self.useFixture(
mockpatch.PatchObject(javelin,
"_resolve_image",
return_value=(None, None)))
with mock.patch('six.moves.builtins.open',
mock.mock_open(),
create=True) as open_mock:
javelin.create_images([self.fake_object])
mocked_function = self.fake_client.images.create_image
mocked_function.assert_called_once_with(self.fake_object['name'],
self.fake_object['format'],
self.fake_object['format'])
mocked_function = self.fake_client.images.store_image_file
fake_image_id = self.fake_object['body'].get('id')
mocked_function.assert_called_once_with(fake_image_id, open_mock())
def test_create_images(self):
self.fake_client.images.create_image.return_value = self.fake_object["body"]
self.useFixture(mockpatch.PatchObject(javelin, "client_for_user", return_value=self.fake_client))
self.useFixture(mockpatch.PatchObject(javelin, "_get_image_by_name", return_value=[]))
self.useFixture(mockpatch.PatchObject(javelin, "_resolve_image", return_value=(None, None)))
with mock.patch("six.moves.builtins.open", mock.mock_open(), create=True) as open_mock:
javelin.create_images([self.fake_object])
mocked_function = self.fake_client.images.create_image
mocked_function.assert_called_once_with(
self.fake_object["name"], self.fake_object["format"], self.fake_object["format"]
)
mocked_function = self.fake_client.images.store_image_file
fake_image_id = self.fake_object["body"].get("id")
mocked_function.assert_called_once_with(fake_image_id, open_mock())
