def test_to_image_sets_image_path(self):
# Arrange
cm = StoreWriter(self.directory, self.file_name)
# Act
self.assertIsNone((cm.get_img_path()))
cm.to_image(MagicMock(), MagicMock(), 'name')
# Assert
self.assertIsNotNone((cm.get_img_path()))
def test_get_image_returns_image_path(self):
cm = StoreWriter(self.directory, self.file_name)
cm.to_image(MagicMock(), 'test_image')
path = cm.get_img_path()
self.assertIsNotNone(path)
self.assertIn('test_image', path)
self.assertIn("img_dir", path)
self.assertIn(self.directory, path)
def test_get_img_path_returns_none_if_to_image_not_called(self):
# Arrange
cm = StoreWriter(self.directory, self.file_name)
cm.to_image = MagicMock()
# Act
cm.to_image.assert_not_called()
# Assert
self.assertIsNone(cm.get_img_path())
