@@ -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