X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/af968d49039f5cd8f0ef5839558a57867d7ef111..3f73a363f42cb3cb6627ddcea371e3a22810daa6:/test/python/dummy_tokenizer.py?ds=sidebyside diff --git a/test/python/dummy_tokenizer.py b/test/python/dummy_tokenizer.py index 47cc580c..0a86ba8d 100644 --- a/test/python/dummy_tokenizer.py +++ b/test/python/dummy_tokenizer.py @@ -13,9 +13,10 @@ class DummyTokenizer: self.dsn = dsn self.data_dir = data_dir self.init_state = None + self.analyser_cache = {} - def init_new_db(self, config): + def init_new_db(self, *args, **kwargs): assert self.init_state == None self.init_state = "new" @@ -23,3 +24,44 @@ class DummyTokenizer: def init_from_project(self): assert self.init_state == None self.init_state = "loaded" + + + def finalize_import(self, _): + pass + + + def name_analyzer(self): + return DummyNameAnalyzer(self.analyser_cache) + + +class DummyNameAnalyzer: + + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_value, traceback): + self.close() + + + def __init__(self, cache): + self.analyser_cache = cache + cache['countries'] = [] + + + def close(self): + pass + + def normalize_postcode(self, postcode): + return postcode + + def update_postcodes_from_db(self): + pass + + def update_special_phrases(self, phrases): + self.analyser_cache['special_phrases'] = phrases + + def add_country_names(self, code, names): + self.analyser_cache['countries'].append((code, names)) + + def process_place(self, place): + return {}