test: 补充知识处理、检索与 Agent 回归用例
覆盖切块、向量检索、工具调用和历史预算;模型相关用例使用桩。
This commit is contained in:
1 parent
2dd062787d
commit
d9fe727c3f
123 files changed
+29005
No files matched your search
@@ -0,0 +1,183 @@
|
||||
"""
|
||||
Copyright 2024-2026 ReplyPier
|
||||
|
||||
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.user import User
|
||||
from app.repositories.user import UserRepository
|
||||
from uuid import UUID, uuid4
|
||||
from app.models.organization import Organization
|
||||
|
||||
@pytest.fixture
|
||||
def user_repo(db):
|
||||
"""Create a user repository instance"""
|
||||
return UserRepository(db)
|
||||
|
||||
def test_create_user(user_repo, test_organization_id):
|
||||
"""Test creating a new user"""
|
||||
email = "test@example.com"
|
||||
password = "testpassword"
|
||||
full_name = "Test User"
|
||||
hashed_password = User.get_password_hash(password)
|
||||
|
||||
user = user_repo.create_user(
|
||||
email=email,
|
||||
hashed_password=hashed_password,
|
||||
full_name=full_name,
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
assert user.email == email
|
||||
assert user.full_name == full_name
|
||||
assert user.organization_id == test_organization_id
|
||||
assert user.verify_password(password) # Verify the password works
|
||||
|
||||
def test_get_user_by_email(user_repo, test_organization_id):
|
||||
"""Test retrieving a user by email"""
|
||||
# Create test user
|
||||
email = "test@example.com"
|
||||
hashed_password = User.get_password_hash("testpassword")
|
||||
user = user_repo.create_user(
|
||||
email=email,
|
||||
hashed_password=hashed_password,
|
||||
full_name="Test User",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
# Retrieve user
|
||||
retrieved_user = user_repo.get_user_by_email(email)
|
||||
assert retrieved_user.id == user.id
|
||||
assert retrieved_user.email == email
|
||||
|
||||
def test_get_user_by_id(user_repo, test_organization_id):
|
||||
"""Test retrieving a user by ID"""
|
||||
# Create test user
|
||||
hashed_password = User.get_password_hash("testpassword")
|
||||
user = user_repo.create_user(
|
||||
email="test@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name="Test User",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
# Retrieve user
|
||||
retrieved_user = user_repo.get_user(user.id)
|
||||
assert retrieved_user.id == user.id
|
||||
assert retrieved_user.email == user.email
|
||||
|
||||
def test_get_users_by_organization(user_repo, test_organization_id):
|
||||
"""Test retrieving all users for an organization"""
|
||||
# Create multiple users
|
||||
hashed_password = User.get_password_hash("testpassword")
|
||||
user1 = user_repo.create_user(
|
||||
email="user1@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name="User One",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
user2 = user_repo.create_user(
|
||||
email="user2@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name="User Two",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
# Create another organization for testing
|
||||
other_org = Organization(
|
||||
name="Other Organization",
|
||||
domain="other.com",
|
||||
timezone="UTC"
|
||||
)
|
||||
user_repo.db.add(other_org)
|
||||
user_repo.db.commit()
|
||||
user_repo.db.refresh(other_org)
|
||||
|
||||
# Create user in different organization
|
||||
user_repo.create_user(
|
||||
email="other@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name="Other User",
|
||||
organization_id=other_org.id
|
||||
)
|
||||
|
||||
# Get users for test organization
|
||||
users = user_repo.get_users_by_organization(test_organization_id)
|
||||
assert len(users) == 2
|
||||
assert all(u.organization_id == test_organization_id for u in users)
|
||||
|
||||
def test_update_user(user_repo, test_organization_id):
|
||||
"""Test updating a user"""
|
||||
# Create test user
|
||||
hashed_password = User.get_password_hash("testpassword")
|
||||
user = user_repo.create_user(
|
||||
email="test@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name="Test User",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
# Update user
|
||||
new_name = "Updated User"
|
||||
updated_user = user_repo.update_user(
|
||||
user_id=str(user.id),
|
||||
full_name=new_name
|
||||
)
|
||||
|
||||
assert updated_user.full_name == new_name
|
||||
assert updated_user.id == user.id
|
||||
|
||||
def test_authenticate_user(user_repo, test_organization_id):
|
||||
"""Test user authentication"""
|
||||
email = "test@example.com"
|
||||
password = "testpassword"
|
||||
hashed_password = User.get_password_hash(password)
|
||||
|
||||
# Create test user
|
||||
user = user_repo.create_user(
|
||||
email=email,
|
||||
hashed_password=hashed_password,
|
||||
full_name="Test User",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
|
||||
# Test valid password
|
||||
assert user.verify_password(password)
|
||||
|
||||
# Test invalid password
|
||||
assert not user.verify_password("wrongpassword")
|
||||
|
||||
def test_get_multi(user_repo, test_organization_id):
|
||||
"""Test retrieving multiple users with pagination"""
|
||||
# Create multiple users
|
||||
users = []
|
||||
hashed_password = User.get_password_hash("testpassword")
|
||||
for i in range(5):
|
||||
user = user_repo.create_user(
|
||||
email=f"user{i}@example.com",
|
||||
hashed_password=hashed_password,
|
||||
full_name=f"User {i}",
|
||||
organization_id=test_organization_id
|
||||
)
|
||||
users.append(user)
|
||||
|
||||
# Test pagination
|
||||
first_page = user_repo.get_users_by_organization(test_organization_id)[:2]
|
||||
assert len(first_page) == 2
|
||||
|
||||
second_page = user_repo.get_users_by_organization(test_organization_id)[2:4]
|
||||
assert len(second_page) == 2
|
||||
|
||||
last_page = user_repo.get_users_by_organization(test_organization_id)[4:]
|
||||
assert len(last_page) == 1
|
||||
Reference in new issue
Block a user