Skip to content

Commit dad1eb9

Browse files
committed
tests(db): Add database connector initialization test
1 parent d064d99 commit dad1eb9

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

src/vectorcode/database/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
logger = logging.getLogger(name=__name__)
88

99

10-
def get_database_connector(config: Config) -> DatabaseConnectorBase: # pragma: nocover
10+
def get_database_connector(config: Config) -> DatabaseConnectorBase:
1111
"""
1212
It's CRUCIAL to keep the `import`s of the database connectors in the branches.
1313
This allow them to be lazy-imported. This also allow us to keep the main package

tests/database/test_db_init.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
import pytest
2+
3+
from vectorcode.cli_utils import Config
4+
from vectorcode.database import get_database_connector
5+
from vectorcode.database.chroma0 import ChromaDB0Connector
6+
7+
8+
def test_get_database_connector():
9+
assert isinstance(
10+
get_database_connector(Config(db_type="ChromaDB0")), ChromaDB0Connector
11+
)
12+
13+
14+
def test_get_database_connector_invalid_type():
15+
with pytest.raises(ValueError):
16+
get_database_connector(Config(db_type="InvalidDB"))

0 commit comments

Comments
 (0)