forked from Azure-Samples/rag-postgres-openai-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_postgres_engine.py
More file actions
63 lines (55 loc) · 2.26 KB
/
Copy pathtest_postgres_engine.py
File metadata and controls
63 lines (55 loc) · 2.26 KB
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import os
import pytest
from fastapi_app.postgres_engine import (
create_postgres_engine,
create_postgres_engine_from_args,
create_postgres_engine_from_env,
)
from tests.conftest import POSTGRES_DATABASE, POSTGRES_HOST, POSTGRES_PASSWORD, POSTGRES_SSL, POSTGRES_USERNAME
@pytest.mark.asyncio
async def test_create_postgres_engine(mock_session_env, mock_azure_credential):
engine = await create_postgres_engine(
host=os.environ["POSTGRES_HOST"],
username=os.environ["POSTGRES_USERNAME"],
database=os.environ["POSTGRES_DATABASE"],
password=os.environ.get("POSTGRES_PASSWORD"),
sslmode=os.environ.get("POSTGRES_SSL"),
azure_credential=mock_azure_credential,
)
assert engine.url.host == "localhost"
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
assert engine.url.query["ssl"] == "prefer"
@pytest.mark.asyncio
async def test_create_postgres_engine_from_env(mock_session_env, mock_azure_credential):
engine = await create_postgres_engine_from_env(
azure_credential=mock_azure_credential,
)
assert engine.url.host == "localhost"
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
assert engine.url.query["ssl"] == "prefer"
@pytest.mark.asyncio
async def test_create_postgres_engine_from_args(mock_azure_credential):
args = type(
"Args",
(),
{
"host": POSTGRES_HOST,
"username": POSTGRES_USERNAME,
"database": POSTGRES_DATABASE,
"password": POSTGRES_PASSWORD,
"sslmode": POSTGRES_SSL,
},
)
engine = await create_postgres_engine_from_args(
args=args,
azure_credential=mock_azure_credential,
)
assert engine.url.host == "localhost"
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
assert engine.url.query["ssl"] == "prefer"