jpmschweitzer/odysseus
1
0
Fork 0
mirror of https://github.com/pewdiepie-archdaemon/odysseus.git synced 2026-10-11 17:32:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

test(database): restore model endpoint isolation

Browse Source
This commit is contained in:
Alexandre Teixeira
2026-09-22 18:07:09 +01:00
parent dbba4978ee
commit ebcc6c524f
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_research_endpoint_owner_scope.py
+7 -1
View File
@@ -15,6 +15,7 @@ import sys
import types import types
from types import SimpleNamespace from types import SimpleNamespace
from unittest.mock import MagicMock from unittest.mock import MagicMock
import pytest
# The helper resolves `from src.database import ModelEndpoint` at call time. # 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 # Stub the module so we can hand it a fake declarative class whose column
@@ -73,12 +74,17 @@ class _DB:
return _Query(self._rows) 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): def _ep(eid, owner, *, is_enabled=True):
return SimpleNamespace(id=eid, owner=owner, is_enabled=is_enabled, api_key="sk-secret") return SimpleNamespace(id=eid, owner=owner, is_enabled=is_enabled, api_key="sk-secret")
def _resolve(rows, owner, endpoint_id=None): def _resolve(rows, owner, endpoint_id=None):
sys.modules["src.database"].ModelEndpoint = _ModelEndpoint
return _owned_enabled_endpoint(_DB(rows), owner, endpoint_id) return _owned_enabled_endpoint(_DB(rows), owner, endpoint_id)