fix: search as client test (#83)

tattle-made · Feb 16, 2024 · 9855e96 · 9855e96
1 parent f6987a6
commit 9855e96
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/src/api/tests/sample_data/07ba4a2f-c0a2-44ba-96d8-7b4cc94c8ee7.mp4 b/src/api/tests/sample_data/07ba4a2f-c0a2-44ba-96d8-7b4cc94c8ee7.mp4
diff --git a/src/api/tests/sample_data/c8709f21-bd7d-4e22-af14-50ad8a429f84.jpeg b/src/api/tests/sample_data/c8709f21-bd7d-4e22-af14-50ad8a429f84.jpeg
diff --git a/src/api/tests/sample_data/people.jpg b/src/api/tests/sample_data/people.jpg
diff --git a/src/api/tests/sample_data/sample-cat-video.mp4 b/src/api/tests/sample_data/sample-cat-video.mp4
diff --git a/src/api/tests/test_index_api_as_client.py b/src/api/tests/test_index_api_as_client.py
@@ -80,28 +80,29 @@ def testIndexVideo(self):
             # "media": open("sample_data/simple-text.txt", "rb"),
             "data": json.dumps(data),
         }
-        # axios.post("endpoint", data)
         response = requests.post(url, json=data, headers=headers)
         print(response.text)
         self.assertEqual(response.status_code, 200)
 
     @skip
     def testIndexEnqueueImage(self):
         url = API_URL + "/search"
+        # headers = {"Content-Type": "application/json"}
+        headers = {"Content-Type": "multipart/form-data"}
         data = {
             "post": {
                 "id": "1234",
                 "media_type": "image",
-                "media_url": "https://fs.tattle.co.in/service/kosh/file/c8709f21-bd7d-4e22-af14-50ad8a429f84",
+                "media_url": "https://raw.githubusercontent.com/tattle-made/feluda/master/src/api/core/operators/sample_data/people.jpg",
                 "datasource_id": "asdfasdf-asdfasdf-asdf",
                 "client_id": "123-12312",
             },
             "metadata": {"domain": "hate_speech", "type": ["gender", "caste"]},
             "config": {"mode": "enqueue", "version": "0.1"},
         }
         files = {"data": json.dumps(data)}
-        response = requests.post(url, files=files)
-        print(response.json())
+        response = requests.post(url, json=data, headers=headers)
+        print(response.text)
         self.assertEqual(response.status_code, 200)
 
     @skip

diff --git a/src/api/tests/test_search_api_as_client.py b/src/api/tests/test_search_api_as_client.py
@@ -28,7 +28,7 @@ def testSearchRawQuery(self):
     def testSearchImage(self):
         url = API_URL + "/search"
         data = {"query_type": "image"}
-        with open("sample_data/c8709f21-bd7d-4e22-af14-50ad8a429f84.jpeg", "rb") as file:
+        with open("sample_data/people.jpg", "rb") as file:
             data = {"data": json.dumps(data)}
             files = {"media": file}
             response = requests.post(url, data=data, files=files)
@@ -39,7 +39,7 @@ def testSearchImage(self):
     # @skip
     def testIndexVideo(self):
         url = API_URL + "/search"
-        with open("sample_data/07ba4a2f-c0a2-44ba-96d8-7b4cc94c8ee7.mp4", "rb") as file:
+        with open("sample_data/sample-cat-video.mp4", "rb") as file:
             data = {"data": json.dumps({"query_type": "video"})}
             files = {"media": file}
             response = requests.post(url, data=data, files=files)