@@ -21,9 +21,9 @@ async def test_create_postgres_engine(mock_session_env, mock_azure_credential):
21
21
azure_credential = mock_azure_credential ,
22
22
)
23
23
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" )
27
27
assert engine .url .query ["ssl" ] == "prefer"
28
28
29
29
@@ -33,9 +33,9 @@ async def test_create_postgres_engine_from_env(mock_session_env, mock_azure_cred
33
33
azure_credential = mock_azure_credential ,
34
34
)
35
35
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" )
39
39
assert engine .url .query ["ssl" ] == "prefer"
40
40
41
41
@@ -57,7 +57,7 @@ async def test_create_postgres_engine_from_args(mock_azure_credential):
57
57
azure_credential = mock_azure_credential ,
58
58
)
59
59
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" )
63
63
assert engine .url .query ["ssl" ] == "prefer"
0 commit comments