def import_naptan(previous_result=None, url=None, force_update=False):
if previous_result in (None, True):
try:
app = create_app()
with app.blueprint_context(BLUEPRINT_NAME):
url = url or app.config['NAPTAN_IMPORT_URL']
naptan = get_resource(url, force_update)
if naptan:
archive = zipfile.ZipFile(open(naptan))
f = archive.open('NaPTAN.xml')
naptan = NaPTANImporter(search_importer, 10, f, ['340'], 'identifiers')
naptan.run()
return True
else:
logger.info("Naptan hasn't been imported - resource not loaded")
except:
logger.error("Error running NaPTAN importer", exc_info=True)
return False
def test_search_called_each_result(self):
naptan_importer = NaPTANImporter(self.mock_solr, 10, file(self.naptan_file), ['639'], 'identifiers')
naptan_importer.run()
self.assertEqual(self.mock_solr.search_for_ids.call_count,
len(naptan_importer.handler.stop_points)+len(naptan_importer.handler.stop_areas))
