Mock the environment variables for testing.
(monkeypatch_session)
| 107 | |
| 108 | @pytest.fixture(scope="session") |
| 109 | def mock_session_env_ollama(monkeypatch_session): |
| 110 | """Mock the environment variables for testing.""" |
| 111 | # Note that this does *not* clear existing env variables by default- |
| 112 | # we used to specify clear=True but this caused issues with Playwright tests |
| 113 | # https://github.com/microsoft/playwright-python/issues/2506 |
| 114 | with mock.patch.dict(os.environ): |
| 115 | # Database |
| 116 | monkeypatch_session.setenv("POSTGRES_HOST", POSTGRES_HOST) |
| 117 | monkeypatch_session.setenv("POSTGRES_USERNAME", POSTGRES_USERNAME) |
| 118 | monkeypatch_session.setenv("POSTGRES_DATABASE", POSTGRES_DATABASE) |
| 119 | monkeypatch_session.setenv("POSTGRES_PASSWORD", POSTGRES_PASSWORD) |
| 120 | monkeypatch_session.setenv("POSTGRES_SSL", POSTGRES_SSL) |
| 121 | monkeypatch_session.setenv("RUNNING_IN_PRODUCTION", "False") |
| 122 | # Azure Subscription |
| 123 | monkeypatch_session.setenv("AZURE_SUBSCRIPTION_ID", "test-storage-subid") |
| 124 | # Ollama OpenAI |
| 125 | monkeypatch_session.setenv("OPENAI_CHAT_HOST", "ollama") |
| 126 | monkeypatch_session.setenv("OPENAI_EMBED_HOST", "ollama") |
| 127 | monkeypatch_session.setenv("OLLAMA_ENDPOINT", "http://host.docker.internal:11434/v1") |
| 128 | monkeypatch_session.setenv("OLLAMA_CHAT_MODEL", "llama3.1") |
| 129 | monkeypatch_session.setenv("OLLAMA_EMBED_MODEL", "nomic-embed-text") |
| 130 | monkeypatch_session.setenv("OLLAMA_EMBEDDING_COLUMN", "embedding_nomic") |
| 131 | |
| 132 | yield |
| 133 | |
| 134 | |
| 135 | async def create_and_seed_db(): |
nothing calls this directly
no outgoing calls
no test coverage detected