Updated the TCs

Author: Your Name

sde_collections/tests/test_import_fulltexts.py

@@ -33,6 +33,8 @@ def mock_generator():
         yield (mock_batch)
 
     with patch("sde_collections.sinequa_api.Api.get_full_texts") as mock_get_full_texts, patch(
+        "sde_collections.sinequa_api.Api.get_total_count", return_value=2
+    ), patch(
         "sde_collections.utils.slack_utils.send_detailed_import_notification"
     ):
         mock_get_full_texts.return_value = mock_generator()
@@ -64,6 +66,8 @@ def mock_batch_generator():
             yield (create_batch(start, min(batch_size, total_records - start)))
 
     with patch("sde_collections.sinequa_api.Api.get_full_texts") as mock_get_full_texts, patch(
+        "sde_collections.sinequa_api.Api.get_total_count", return_value=20000
+    ), patch(
         "sde_collections.utils.slack_utils.send_detailed_import_notification"
     ):
         mock_get_full_texts.return_value = mock_batch_generator()