- def test_import_missing_file(self):
- assert self.call_nominatim('import', '--osm-file', 'sfsafegwedgw.reh.erh') == 1
-
-
- def test_import_bad_file(self):
- assert self.call_nominatim('import', '--osm-file', '.') == 1
-
-
- def test_import_full(self, mock_func_factory):
- mocks = [
- mock_func_factory(nominatim.tools.database_import, 'setup_database_skeleton'),
- mock_func_factory(nominatim.tools.country_info, 'setup_country_tables'),
- mock_func_factory(nominatim.tools.database_import, 'import_osm_data'),
- mock_func_factory(nominatim.tools.refresh, 'import_wikipedia_articles'),
- mock_func_factory(nominatim.tools.database_import, 'truncate_data_tables'),
- mock_func_factory(nominatim.tools.database_import, 'load_data'),
- mock_func_factory(nominatim.tools.database_import, 'create_tables'),
- mock_func_factory(nominatim.tools.database_import, 'create_table_triggers'),
- mock_func_factory(nominatim.tools.database_import, 'create_partition_tables'),
- mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
- mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
- mock_func_factory(nominatim.tools.refresh, 'load_address_levels_from_config'),
- mock_func_factory(nominatim.tools.postcodes, 'update_postcodes'),
- mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
- mock_func_factory(nominatim.tools.refresh, 'setup_website'),
- mock_func_factory(nominatim.db.properties, 'set_property')
- ]
-
- cf_mock = mock_func_factory(nominatim.tools.refresh, 'create_functions')
-
- assert self.call_nominatim('import', '--osm-file', __file__) == 0
- assert self.tokenizer_mock.finalize_import_called
-
- assert cf_mock.called > 1
-
- for mock in mocks:
- assert mock.called == 1, "Mock '{}' not called".format(mock.func_name)
-
-
- def test_import_continue_load_data(self, mock_func_factory):
- mocks = [
- mock_func_factory(nominatim.tools.database_import, 'truncate_data_tables'),
- mock_func_factory(nominatim.tools.database_import, 'load_data'),
- mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
- mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
- mock_func_factory(nominatim.tools.postcodes, 'update_postcodes'),
- mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
- mock_func_factory(nominatim.tools.refresh, 'setup_website'),
- mock_func_factory(nominatim.db.properties, 'set_property')
- ]
-
- assert self.call_nominatim('import', '--continue', 'load-data') == 0
- assert self.tokenizer_mock.finalize_import_called
-
- for mock in mocks:
- assert mock.called == 1, "Mock '{}' not called".format(mock.func_name)
-
-
- def test_import_continue_indexing(self, mock_func_factory, placex_table,
- temp_db_conn):
- mocks = [
- mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
- mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
- mock_func_factory(nominatim.indexer.indexer.Indexer, 'index_full'),
- mock_func_factory(nominatim.tools.refresh, 'setup_website'),
- mock_func_factory(nominatim.db.properties, 'set_property')
- ]
-
- assert self.call_nominatim('import', '--continue', 'indexing') == 0
-
- for mock in mocks:
- assert mock.called == 1, "Mock '{}' not called".format(mock.func_name)
-
- assert temp_db_conn.index_exists('idx_placex_pendingsector')
-
- # Calling it again still works for the index
- assert self.call_nominatim('import', '--continue', 'indexing') == 0
- assert temp_db_conn.index_exists('idx_placex_pendingsector')
-
-
- def test_import_continue_postprocess(self, mock_func_factory):
- mocks = [
- mock_func_factory(nominatim.tools.database_import, 'create_search_indices'),
- mock_func_factory(nominatim.tools.country_info, 'create_country_names'),
- mock_func_factory(nominatim.tools.refresh, 'setup_website'),
- mock_func_factory(nominatim.db.properties, 'set_property')
- ]
-
- assert self.call_nominatim('import', '--continue', 'db-postprocess') == 0
-
- assert self.tokenizer_mock.finalize_import_called
-
- for mock in mocks:
- assert mock.called == 1, "Mock '{}' not called".format(mock.func_name)