class ZippedExportWriterTests(SimpleTestCase):
def setUp(self):
self.zip_file_patch = patch('zipfile.ZipFile')
self.MockZipFile = self.zip_file_patch.start()
self.path_mock = Mock()
self.path_mock.get_path.return_value = 'tmp'
self.writer = ZippedExportWriter()
self.writer.archive_basepath = '✓path'
self.writer.tables = [self.path_mock]
self.writer.file = Mock()
def tearDown(self):
self.zip_file_patch.stop()
del self.writer
def test_zipped_export_writer_unicode(self):
mock_zip_file = self.MockZipFile.return_value
self.writer.table_names = {0: 'ひらがな'}
self.writer._write_final_result()
filename = os.path.join(self.writer.archive_basepath, 'ひらがな.csv')
mock_zip_file.write.assert_called_with('tmp', filename)
def test_zipped_export_writer_utf8(self):
mock_zip_file = self.MockZipFile.return_value
self.writer.table_names = {0: b'\xe3\x81\xb2\xe3\x82\x89\xe3\x81\x8c\xe3\x81\xaa'}
self.writer._write_final_result()
filename = os.path.join(self.writer.archive_basepath, 'ひらがな.csv')
mock_zip_file.write.assert_called_with('tmp', filename)
class ZippedExportWriterTests(SimpleTestCase):
def setUp(self):
self.zip_file_patch = patch('zipfile.ZipFile')
self.MockZipFile = self.zip_file_patch.start()
self.path_mock = Mock()
self.path_mock.get_path.return_value = 'tmp'
self.writer = ZippedExportWriter()
self.writer.tables = [self.path_mock]
self.writer.file = Mock()
def tearDown(self):
self.zip_file_patch.stop()
del self.writer
def test_zipped_export_writer_unicode(self):
mock_zip_file = self.MockZipFile.return_value
self.writer.table_names = {0: u'ひらがな'}
self.writer._write_final_result()
mock_zip_file.write.assert_called_with('tmp', 'ひらがな.csv')
def test_zipped_export_writer_utf8(self):
mock_zip_file = self.MockZipFile.return_value
self.writer.table_names = {0: '\xe3\x81\xb2\xe3\x82\x89\xe3\x81\x8c\xe3\x81\xaa'}
self.writer._write_final_result()
mock_zip_file.write.assert_called_with('tmp', 'ひらがな.csv')
def setUp(self):
self.zip_file_patch = patch('zipfile.ZipFile')
self.MockZipFile = self.zip_file_patch.start()
self.path_mock = Mock()
self.path_mock.get_path.return_value = 'tmp'
self.writer = ZippedExportWriter()
self.writer.tables = [self.path_mock]
self.writer.file = Mock()
def setUp(self):
self.zip_file_patch = patch('zipfile.ZipFile')
self.MockZipFile = self.zip_file_patch.start()
self.path_mock = Mock()
self.path_mock.get_path.return_value = 'tmp'
self.writer = ZippedExportWriter()
self.writer.tables = [self.path_mock]
self.writer.file = Mock()
