From ebcc6c524f76246164e2efa03f3dbdca9555b7d5 Mon Sep 17 00:00:00 2001 From: Alexandre Teixeira <111787685+alteixeira20@users.noreply.github.com> Date: Tue, 22 Sep 2026 18:07:09 +0100 Subject: [PATCH] test(database): restore model endpoint isolation --- tests/test_research_endpoint_owner_scope.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_research_endpoint_owner_scope.py b/tests/test_research_endpoint_owner_scope.py index e30e5d994..a4a4b661b 100644 --- a/tests/test_research_endpoint_owner_scope.py +++ b/tests/test_research_endpoint_owner_scope.py @@ -15,6 +15,7 @@ import sys import types from types import SimpleNamespace from unittest.mock import MagicMock +import pytest # The helper resolves `from src.database import ModelEndpoint` at call time. # Stub the module so we can hand it a fake declarative class whose column @@ -73,12 +74,17 @@ class _DB: return _Query(self._rows) +@pytest.fixture(autouse=True) +def _isolate_model_endpoint(monkeypatch): + import src.database + monkeypatch.setattr(src.database, "ModelEndpoint", _ModelEndpoint) + + def _ep(eid, owner, *, is_enabled=True): return SimpleNamespace(id=eid, owner=owner, is_enabled=is_enabled, api_key="sk-secret") def _resolve(rows, owner, endpoint_id=None): - sys.modules["src.database"].ModelEndpoint = _ModelEndpoint return _owned_enabled_endpoint(_DB(rows), owner, endpoint_id)