def test_remove_image_file_attempts_to_remove_file_when_it_exists(self):
# Arrange
cm = StoreWriter(self.directory, self.file_name)
cm._file_manager = MagicMock()
cm._file_manager.is_file.return_value = True
# Act
cm.remove_img_file(MagicMock())
# Assert
cm._file_manager.remove.assert_called_once()
def test_make_image_dir_attempts_to_create_img_dir(self):
# Arrange
cm = StoreWriter(self.directory, self.file_name)
cm._file_manager = MagicMock()
img_dir = os.path.join(self.directory, "img_dir")
# Act
cm._make_img_dir()
# Assert
self.assertEqual(cm._file_manager.make_dir_when_no_dir.call_count, 2)
cm._file_manager.make_dir_when_no_dir.assert_called_with(img_dir)
def test_to_file_csv_store_in_csv_file(self):
# Arrange
cm = StoreWriter(self.directory, self.file_name)
cm._file_manager = MagicMock()
file_name = os.path.join(self.directory, self.file_name + ".csv")
# Act
cm.to_csv_file(self.records)
# Assert
((filename_used, record_lines_used),
kwargs) = cm._file_manager.write_lines.call_args_list[0]
self.assertEqual(file_name, filename_used)
for r, l in zip(self.records, record_lines_used):
self.assertIn(r.to_csv_string(), l)
