[BUG] Shut down test system to hopefully fix test flakiness

amikos-tech · Mar 13, 2024 · 0e2776c · 0e2776c
1 parent 17d427b
commit 0e2776c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chromadb/test/db/test_system.py b/chromadb/test/db/test_system.py
@@ -120,6 +120,7 @@ def grpc_with_mock_server() -> Generator[SysDB, None, None]:
     system.start()
     client.reset_and_wait_for_ready()
     yield client
+    system.stop()
 
 
 def grpc_with_real_server() -> Generator[SysDB, None, None]:
@@ -145,7 +146,8 @@ def db_fixtures() -> List[Callable[[], Generator[SysDB, None, None]]]:
 
 @pytest.fixture(scope="module", params=db_fixtures())
 def sysdb(request: FixtureRequest) -> Generator[SysDB, None, None]:
-    yield next(request.param())
+    sysdb = request.param()
+    yield next(sysdb)
 
 
 # region Collection tests