feat(chat): add Querit web search provider (#17813)

test/unit_test/rag/utils/test_web_search_conn.py

@@ -0,0 +1,101 @@
+#
+#  Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+#
+
+from rag.utils import web_search_conn
+
+
+def test_create_web_search_provider_uses_existing_tavily_config_without_provider_field(monkeypatch):
+    created_with = []
+    provider = object()
+
+    monkeypatch.setattr(web_search_conn, "Tavily", lambda api_key: created_with.append(api_key) or provider)
+
+    result = web_search_conn.create_web_search_provider({"tavily_api_key": "tvly-test"})
+
+    assert result is provider
+    assert created_with == ["tvly-test"]
+
+
+def test_create_web_search_provider_uses_selected_querit_config(monkeypatch):
+    created_with = []
+    provider = object()
+
+    monkeypatch.setattr(web_search_conn, "Querit", lambda api_key: created_with.append(api_key) or provider)
+
+    result = web_search_conn.create_web_search_provider(
+        {
+            "web_search_provider": "querit",
+            "querit_api_key": "querit-test",
+            "tavily_api_key": "tvly-test",
+        }
+    )
+
+    assert result is provider
+    assert created_with == ["querit-test"]
+
+
+def test_create_web_search_provider_trims_selected_key(monkeypatch):
+    created_with = []
+    provider = object()
+
+    monkeypatch.setattr(web_search_conn, "Querit", lambda api_key: created_with.append(api_key) or provider)
+
+    result = web_search_conn.create_web_search_provider(
+        {
+            "web_search_provider": "querit",
+            "querit_api_key": "  querit-test  ",
+        }
+    )
+
+    assert result is provider
+    assert created_with == ["querit-test"]
+
+
+def test_create_web_search_provider_requires_key_for_selected_provider():
+    assert web_search_conn.create_web_search_provider({}) is None
+    assert web_search_conn.create_web_search_provider(None) is None
+    assert web_search_conn.create_web_search_provider({"web_search_provider": "tavily"}) is None
+    assert web_search_conn.create_web_search_provider({"web_search_provider": "querit"}) is None
+    assert web_search_conn.create_web_search_provider({"tavily_api_key": "   "}) is None
+    assert (
+        web_search_conn.create_web_search_provider(
+            {
+                "web_search_provider": "querit",
+                "querit_api_key": "   ",
+            }
+        )
+        is None
+    )
+
+
+def test_has_web_search_provider_follows_selected_provider():
+    assert web_search_conn.has_web_search_provider({"tavily_api_key": "tvly-test"})
+    assert not web_search_conn.has_web_search_provider({"tavily_api_key": ""})
+    assert web_search_conn.has_web_search_provider({"web_search_provider": "querit", "querit_api_key": "querit-test"})
+    assert not web_search_conn.has_web_search_provider(
+        {
+            "web_search_provider": "querit",
+            "querit_api_key": "",
+            "tavily_api_key": "tvly-test",
+        }
+    )
+    assert not web_search_conn.has_web_search_provider(
+        {
+            "web_search_provider": "unsupported",
+            "querit_api_key": "querit-test",
+            "tavily_api_key": "tvly-test",
+        }
+    )