def test_multiple_write_export_instance_calls(self):
"""
Confirm that calling _write_export_instance() multiple times
(as part of a bulk export) works as expected.
"""
export_instances = [
ExportInstance(
# export_format=Format.JSON,
tables=[
TableConfiguration(
label="My table",
path=[],
columns=[
ExportColumn(
label="Q3",
item=ScalarItem(
path=['form', 'q3'],
),
selected=True,
),
]
),
]
),
ExportInstance(
# export_format=Format.JSON,
tables=[
TableConfiguration(
label="My other table",
path=['form', 'q2'],
columns=[
ExportColumn(
label="Q4",
item=ScalarItem(
path=['form', 'q2', 'q4'],
),
selected=True,
),
]
)
]
)
]
writer = _Writer(get_writer(Format.JSON))
with writer.open(_get_tables(export_instances)):
_write_export_instance(writer, export_instances[0], self.docs)
_write_export_instance(writer, export_instances[1], self.docs)
with ExportFile(writer.path, writer.format) as export:
self.assertEqual(
json.loads(export),
{
u'My table': {
u'headers': [u'Q3'],
u'rows': [[u'baz'], [u'bop']],
},
u'My other table': {
u'headers': [u'Q4'],
u'rows': [[u'bar'], [u'boop']],
}
}
)
def test_multiple_write_export_instance_calls(self):
"""
Confirm that calling _write_export_instance() multiple times
(as part of a bulk export) works as expected.
"""
export_instances = [
FormExportInstance(
# export_format=Format.JSON,
tables=[
TableConfiguration(
label="My table",
selected=True,
path=[],
columns=[
ExportColumn(
label="Q3",
item=ScalarItem(
path=[PathNode(name='form'), PathNode(name='q3')],
),
selected=True,
),
]
),
]
),
FormExportInstance(
# export_format=Format.JSON,
tables=[
TableConfiguration(
label="My other table",
selected=True,
path=[PathNode(name="form", is_repeat=False), PathNode(name="q2", is_repeat=False)],
columns=[
ExportColumn(
label="Q4",
item=ScalarItem(
path=[PathNode(name='form'), PathNode(name='q2'), PathNode(name='q4')],
),
selected=True,
),
]
)
]
)
]
writer = _Writer(get_writer(Format.JSON))
with writer.open(export_instances):
_write_export_instance(writer, export_instances[0], self.docs)
_write_export_instance(writer, export_instances[1], self.docs)
with ExportFile(writer.path, writer.format) as export:
self.assertEqual(
json.loads(export.read()),
{
u'Export1-My table': {
u'headers': [u'Q3'],
u'rows': [[u'baz'], [u'bop']],
},
u'Export2-My other table': {
u'headers': [u'Q4'],
u'rows': [[u'bar'], [u'boop']],
}
}
)