1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
import pytest
from pymongo import AsyncMongoClient
from beanie.odm.utils.pydantic import IS_PYDANTIC_V2
if IS_PYDANTIC_V2:
from pydantic_settings import BaseSettings
else:
from pydantic import BaseSettings
class Settings(BaseSettings):
mongodb_dsn: str = "mongodb://localhost:27017/beanie_db"
mongodb_db_name: str = "beanie_db"
@pytest.fixture
def settings():
return Settings()
@pytest.fixture
async def cli(settings):
client = AsyncMongoClient(settings.mongodb_dsn)
yield client
await client.close()
@pytest.fixture
def db(cli, settings):
return cli[settings.mongodb_db_name]
|
