update unit tests in 'tests/unit/test_worker'

tdspora · Nov 25, 2024 · 2f1a88c · 2f1a88c
1 parent 8dfdaa0
commit 2f1a88c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/tests/unit/test_worker/test_worker.py b/src/tests/unit/test_worker/test_worker.py
@@ -1936,6 +1936,8 @@ def test_launch_infer_of_not_pretrained_table_and_absent_success_file(
                 in caplog.text
             )
     mock_infer_table.assert_not_called()
+    mock_check_existence_of_destination.assert_called_once()
+    mock_validate_metadata.assert_called_once_with("test_table")
     rp_logger.info(SUCCESSFUL_MESSAGE)
 
 
@@ -1985,4 +1987,6 @@ def test_launch_infer_of_not_pretrained_table_and_success_file_with_wrong_conten
                 in caplog.text
             )
     mock_infer_table.assert_not_called()
+    mock_check_existence_of_destination.assert_called_once()
+    mock_validate_metadata.assert_called_once_with("test_table")
     rp_logger.info(SUCCESSFUL_MESSAGE)