searxng/tests/__init__.py
Ivan Gabaldon ce8929cabe
[mod] limiter: trusted proxies (#4911)
Replaces `x_for` functionality with `trusted_proxies`. This allows defining
which IP / ranges to trust extracting the client IP address from X-Forwarded-For
and X-Real-IP headers.

We don't know if the proxy chain will give us the proper client
address (REMOTE_ADDR in the WSGI environment), so we rely on reading the headers
of the proxy before SearXNG (if there is one, in that case it must be added to
trusted_proxies) hoping it has done the proper checks. In case a proxy in the
chain does not check the client address correctly, integrity is compromised and
this should be fixed by whoever manages the proxy, not us.

Closes:

- https://github.com/searxng/searxng/issues/4940
- https://github.com/searxng/searxng/issues/4939
- https://github.com/searxng/searxng/issues/4907
- https://github.com/searxng/searxng/issues/3632
- https://github.com/searxng/searxng/issues/3191
- https://github.com/searxng/searxng/issues/1237

Related:

- https://github.com/searxng/searxng-docker/issues/386
- https://github.com/inetol-infrastructure/searxng-container/issues/81
2025-08-09 23:03:30 +02:00

92 lines
2.6 KiB
Python

# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring,disable=missing-class-docstring,invalid-name
import pathlib
import os
import aiounittest
os.environ.pop('SEARXNG_SETTINGS_PATH', None)
os.environ['SEARXNG_DISABLE_ETC_SETTINGS'] = '1'
class SearxTestLayer:
"""Base layer for non-robot tests."""
__name__ = 'SearxTestLayer'
@classmethod
def setUp(cls):
pass
@classmethod
def tearDown(cls):
pass
@classmethod
def testSetUp(cls):
pass
@classmethod
def testTearDown(cls):
pass
class SearxTestCase(aiounittest.AsyncTestCase):
"""Base test case for non-robot tests."""
layer = SearxTestLayer
SETTINGS_FOLDER = pathlib.Path(__file__).parent / "unit" / "settings"
TEST_SETTINGS = "test_settings.yml"
def setUp(self):
self.init_test_settings()
def setattr4test(self, obj, attr, value):
"""setattr(obj, attr, value) but reset to the previous value in the
cleanup."""
previous_value = getattr(obj, attr)
def cleanup_patch():
setattr(obj, attr, previous_value)
self.addCleanup(cleanup_patch)
setattr(obj, attr, value)
def init_test_settings(self):
"""Sets ``SEARXNG_SETTINGS_PATH`` environment variable an initialize
global ``settings`` variable and the ``logger`` from a test config in
:origin:`tests/unit/settings/`.
"""
os.environ['SEARXNG_SETTINGS_PATH'] = str(self.SETTINGS_FOLDER / self.TEST_SETTINGS)
# pylint: disable=import-outside-toplevel
import searx
import searx.locales
import searx.plugins
import searx.search
import searx.webapp
# https://flask.palletsprojects.com/en/stable/config/#builtin-configuration-values
# searx.webapp.app.config["DEBUG"] = True
searx.webapp.app.config["TESTING"] = True # to get better error messages
searx.webapp.app.config["EXPLAIN_TEMPLATE_LOADING"] = True
searx.init_settings()
searx.plugins.initialize(searx.webapp.app)
# searx.search.initialize will:
# - load the engines and
# - initialize searx.network, searx.metrics, searx.processors and searx.search.checker
searx.search.initialize(
enable_checker=True,
check_network=True,
enable_metrics=searx.get_setting("general.enable_metrics"), # type: ignore
)
# pylint: disable=attribute-defined-outside-init
self.app = searx.webapp.app
self.client = self.app.test_client()