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()