File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments