Skip to content

Commit b244594

Browse files
committed
fix failing test
Signed-off-by: Gabriel Costa <gabrielcg@proton.me>
1 parent c024657 commit b244594

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/unit/mcpgateway/test_admin_policy.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,8 @@ def _make_native(rules=None):
5353
"""Build a mock native engine with a given rules list."""
5454
native = MagicMock()
5555
native._rules = rules if rules is not None else []
56+
# Mock the rules property to return the _rules list
57+
type(native).rules = property(lambda self: self._rules)
5658
return native
5759

5860

@@ -62,6 +64,7 @@ def _make_pdp(native=MagicMock()):
6264

6365
pdp = MagicMock()
6466
pdp._engines = {EngineType.NATIVE: native}
67+
pdp.get_engine = MagicMock(return_value=native)
6568
pdp.cache_stats = MagicMock(return_value={"hits": 0, "misses": 0})
6669
pdp.health = AsyncMock(return_value=MagicMock(healthy=True, engines=[]))
6770
return pdp
@@ -106,6 +109,7 @@ async def test_uses_empty_rules_when_native_engine_absent(self):
106109

107110
pdp = MagicMock()
108111
pdp._engines = {} # no NATIVE engine
112+
pdp.get_engine = MagicMock(return_value=None) # Return None when engine is absent
109113
pdp.cache_stats = MagicMock(return_value={})
110114
pdp.health = AsyncMock(return_value=MagicMock())
111115

0 commit comments

Comments
 (0)