mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-21 01:00:28 +02:00
8d443ada5b
* initial commit * almost done * finished 3 tests * minor refactor * built out initial permisison tests * reworked test_deletion * removed logging * all original tests have been converted * renamed user_groups to user_group * mypy * added test for doc set permissions * unified naming for manager methods * Refactored models and added new deletion test * minor additions * better logging+fixed input variables * commented out failed tests * Added readme * readme update * Added auth to IT set auth_type to basic and require_email_verification to false * Update run-it.yml * used verify and added to readme * added api key manager
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import os
|
|
from collections.abc import Generator
|
|
|
|
import pytest
|
|
from sqlalchemy.orm import Session
|
|
|
|
from danswer.db.engine import get_session_context_manager
|
|
from danswer.db.search_settings import get_current_search_settings
|
|
from tests.integration.common_utils.reset import reset_all
|
|
from tests.integration.common_utils.vespa import TestVespaClient
|
|
|
|
|
|
def load_env_vars(env_file: str = ".env") -> None:
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
env_path = os.path.join(current_dir, env_file)
|
|
try:
|
|
with open(env_path, "r") as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if line and not line.startswith("#"):
|
|
key, value = line.split("=", 1)
|
|
os.environ[key] = value.strip()
|
|
print("Successfully loaded environment variables")
|
|
except FileNotFoundError:
|
|
print(f"File {env_file} not found")
|
|
|
|
|
|
# Load environment variables at the module level
|
|
load_env_vars()
|
|
|
|
|
|
@pytest.fixture
|
|
def db_session() -> Generator[Session, None, None]:
|
|
with get_session_context_manager() as session:
|
|
yield session
|
|
|
|
|
|
@pytest.fixture
|
|
def vespa_client(db_session: Session) -> TestVespaClient:
|
|
search_settings = get_current_search_settings(db_session)
|
|
return TestVespaClient(index_name=search_settings.index_name)
|
|
|
|
|
|
@pytest.fixture
|
|
def reset() -> None:
|
|
reset_all()
|