def test_custom_writer_with_json_file_formatter(self): # given: options = { 'name': 'exporters.export_formatter.json_export_formatter.JSONExportFormatter', 'options': { 'jsonlines': False } } formatter = JsonExportFormatter(options, meta()) writer = FakeWriter({}, {}, export_formatter=formatter) # when: try: writer.write_batch(self.batch) writer.flush() finally: writer.close() # then: output = writer.custom_output[()] out = json.loads(output) self.assertEquals(self.batch, out) self.assertEquals('json', writer.write_buffer.items_group_files.file_extension)
def setUp(self): item_writer = GroupingBufferFilesTracker( JsonExportFormatter({}, meta()), 'gz') self.write_buffer = WriteBuffer({}, meta(), items_per_buffer_write=1000, size_per_buffer_write=1000, items_group_files_handler=item_writer)
class JsonFormatterTest(unittest.TestCase): def setUp(self): self.options = { } self.export_formatter = JsonExportFormatter(self.options, meta()) def test_format(self): item = BaseRecord() item['key'] = 0 item['value'] = random.randint(0, 10000) item = self.export_formatter.format(item) self.assertIsInstance(json.loads(item), dict)
def setUp(self): self.options = { } self.export_formatter = JsonExportFormatter(self.options, meta())