Unbreak test broken in #1185

MagicStack · Oct 17, 2024 · 8f2be4c · 8f2be4c
1 parent aa2d0e6
commit 8f2be4c
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
@@ -39,12 +39,18 @@ async def test_exceptions_str(self):
                  CREATE FUNCTION foo() RETURNS bool AS $$ $$ LANGUAGE SQL;
             ''')
         except asyncpg.InvalidFunctionDefinitionError as e:
-            self.assertEqual(
-                e.detail,
-                "Function's final statement must be SELECT or "
-                "INSERT/UPDATE/DELETE/MERGE RETURNING.")
-            self.assertIn(
-                'DETAIL:  Function', str(e)
-            )
+            if self.server_version < (17, 0):
+                detail = (
+                    "Function's final statement must be SELECT or "
+                    "INSERT/UPDATE/DELETE RETURNING."
+                )
+            else:
+                detail = (
+                    "Function's final statement must be SELECT or "
+                    "INSERT/UPDATE/DELETE/MERGE RETURNING."
+                )
+
+            self.assertEqual(e.detail, detail)
+            self.assertIn('DETAIL:  Function', str(e))
         else:
             self.fail('InvalidFunctionDefinitionError not raised')