some test fixes

Author: bubriks

python/tests/core/test_kafka_engine.py

@@ -340,7 +340,7 @@ def test_kafka_get_offsets_high(self, mocker):
         )
 
         # Assert
-        assert result == f" -initialCheckPointString {topic_name},0:11"
+        assert result == f"{topic_name},0:11"
 
     def test_kafka_get_offsets_low(self, mocker):
         # Arrange
@@ -372,7 +372,7 @@ def test_kafka_get_offsets_low(self, mocker):
         )
 
         # Assert
-        assert result == f" -initialCheckPointString {topic_name},0:0"
+        assert result == f"{topic_name},0:0"
 
     def test_kafka_get_offsets_no_topic(self, mocker):
         # Arrange

python/tests/engine/test_python.py

@@ -2962,7 +2962,7 @@ def test_materialization_kafka_first_job_execution(self, mocker):
         mocker.patch("hsfs.util.get_job_url")
         mocker.patch(
             "hsfs.core.kafka_engine.kafka_get_offsets",
-            return_value=" tests_offsets",
+            return_value="tests_offsets",
         )
         mocker.patch(
             "hsfs.core.job_api.JobApi.last_execution",
@@ -3004,7 +3004,7 @@ def test_materialization_kafka_first_job_execution(self, mocker):
         # Assert
         assert mock_python_engine_kafka_produce.call_count == 4
         job_mock.run.assert_called_once_with(
-            args="defaults tests_offsets",
+            args="defaults -initialCheckPointString tests_offsets",
             await_termination=False,
         )
 
@@ -3020,7 +3020,7 @@ def test_materialization_kafka_skip_offsets(self, mocker):
         mocker.patch("hsfs.util.get_job_url")
         mocker.patch(
             "hsfs.core.kafka_engine.kafka_get_offsets",
-            return_value=" tests_offsets",
+            return_value="tests_offsets",
         )
 
         mocker.patch("hopsworks_common.client.get_instance")
@@ -3061,7 +3061,7 @@ def test_materialization_kafka_skip_offsets(self, mocker):
         # Assert
         assert mock_python_engine_kafka_produce.call_count == 4
         job_mock.run.assert_called_once_with(
-            args="defaults tests_offsets",
+            args="defaults -initialCheckPointString tests_offsets",
             await_termination=False,
         )
 
@@ -3077,7 +3077,7 @@ def test_materialization_kafka_topic_doesnt_exist(self, mocker):
         mocker.patch("hsfs.util.get_job_url")
         mocker.patch(
             "hsfs.core.kafka_engine.kafka_get_offsets",
-            side_effect=["", " tests_offsets"],
+            side_effect=["", "tests_offsets"],
         )
 
         mocker.patch("hopsworks_common.client.get_instance")
@@ -3115,7 +3115,7 @@ def test_materialization_kafka_topic_doesnt_exist(self, mocker):
         # Assert
         assert mock_python_engine_kafka_produce.call_count == 4
         job_mock.run.assert_called_once_with(
-            args="defaults tests_offsets",
+            args="defaults -initialCheckPointString tests_offsets",
             await_termination=False,
         )