move websearch tests back to main tests suite (#5842)

deepset-ai · Sep 20, 2023 · 44f0c46 · 44f0c46
1 parent 57d33ee
commit 44f0c46
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 15 deletions.
diff --git a/e2e/preview/components/test_serperdev_websearch.py b/e2e/preview/components/test_serperdev_websearch.py
diff --git a/test/preview/components/websearch/test_serperdev.py b/test/preview/components/websearch/test_serperdev.py
@@ -1,3 +1,4 @@
+import os
 from unittest.mock import Mock, patch
 
 import pytest
@@ -176,3 +177,14 @@ def test_bad_response_code(self, mock_post):
 
         with pytest.raises(SerperDevError):
             ws.run(query="Who is the boyfriend of Olivia Wilde?")
+
+    @pytest.mark.skipif(
+        not os.environ.get("SERPERDEV_API_KEY", None),
+        reason="Export an env var called SERPERDEV_API_KEY containing the SerperDev API key to run this test.",
+    )
+    @pytest.mark.integration
+    def test_web_search():
+        ws = SerperDevWebSearch(api_key=os.environ.get("SERPERDEV_API_KEY", None), top_k=10)
+        results = ws.run(query="Who is the boyfriend of Olivia Wilde?")["documents"]
+        assert len(results) == 10
+        assert all(isinstance(doc, Document) for doc in results)