Set up mongo connection in pytest & fix authentication url

This commit is contained in:
stijndcl 2022-07-25 21:08:06 +02:00
parent 52b452c85a
commit 6bebd109bb
2 changed files with 17 additions and 4 deletions

View file

@ -2,10 +2,12 @@ import asyncio
from typing import AsyncGenerator, Generator
from unittest.mock import MagicMock
import motor.motor_asyncio
import pytest
from sqlalchemy.ext.asyncio import AsyncSession
from database.engine import postgres_engine
import settings
from database.engine import mongo_client, postgres_engine
from database.migrations import ensure_latest_migration, migrate
from didier import Didier
@ -54,6 +56,14 @@ async def postgres(tables) -> AsyncGenerator[AsyncSession, None]:
await connection.close()
@pytest.fixture
async def mongodb() -> motor.motor_asyncio.AsyncIOMotorDatabase:
"""Fixture to get a MongoDB connection"""
database = mongo_client[settings.MONGO_DB]
yield database
mongo_client.drop_database(settings.MONGO_DB)
@pytest.fixture
def mock_client() -> Didier:
"""Fixture to get a mock Didier instance