@@ -144,7 +144,7 @@ def test_insert_csv_files(self, mock_file_descriptor_class, mock_csv_files):
144144 mock_ingest_client .ingest_from_file .side_effect = [mock_ingestion_result1 , mock_ingestion_result2 ]
145145
146146 # Act
147- insert_csv_files (files_data , mock_kusto_client , mock_ingest_client , runner_id , database , wait = False )
147+ insert_csv_files (files_data , mock_ingest_client , runner_id , database , wait = False )
148148
149149 # Assert
150150 # Verify that ingest_from_file was called for each CSV file
@@ -182,7 +182,7 @@ def test_insert_csv_files_with_wait(self, mock_file_descriptor_class, mock_check
182182 ]
183183
184184 # Act
185- insert_csv_files (files_data , mock_kusto_client , mock_ingest_client , runner_id , database , wait = True )
185+ insert_csv_files (files_data , mock_ingest_client , runner_id , database , wait = True )
186186
187187 # Assert
188188 # Verify that check_ingestion_status was called
@@ -223,16 +223,7 @@ def test_insert_csv_files_with_wait_max_retries(
223223 ]
224224
225225 # Act
226- insert_csv_files (
227- files_data ,
228- mock_kusto_client ,
229- mock_ingest_client ,
230- runner_id ,
231- database ,
232- wait = True ,
233- wait_limit = 2 ,
234- wait_duration = 0 ,
235- )
226+ insert_csv_files (files_data , mock_ingest_client , runner_id , database , wait = True , wait_limit = 2 , wait_duration = 0 )
236227
237228 # Assert
238229 # Verify that check_ingestion_status was called
@@ -319,7 +310,6 @@ def test_send_runner_data(
319310 # Verify that insert_csv_files was called
320311 mock_insert_csv_files .assert_called_once_with (
321312 files_data = mock_csv_content ,
322- kusto_client = mock_kusto_client ,
323313 ingest_client = mock_ingest_client ,
324314 runner_id = runner_id ,
325315 database = database_name ,
0 commit comments