test: 补充知识处理、检索与 Agent 回归用例

覆盖切块、向量检索、工具调用和历史预算；模型相关用例使用桩。

backend/tests/repo/test_knowledge_queue_repo.py

@@ -0,0 +1,150 @@
+"""
+Copyright 2024-2026 ChatterMate
+
+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.
+"""
+
+import pytest
+from app.repositories.knowledge_queue import KnowledgeQueueRepository
+from app.models.knowledge_queue import KnowledgeQueue, QueueStatus
+from app.models.user import User
+from uuid import uuid4
+
+@pytest.fixture
+def queue_repo(db):
+    return KnowledgeQueueRepository(db)
+
+@pytest.fixture
+def test_user(db, test_organization_id):
+    """Create a test user"""
+    user = User(
+        email="test@example.com",
+        hashed_password="hashed_password",
+        organization_id=test_organization_id,
+        is_active=True,
+        full_name="Test User"
+    )
+    db.add(user)
+    db.commit()
+    db.refresh(user)
+    return user
+
+@pytest.fixture
+def test_queue_item(db, test_organization_id, test_user):
+    """Create a test queue item"""
+    queue_item = KnowledgeQueue(
+        organization_id=test_organization_id,
+        user_id=test_user.id,
+        source_type="pdf_file",
+        source="/test/path/document.pdf",
+        status=QueueStatus.PENDING,
+        error=None,
+        queue_metadata={"test": "metadata"}
+    )
+    db.add(queue_item)
+    db.commit()
+    db.refresh(queue_item)
+    return queue_item
+
+def test_create_queue_item(queue_repo, test_organization_id, test_user):
+    """Test creating a new queue item"""
+    queue_item = KnowledgeQueue(
+        organization_id=test_organization_id,
+        user_id=test_user.id,
+        source_type="pdf_file",
+        source="/test/path/new.pdf",
+        status=QueueStatus.PENDING,
+        queue_metadata={"test": "metadata"}
+    )
+    
+    created_item = queue_repo.create(queue_item)
+    assert created_item is not None
+    assert created_item.source == "/test/path/new.pdf"
+    assert created_item.status == QueueStatus.PENDING
+    assert created_item.error is None
+
+def test_get_by_id(queue_repo, test_queue_item):
+    """Test retrieving a queue item by ID"""
+    item = queue_repo.get_by_id(test_queue_item.id)
+    assert item is not None
+    assert item.id == test_queue_item.id
+    assert item.source == "/test/path/document.pdf"
+    assert item.status == QueueStatus.PENDING
+
+def test_get_by_id_nonexistent(queue_repo):
+    """Test retrieving a nonexistent queue item"""
+    item = queue_repo.get_by_id(999)
+    assert item is None
+
+def test_get_pending(queue_repo, test_queue_item, db, test_organization_id, test_user):
+    """Test retrieving pending queue items"""
+    # Create another pending item
+    another_item = KnowledgeQueue(
+        organization_id=test_organization_id,
+        user_id=test_user.id,
+        source_type="pdf_file",
+        source="/test/path/another.pdf",
+        status=QueueStatus.PENDING,
+        queue_metadata={"test": "metadata"}
+    )
+    db.add(another_item)
+    
+    # Create a completed item
+    completed_item = KnowledgeQueue(
+        organization_id=test_organization_id,
+        user_id=test_user.id,
+        source_type="pdf_file",
+        source="/test/path/completed.pdf",
+        status=QueueStatus.COMPLETED,
+        queue_metadata={"test": "metadata"}
+    )
+    db.add(completed_item)
+    db.commit()
+
+    # Get pending items
+    pending_items = queue_repo.get_pending()
+    assert len(pending_items) == 2
+    assert all(item.status == QueueStatus.PENDING for item in pending_items)
+
+def test_update_status_success(queue_repo, test_queue_item):
+    """Test successfully updating queue item status"""
+    success = queue_repo.update_status(
+        test_queue_item.id,
+        QueueStatus.COMPLETED
+    )
+    assert success is True
+
+    # Verify the update
+    updated_item = queue_repo.get_by_id(test_queue_item.id)
+    assert updated_item.status == QueueStatus.COMPLETED
+    assert updated_item.error is None
+
+def test_update_status_with_error(queue_repo, test_queue_item):
+    """Test updating queue item status with error"""
+    error_message = "Processing failed"
+    success = queue_repo.update_status(
+        test_queue_item.id,
+        QueueStatus.FAILED,
+        error=error_message
+    )
+    assert success is True
+
+    # Verify the update
+    updated_item = queue_repo.get_by_id(test_queue_item.id)
+    assert updated_item.status == QueueStatus.FAILED
+    assert updated_item.error == error_message
+
+def test_update_status_nonexistent(queue_repo):
+    """Test updating status of nonexistent queue item"""
+    success = queue_repo.update_status(999, QueueStatus.COMPLETED)
+    assert success is False