Engage!

1 files changed, 21 insertions, 0 deletions

diff --git a/llama.cpp/tools/server/tests/conftest.py b/llama.cpp/tools/server/tests/conftest.py
new file mode 100644
index 0000000..c7ed775
--- /dev/null
+++ b/llama.cpp/tools/server/tests/conftest.py
@@ -0,0 +1,21 @@
+import pytest
+from utils import *
+
+
+# ref: https://stackoverflow.com/questions/22627659/run-code-before-and-after-each-test-in-py-test
+@pytest.fixture(autouse=True)
+def stop_server_after_each_test():
+    # do nothing before each test
+    yield
+    # stop all servers after each test
+    instances = set(
+        server_instances
+    )  # copy the set to prevent 'Set changed size during iteration'
+    for server in instances:
+        server.stop()
+
+
+@pytest.fixture(scope="module", autouse=True)
+def do_something():
+    # this will be run once per test session, before any tests
+    ServerPreset.load_all()