fixed the RAg in test pipeline issue

tests/test_url_refresh.py

@@ -0,0 +1,115 @@
+"""Tests for URL source refresh endpoint."""
+import pytest
+from unittest.mock import MagicMock, patch, AsyncMock
+
+AUTH = {"Authorization": "Bearer test-token"}
+
+
+def _make_sb(source=None, chatbot_company="company-1"):
+    sb = MagicMock()
+    default_source = {
+        "id": "src-1",
+        "chatbot_id": "cb-1",
+        "url": "https://example.com/faq",
+        "status": "completed",
+        "page_title": "FAQ",
+        "chunk_count": 10,
+        "error_message": None,
+    }
+
+    def table_side(name):
+        m = MagicMock()
+        m.select.return_value = m
+        m.insert.return_value = m
+        m.update.return_value = m
+        m.delete.return_value = m
+        m.eq.return_value = m
+        m.in_.return_value = m
+        m.returning.return_value = m
+        m.limit.return_value = m
+        m.order.return_value = m
+
+        if name == "chatbots":
+            m.execute.return_value = MagicMock(data=[{
+                "id": "cb-1",
+                "company_id": chatbot_company,
+                "qdrant_collection_name": "col-1",
+            }])
+        elif name == "companies":
+            m.execute.return_value = MagicMock(data=[{"id": chatbot_company, "owner_id": "user-1"}])
+        elif name == "url_sources":
+            m.execute.return_value = MagicMock(data=[source or default_source])
+        else:
+            m.execute.return_value = MagicMock(data=[], count=0)
+        return m
+
+    sb.table.side_effect = table_side
+    sb.auth = MagicMock()
+    return sb
+
+
+class TestUrlRefresh:
+    def _refresh(self, client, source=None):
+        with patch("app.routers.documents.get_supabase") as mock_sb, \
+             patch("app.routers.documents.vector_store") as mock_vs, \
+             patch("app.routers.documents.response_cache") as mock_cache, \
+             patch("app.dependencies.get_current_user") as mock_user, \
+             patch("app.routers.documents._process_url_source", new_callable=AsyncMock):
+            mock_sb.return_value = _make_sb(source=source)
+            mock_vs.delete_by_document_id = MagicMock()
+            mock_vs.collection_exists = MagicMock(return_value=True)
+            mock_cache.invalidate = MagicMock()
+            user = MagicMock()
+            user.id = "user-1"
+            mock_user.return_value = user
+            return client.post(
+                "/api/v1/chatbots/cb-1/url-sources/src-1/refresh",
+                headers=AUTH,
+            )
+
+    def test_returns_200(self, client):
+        resp = self._refresh(client)
+        assert resp.status_code == 200
+
+    def test_source_reset_to_pending(self, client):
+        resp = self._refresh(client)
+        body = resp.json()
+        assert body["status"] == "pending"
+        assert body["chunk_count"] == 0
+
+    def test_returns_404_for_unknown_source(self, client):
+        with patch("app.routers.documents.get_supabase") as mock_sb, \
+             patch("app.dependencies.get_current_user") as mock_user:
+            sb = _make_sb()
+            # Override url_sources to return empty
+            def table_side(name):
+                m = MagicMock()
+                m.select.return_value = m
+                m.update.return_value = m
+                m.eq.return_value = m
+                if name == "chatbots":
+                    m.execute.return_value = MagicMock(data=[{
+                        "id": "cb-1", "company_id": "company-1",
+                        "qdrant_collection_name": "col-1",
+                    }])
+                elif name == "companies":
+                    m.execute.return_value = MagicMock(data=[{"id": "company-1", "owner_id": "user-1"}])
+                else:
+                    m.execute.return_value = MagicMock(data=[])
+                return m
+            sb2 = MagicMock()
+            sb2.table.side_effect = table_side
+            sb2.auth = MagicMock()
+            mock_sb.return_value = sb2
+            user = MagicMock()
+            user.id = "user-1"
+            mock_user.return_value = user
+            resp = client.post(
+                "/api/v1/chatbots/cb-1/url-sources/no-such-src/refresh",
+                headers=AUTH,
+            )
+        assert resp.status_code == 404
+
+    def test_requires_authentication(self, client):
+        resp = client.post("/api/v1/chatbots/cb-1/url-sources/src-1/refresh")
+        assert resp.status_code == 401