Test fixes

grai-io · Nov 13, 2023 · bc6ae5c · bc6ae5c
1 parent 77b18b6
commit bc6ae5c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/grai-server/app/connections/tests/test_tasks.py b/grai-server/app/connections/tests/test_tasks.py
@@ -259,6 +259,8 @@ def test_run_update_server_postgres_wrong_password(self, test_workspace, test_po
         assert (
             run.metadata["message"]
             == 'connection to server at "localhost" (127.0.0.1), port 5432 failed: FATAL:  password authentication failed for user "grai"\n'
+            or run.metadata["message"]
+            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  password authentication failed for user "grai"'
         )
 
     def test_run_update_server_postgres_no_database(self, test_workspace, test_postgres_connector, test_source):
@@ -287,7 +289,7 @@ def test_run_update_server_postgres_no_database(self, test_workspace, test_postg
             run.metadata["message"]
             == 'connection to server at "localhost" (127.0.0.1), port 5432 failed: FATAL:  database "wrong" does not exist\n'
             or run.metadata["message"]
-            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  password authentication failed for user "grai"'
+            == 'connection to server at "127.0.0.1", port 5432 failed: FATAL:  database "wrong" does not exist'
         )
 
     def test_run_update_server_no_connector(self, test_workspace, test_connector, test_source):