diff --git a/tests/test_refinement_keys.py b/tests/test_refinement_keys.py index 1b2f21ac..261b4342 100644 --- a/tests/test_refinement_keys.py +++ b/tests/test_refinement_keys.py @@ -7,15 +7,23 @@ sys.path.insert(0, str(Path(__file__).resolve().parents[1])) class _FakeRoutes: + # Mirror every verb server.py registers (this fake is installed into sys.modules + # as `server`, so a real server.py import elsewhere registers against it). def get(self, _p): return lambda f: f def post(self, _p): return lambda f: f + def put(self, _p): + return lambda f: f + def delete(self, _p): return lambda f: f + def patch(self, _p): + return lambda f: f + sys.modules.setdefault("folder_paths", types.SimpleNamespace(models_dir="", get_user_directory=lambda: "")) sys.modules.setdefault(