class TestObjectWriterFactory(unittest.TestCase):
def setUp(self):
compatibility_factory = mock.create_autospec(CompatibilityFactory).return_value
self.object_writer_factory = ObjectWriterFactory(compatibility_factory)
@mock.patch('xcrawler.files.writers.object_writer_factory.ObjectWriterCsv')
def test_create_object_writer_csv(self, mock_object_writer_csv_class):
mock_write_opener = mock.create_autospec(CompatibleWriteOpener).return_value
mock_object_converter = mock.create_autospec(CompatibleObjectConverter).return_value
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.object_writer_factory.compatibility_factory.create_compatible_write_opener.return_value = mock_write_opener
self.object_writer_factory.compatibility_factory.create_compatible_object_converter.return_value = mock_object_converter
mock_object_writer_csv_class.return_value = mock_object_writer
result = self.object_writer_factory.create_object_writer_csv()
mock_object_writer_csv_class.assert_called_once_with(mock_write_opener, mock_object_converter)
self.assertEquals(result, mock_object_writer)
class TestObjectWriterFactory(unittest.TestCase):
def setUp(self):
compatibility_factory = mock.create_autospec(
CompatibilityFactory).return_value
self.object_writer_factory = ObjectWriterFactory(compatibility_factory)
@mock.patch('xcrawler.files.writers.object_writer_factory.ObjectWriterCsv')
def test_create_object_writer_csv(self, mock_object_writer_csv_class):
mock_write_opener = mock.create_autospec(
CompatibleWriteOpener).return_value
mock_object_converter = mock.create_autospec(
CompatibleObjectConverter).return_value
mock_object_writer = mock.create_autospec(ObjectWriter).return_value
self.object_writer_factory.compatibility_factory.create_compatible_write_opener.return_value = mock_write_opener
self.object_writer_factory.compatibility_factory.create_compatible_object_converter.return_value = mock_object_converter
mock_object_writer_csv_class.return_value = mock_object_writer
result = self.object_writer_factory.create_object_writer_csv()
mock_object_writer_csv_class.assert_called_once_with(
mock_write_opener, mock_object_converter)
self.assertEquals(result, mock_object_writer)
def setUp(self):
compatibility_factory = mock.create_autospec(
CompatibilityFactory).return_value
self.object_writer_factory = ObjectWriterFactory(compatibility_factory)
def setUp(self):
compatibility_factory = mock.create_autospec(CompatibilityFactory).return_value
self.object_writer_factory = ObjectWriterFactory(compatibility_factory)
def __init__(self,
filepath_splitter=FilePathSplitter(),
object_writer_factory=ObjectWriterFactory()):
self.filepath_splitter = filepath_splitter
self.object_writer_factory = object_writer_factory
