chore: fix tests

Signed-off-by: Sidhant Kohli <sidhant.kohli@gmail.com>
numaproj · Oct 30, 2024 · e627427 · e627427
1 parent 9cb5f0d
commit e627427
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/map/test_async_mapper.py b/tests/map/test_async_mapper.py
@@ -194,7 +194,7 @@ def test_map_grpc_error_no_handshake(self) -> None:
         except grpc.RpcError as e:
             logging.error(e)
             grpc_exception = e
-            self.assertTrue("MapFn: expected handshake message" in e.__str__())
+            self.assertTrue("MapFn: expected handshake as the first message" in e.__str__())
 
         self.assertEqual(0, len(responses))
         self.assertIsNotNone(grpc_exception)

diff --git a/tests/map/test_multiproc_mapper.py b/tests/map/test_multiproc_mapper.py
@@ -64,7 +64,7 @@ def test_udf_map_err_handshake(self):
                     break
 
         metadata, code, details = method.termination()
-        self.assertTrue("MapFn: expected handshake message" in details)
+        self.assertTrue("MapFn: expected handshake as the first message" in details)
         self.assertEqual(grpc.StatusCode.UNKNOWN, code)
 
     def test_udf_map_err(self):

diff --git a/tests/map/test_sync_mapper.py b/tests/map/test_sync_mapper.py
@@ -58,7 +58,7 @@ def test_udf_map_err_handshake(self):
                     break
 
         metadata, code, details = method.termination()
-        self.assertTrue("MapFn: expected handshake message" in details)
+        self.assertTrue("MapFn: expected handshake as the first message" in details)
         self.assertEqual(grpc.StatusCode.UNKNOWN, code)
 
     def test_udf_map_error_response(self):