def test_write_dump_information(self): dump_id = "foobar" data_source_item_id = "Q42" identifier_property_ids = ["P42"] language = "en" source_url = "http://foo.bar" size = 42 license_item_id = "Q21" dump_information_file = StringIO() external_data_file = StringIO() result = ResultWriter(external_data_file, dump_information_file) result.write_dump_information(dump_id, data_source_item_id, identifier_property_ids, language, source_url, size, license_item_id) actual_row_fields = self.get_first_line_csv(dump_information_file) assert dump_id == actual_row_fields[0] assert data_source_item_id == actual_row_fields[1] assert identifier_property_ids == json.loads(actual_row_fields[2]) assert language == actual_row_fields[4] assert source_url == actual_row_fields[5] assert str(size) == actual_row_fields[6] assert license_item_id == actual_row_fields[7]
def test_write_external_data(self): dump_id = "foobar" external_id = "foobar" property_id = "P42" value = "foobar" dump_information_file = StringIO() external_data_file = StringIO() result = ResultWriter(external_data_file, dump_information_file) result.write_external_value(dump_id, external_id, property_id, value) actual_row_fields = self.get_first_line_csv(external_data_file) assert dump_id == actual_row_fields[0] assert external_id == actual_row_fields[1] assert property_id == actual_row_fields[2] assert value == actual_row_fields[3]
def test_write_external_data(): expected_dump_id = "foobar" expected_value_triple = ("foobar", "P42", ["foobar"]) gnd_converter = GndDumpConverter(False) xml_dump_converter_mock = XmlDumpConverter(None, None, None) def process_dump_mock(dump_file): yield expected_value_triple xml_dump_converter_mock.process_dump = process_dump_mock gnd_converter.xml_dump_converter = xml_dump_converter_mock result_mock = ResultWriter(StringIO(), StringIO()) def write_external_value_mock(dump_id, external_id, property_id, external_value): assert expected_dump_id == dump_id assert expected_value_triple[0] == external_id assert expected_value_triple[1] == property_id assert external_value in expected_value_triple[2] result_mock.write_external_value = write_external_value_mock gnd_converter.write_external_data(expected_dump_id, None, result_mock)