Fix tests

surrealdb · Feb 7, 2025 · 666f1c8 · 666f1c8
1 parent 092cdf5
commit 666f1c8
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 7 deletions.
diff --git a/tests/unit_tests/connections/insert/test_async_http.py b/tests/unit_tests/connections/insert/test_async_http.py
@@ -52,8 +52,9 @@ async def test_insert_record_id_result_error(self):
 
         with self.assertRaises(Exception) as context:
             _ = await self.connection.insert(record_id, self.insert_data)
+        e = str(context.exception)
         self.assertEqual(
-            "There was a problem with the database: Can not execute INSERT statement using value: user:tobie" in str(context.exception),
+            "There was a problem with the database: Can not execute INSERT statement using value" in e and "user:tobie" in e,
             True
         )
         await self.connection.query("DELETE user;")

diff --git a/tests/unit_tests/connections/insert/test_async_ws.py b/tests/unit_tests/connections/insert/test_async_ws.py
@@ -53,8 +53,9 @@ async def test_insert_record_id_result_error(self):
 
         with self.assertRaises(Exception) as context:
             _ = await self.connection.insert(record_id, self.insert_data)
+        e = str(context.exception)
         self.assertEqual(
-            "There was a problem with the database: Can not execute INSERT statement using value: user:tobie" in str(context.exception),
+            "There was a problem with the database: Can not execute INSERT statement using value" in e and "user:tobie" in e,
             True
         )
         await self.connection.query("DELETE user;")

diff --git a/tests/unit_tests/connections/insert/test_blocking_http.py b/tests/unit_tests/connections/insert/test_blocking_http.py
@@ -57,8 +57,10 @@ def test_insert_record_id_result_error(self):
 
         with self.assertRaises(Exception) as context:
             _ = self.connection.insert(record_id, self.insert_data)
-        self.assertTrue(
-            "There was a problem with the database: Can not execute INSERT statement using value: user:tobie" in str(context.exception)
+        e = str(context.exception)
+        self.assertEqual(
+            "There was a problem with the database: Can not execute INSERT statement using value" in e and "user:tobie" in e,
+            True
         )
         self.connection.query("DELETE user;")
 

diff --git a/tests/unit_tests/connections/insert/test_blocking_ws.py b/tests/unit_tests/connections/insert/test_blocking_ws.py
@@ -56,9 +56,10 @@ def test_insert_record_id_result_error(self):
 
         with self.assertRaises(Exception) as context:
             _ = self.connection.insert(record_id, self.insert_data)
-        self.assertTrue(
-            "There was a problem with the database: Can not execute INSERT statement using value: user:tobie"
-            in str(context.exception)
+        e = str(context.exception)
+        self.assertEqual(
+            "There was a problem with the database: Can not execute INSERT statement using value" in e and "user:tobie" in e,
+            True
         )