Skip to content

Commit 4ad3ebe

Browse files
committed
fix test
1 parent 40c4ddf commit 4ad3ebe

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/tests_api/test_server_config.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,9 @@ def test_env_override(self):
3737
"RAGLIGHT_PERSIST_DIR": "/tmp/mydb",
3838
"RAGLIGHT_COLLECTION": "myproject",
3939
"RAGLIGHT_K": "10",
40-
"RAGLIGHT_CHROMA_HOST": "chromadb",
41-
"RAGLIGHT_CHROMA_PORT": "8001",
40+
"RAGLIGHT_DB": "chromadb",
41+
"RAGLIGHT_DB_PORT": "8001",
42+
"RAGLIGHT_DB_HOST": "localhost",
4243
}
4344
with patch.dict(os.environ, {**_clean_env(), **env}, clear=True):
4445
cfg = ServerConfig()
@@ -51,8 +52,9 @@ def test_env_override(self):
5152
self.assertEqual(cfg.persist_dir, "/tmp/mydb")
5253
self.assertEqual(cfg.collection, "myproject")
5354
self.assertEqual(cfg.k, 10)
54-
self.assertEqual(cfg.chroma_host, "chromadb")
55-
self.assertEqual(cfg.chroma_port, 8001)
55+
self.assertEqual(cfg.db_host, "localhost")
56+
self.assertEqual(cfg.db_port, 8001)
57+
self.assertEqual(cfg.db, "chroma")
5658

5759
def test_to_rag_config_returns_correct_type(self):
5860
with patch.dict(os.environ, _clean_env(), clear=True):

0 commit comments

Comments
 (0)