test/python/api/fake_adaptor.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2025 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Provides dummy implementations of ASGIAdaptor for testing.
   9 """
  10 from collections import namedtuple
  11
  12 import nominatim_api.v1.server_glue as glue
  13 from nominatim_api.v1.format import dispatch as formatting
  14 from nominatim_api.config import Configuration
  15
  16
  17 class FakeError(BaseException):
  18
  19     def __init__(self, msg, status):
  20         self.msg = msg
  21         self.status = status
  22
  23     def __str__(self):
  24         return f'{self.status} -- {self.msg}'
  25
  26
  27 FakeResponse = namedtuple('FakeResponse', ['status', 'output', 'content_type'])
  28
  29
  30 class FakeAdaptor(glue.ASGIAdaptor):
  31
  32     def __init__(self, params=None, headers=None, config=None):
  33         self.params = params or {}
  34         self.headers = headers or {}
  35         self._config = config or Configuration(None)
  36
  37     def get(self, name, default=None):
  38         return self.params.get(name, default)
  39
  40     def get_header(self, name, default=None):
  41         return self.headers.get(name, default)
  42
  43     def error(self, msg, status=400):
  44         return FakeError(msg, status)
  45
  46     def create_response(self, status, output, num_results):
  47         return FakeResponse(status, output, self.content_type)
  48
  49     def base_uri(self):
  50         return 'http://test'
  51
  52     def config(self):
  53         return self._config
  54
  55     def formatting(self):
  56         return formatting
  57
  58     def query_stats(self):
  59         return None