X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/af968d49039f5cd8f0ef5839558a57867d7ef111..893490f94e19d113e46b89b3ae62492d40db4a7a:/test/python/dummy_tokenizer.py diff --git a/test/python/dummy_tokenizer.py b/test/python/dummy_tokenizer.py index 47cc580c..d3f006de 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,37 @@ class DummyTokenizer: def init_from_project(self): assert self.init_state == None self.init_state = "loaded" + + + 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 add_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 {}