Skip to content

Commit 7db7fcd

Browse files
committed
Update PG tests
1 parent 06e3bd6 commit 7db7fcd

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/test_postgres_engine.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@ async def test_create_postgres_engine(mock_session_env, mock_azure_credential):
2121
azure_credential=mock_azure_credential,
2222
)
2323
assert engine.url.host == "localhost"
24-
assert engine.url.username == "admin"
25-
assert engine.url.database == "postgres"
26-
assert engine.url.password == "postgres"
24+
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
25+
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
26+
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
2727
assert engine.url.query["ssl"] == "prefer"
2828

2929

@@ -33,9 +33,9 @@ async def test_create_postgres_engine_from_env(mock_session_env, mock_azure_cred
3333
azure_credential=mock_azure_credential,
3434
)
3535
assert engine.url.host == "localhost"
36-
assert engine.url.username == "admin"
37-
assert engine.url.database == "postgres"
38-
assert engine.url.password == "postgres"
36+
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
37+
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
38+
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
3939
assert engine.url.query["ssl"] == "prefer"
4040

4141

@@ -57,7 +57,7 @@ async def test_create_postgres_engine_from_args(mock_azure_credential):
5757
azure_credential=mock_azure_credential,
5858
)
5959
assert engine.url.host == "localhost"
60-
assert engine.url.username == "admin"
61-
assert engine.url.database == "postgres"
62-
assert engine.url.password == "postgres"
60+
assert engine.url.username == os.environ["POSTGRES_USERNAME"]
61+
assert engine.url.database == os.environ["POSTGRES_DATABASE"]
62+
assert engine.url.password == os.environ.get("POSTGRES_PASSWORD")
6363
assert engine.url.query["ssl"] == "prefer"

0 commit comments

Comments
 (0)