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

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

backend/tests/repo/test_organization_repo.py

@@ -0,0 +1,197 @@
+"""
+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.models.organization import Organization
+from app.repositories.organization import OrganizationRepository
+
+@pytest.fixture
+def org_repo(db):
+    """Create an organization repository instance"""
+    return OrganizationRepository(db)
+
+def test_create_organization(org_repo):
+    """Test creating a new organization"""
+    name = "Test Organization"
+    domain = "test.com"
+    timezone = "UTC"
+    business_hours = {
+        "monday": {"start": "09:00", "end": "17:00"},
+        "tuesday": {"start": "09:00", "end": "17:00"}
+    }
+
+    org = org_repo.create_organization(
+        name=name,
+        domain=domain,
+        timezone=timezone,
+        business_hours=business_hours
+    )
+
+    assert org.name == name
+    assert org.domain == domain
+    assert org.timezone == timezone
+    assert org.business_hours == business_hours
+    assert org.is_active is True
+
+def test_get_organization(org_repo):
+    """Test retrieving an organization by ID"""
+    # Create test organization
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain="test.com",
+        timezone="UTC"
+    )
+
+    # Retrieve organization
+    retrieved_org = org_repo.get_organization(org.id)
+    assert retrieved_org.id == org.id
+    assert retrieved_org.name == org.name
+
+def test_get_organization_by_domain(org_repo):
+    """Test retrieving an organization by domain"""
+    domain = "test.com"
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain=domain,
+        timezone="UTC"
+    )
+
+    retrieved_org = org_repo.get_organization_by_domain(domain)
+    assert retrieved_org.id == org.id
+    assert retrieved_org.domain == domain
+
+def test_is_domain_taken(org_repo):
+    """Domain clashes ignore case and the org being updated"""
+    org = org_repo.create_organization(name="Test Organization", domain="test.com", timezone="UTC")
+
+    assert org_repo.is_domain_taken("TEST.com") is True
+    assert org_repo.is_domain_taken("test.com", exclude_org_id=org.id) is False
+    assert org_repo.is_domain_taken("other.com") is False
+
+def test_get_multi(org_repo):
+    """Test retrieving multiple organizations with pagination"""
+    # Create multiple organizations
+    orgs = []
+    for i in range(5):
+        org = org_repo.create_organization(
+            name=f"Organization {i}",
+            domain=f"org{i}.com",
+            timezone="UTC"
+        )
+        orgs.append(org)
+
+    # Test pagination
+    active_orgs = org_repo.get_active_organizations()
+    assert len(active_orgs) == 5
+
+def test_update_organization(org_repo):
+    """Test updating an organization"""
+    # Create test organization
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain="test.com",
+        timezone="UTC"
+    )
+
+    # Update organization
+    new_name = "Updated Organization"
+    new_settings = {"new_setting": "new_value"}
+    updated_org = org_repo.update_organization(
+        org_id=org.id,
+        name=new_name,
+        settings=new_settings
+    )
+
+    assert updated_org.name == new_name
+    assert updated_org.settings == new_settings
+    assert updated_org.id == org.id
+
+def test_delete_organization(org_repo):
+    """Test deactivating an organization"""
+    # Create test organization
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain="test.com",
+        timezone="UTC"
+    )
+
+    # Deactivate organization
+    success = org_repo.deactivate_organization(org.id)
+    assert success is True
+
+    # Verify organization is deactivated
+    deactivated_org = org_repo.get_organization(org.id)
+    assert deactivated_org is not None
+    assert deactivated_org.is_active is False
+
+def test_get_active_organizations(org_repo):
+    """Test retrieving only active organizations"""
+    # Create active and inactive organizations
+    active_org = org_repo.create_organization(
+        name="Active Organization",
+        domain="active.com",
+        timezone="UTC"
+    )
+    inactive_org = org_repo.create_organization(
+        name="Inactive Organization",
+        domain="inactive.com",
+        timezone="UTC"
+    )
+    org_repo.deactivate_organization(inactive_org.id)
+
+    # Get active organizations
+    active_orgs = org_repo.get_active_organizations()
+    assert len(active_orgs) == 1
+    assert active_orgs[0].id == active_org.id
+    assert active_orgs[0].is_active is True
+
+def test_organization_with_business_hours(org_repo):
+    """Test organization with business hours"""
+    business_hours = {
+        "monday": {"start": "09:00", "end": "17:00"},
+        "tuesday": {"start": "09:00", "end": "17:00"},
+        "wednesday": {"start": "09:00", "end": "17:00"},
+        "thursday": {"start": "09:00", "end": "17:00"},
+        "friday": {"start": "09:00", "end": "17:00"}
+    }
+
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain="test.com",
+        timezone="UTC",
+        business_hours=business_hours
+    )
+
+    retrieved_org = org_repo.get_organization(org.id)
+    assert retrieved_org.business_hours == business_hours
+
+def test_organization_with_settings(org_repo):
+    """Test organization with custom settings"""
+    settings = {
+        "chat_timeout": 300,
+        "max_sessions": 10,
+        "welcome_message": "Welcome to our organization!"
+    }
+
+    org = org_repo.create_organization(
+        name="Test Organization",
+        domain="test.com",
+        timezone="UTC"
+    )
+
+    # Update settings
+    updated_org = org_repo.update_settings(org.id, settings)
+    assert updated_org.settings == settings