reinstated mocking unstructured

i-dot-ai · Sep 2, 2024 · 8025c09 · 8025c09
1 parent 06c874d
commit 8025c09
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/django_app/tests/management/test_commands.py b/django_app/tests/management/test_commands.py
@@ -240,10 +240,15 @@ def test_delete_expired_chats(chat: Chat, msg_1_date: datetime, msg_2_date: date
 
 
 @pytest.mark.django_db(transaction=True)
-def test_reingest_files(uploaded_file: File, mocker):
+def test_reingest_files(uploaded_file: File, requests_mock: Mocker, mocker):
     # Given
     assert uploaded_file.status == StatusEnum.processing
 
+    requests_mock.post(
+        f"http://{settings.UNSTRUCTURED_HOST}:8000/general/v0/general",
+        json=[{"text": "hello", "metadata": {"filename": "my-file.txt"}}],
+    )
+
     # When
     with mocker.patch("redbox.chains.ingest.VectorStore.add_documents", return_value=[]):
         call_command("reingest_files", sync=True)