klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix skipped tests (#1385)

Browse Source
This commit is contained in:
Dev Khant
2024-06-07 22:56:54 +05:30
committed by GitHub
parent e0d0f6e94c
commit 827d63d115
4 changed files with 38 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/vectordb/test_chroma_db.py
View File

@@ -34,15 +34,16 @@ def cleanup_db():
print("Error: %s - %s." % (e.filename, e.strerror))
@pytest.mark.skip(reason="ChromaDB client needs to be mocked")
def test_chroma_db_init_with_host_and_port(chroma_db):
settings = chroma_db.client.get_settings()
assert settings.chroma_server_host == "test-host"
assert settings.chroma_server_http_port == "1234"
@patch("embedchain.vectordb.chroma.chromadb.Client")
def test_chroma_db_init_with_host_and_port(mock_client):
chroma_db = ChromaDB(config=ChromaDbConfig(host="test-host", port="1234")) # noqa
called_settings: Settings = mock_client.call_args[0][0]
assert called_settings.chroma_server_host == "test-host"
assert called_settings.chroma_server_http_port == "1234"
@pytest.mark.skip(reason="ChromaDB client needs to be mocked")
def test_chroma_db_init_with_basic_auth():
@patch("embedchain.vectordb.chroma.chromadb.Client")
def test_chroma_db_init_with_basic_auth(mock_client):
chroma_config = {
"host": "test-host",
"port": "1234",
@@ -52,12 +53,17 @@ def test_chroma_db_init_with_basic_auth():
},
}
db = ChromaDB(config=ChromaDbConfig(**chroma_config))
settings = db.client.get_settings()
assert settings.chroma_server_host == "test-host"
assert settings.chroma_server_http_port == "1234"
assert settings.chroma_client_auth_provider == chroma_config["chroma_settings"]["chroma_client_auth_provider"]
assert settings.chroma_client_auth_credentials == chroma_config["chroma_settings"]["chroma_client_auth_credentials"]
ChromaDB(config=ChromaDbConfig(**chroma_config))
called_settings: Settings = mock_client.call_args[0][0]
assert called_settings.chroma_server_host == "test-host"
assert called_settings.chroma_server_http_port == "1234"
assert (
called_settings.chroma_client_auth_provider == chroma_config["chroma_settings"]["chroma_client_auth_provider"]
)
assert (
called_settings.chroma_client_auth_credentials
== chroma_config["chroma_settings"]["chroma_client_auth_credentials"]
)
@patch("embedchain.vectordb.chroma.chromadb.Client")
@@ -84,7 +90,6 @@ def test_app_init_with_host_and_port_none(mock_client):
assert called_settings.chroma_server_http_port is None
@pytest.mark.skip(reason="Logging setup needs to be fixed to make this test to work")
def test_chroma_db_duplicates_throw_warning(caplog):
db = ChromaDB(config=ChromaDbConfig(allow_reset=True, dir="test-db"))
app = App(config=AppConfig(collect_metrics=False), db=db)