class TestItemWriter(unittest.TestCase):
def setUp(self):
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.item_writer = ItemWriter(mock_object_writer)
self.item_writer._ItemWriter__no_items_written_to_file = True
def test_write_headers(self):
mock_item = mock.Mock()
self.item_writer.write_headers(mock_item)
self.item_writer.object_writer.write_headers.assert_called_once_with(
mock_item)
@mock.patch.object(ItemWriter, 'write_headers')
def test_write_item_first_item(self, mock_write_headers):
self.item_writer._ItemWriter__no_items_written_to_file = True
mock_item = mock.Mock()
self.item_writer.write_item(mock_item)
self.assertFalse(
self.item_writer._ItemWriter__no_items_written_to_file)
mock_write_headers.assert_called_once_with(mock_item)
self.item_writer.object_writer.write_object.assert_called_once_with(
mock_item)
def test_write_item_non_first_item(self):
self.item_writer._ItemWriter__no_items_written_to_file = False
mock_item = mock.Mock()
self.item_writer.write_item(mock_item)
self.item_writer.object_writer.write_object.assert_called_once_with(
mock_item)
def test_open_output_file(self):
mock_file_name = "mock_output.csv"
mock_file = mock.Mock()
self.item_writer.object_writer.open_file.return_value = mock_file
self.item_writer.open_output_file(mock_file_name)
self.assertEquals(self.item_writer.output_file, mock_file)
def test_close_output_file(self):
self.item_writer.output_file = mock.Mock()
self.item_writer.close_output_file()
self.item_writer.output_file.close.assert_called_once_with()
class TestItemWriter(unittest.TestCase):
def setUp(self):
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.item_writer = ItemWriter(mock_object_writer)
self.item_writer._ItemWriter__no_items_written_to_file = True
def test_write_headers(self):
mock_item = mock.Mock()
self.item_writer.write_headers(mock_item)
self.item_writer.object_writer.write_headers.assert_called_once_with(mock_item)
@mock.patch.object(ItemWriter, 'write_headers')
def test_write_item_first_item(self, mock_write_headers):
self.item_writer._ItemWriter__no_items_written_to_file = True
mock_item = mock.Mock()
self.item_writer.write_item(mock_item)
self.assertFalse(self.item_writer._ItemWriter__no_items_written_to_file)
mock_write_headers.assert_called_once_with(mock_item)
self.item_writer.object_writer.write_object.assert_called_once_with(mock_item)
def test_write_item_non_first_item(self):
self.item_writer._ItemWriter__no_items_written_to_file = False
mock_item = mock.Mock()
self.item_writer.write_item(mock_item)
self.item_writer.object_writer.write_object.assert_called_once_with(mock_item)
def test_open_output_file(self):
mock_file_name = "mock_output.csv"
mock_file = mock.Mock()
self.item_writer.object_writer.open_file.return_value = mock_file
self.item_writer.open_output_file(mock_file_name)
self.assertEquals(self.item_writer.output_file, mock_file)
def test_close_output_file(self):
self.item_writer.output_file = mock.Mock()
self.item_writer.close_output_file()
self.item_writer.output_file.close.assert_called_once_with()
def setUp(self):
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.item_writer = ItemWriter(mock_object_writer)
self.item_writer._ItemWriter__no_items_written_to_file = True
def setUp(self):
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.item_writer = ItemWriter(mock_object_writer)
self.item_writer._ItemWriter__no_items_written_to_file = True
