于小丘
f93cb3eedb
- Implemented unit tests for Object model including folder and file creation, properties, and path retrieval. - Added unit tests for Setting model covering creation, unique constraints, and type enumeration. - Created unit tests for User model focusing on user creation, uniqueness, and group relationships. - Developed unit tests for Login service to validate login functionality, including 2FA and token generation. - Added utility tests for JWT creation and verification, ensuring token integrity and expiration handling. - Implemented password utility tests for password generation, hashing, and TOTP verification.
453 lines
12 KiB
Python
453 lines
12 KiB
Python
"""
|
|
Object 模型的单元测试
|
|
"""
|
|
import pytest
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
from models.object import Object, ObjectType
|
|
from models.user import User
|
|
from models.group import Group
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_create_folder(db_session: AsyncSession):
|
|
"""测试创建目录"""
|
|
# 创建必要的依赖数据
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="testuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(
|
|
name="本地策略",
|
|
type=PolicyType.LOCAL,
|
|
server="/tmp/test"
|
|
)
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建目录
|
|
folder = Object(
|
|
name="测试目录",
|
|
type=ObjectType.FOLDER,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=0
|
|
)
|
|
folder = await folder.save(db_session)
|
|
|
|
assert folder.id is not None
|
|
assert folder.name == "测试目录"
|
|
assert folder.type == ObjectType.FOLDER
|
|
assert folder.size == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_create_file(db_session: AsyncSession):
|
|
"""测试创建文件"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="testuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(
|
|
name="本地策略",
|
|
type=PolicyType.LOCAL,
|
|
server="/tmp/test"
|
|
)
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建根目录
|
|
root = Object(
|
|
name=user.username,
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
root = await root.save(db_session)
|
|
|
|
# 创建文件
|
|
file = Object(
|
|
name="test.txt",
|
|
type=ObjectType.FILE,
|
|
parent_id=root.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=1024,
|
|
source_name="test_source.txt"
|
|
)
|
|
file = await file.save(db_session)
|
|
|
|
assert file.id is not None
|
|
assert file.name == "test.txt"
|
|
assert file.type == ObjectType.FILE
|
|
assert file.size == 1024
|
|
assert file.source_name == "test_source.txt"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_is_file_property(db_session: AsyncSession):
|
|
"""测试 is_file 属性"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="testuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
file = Object(
|
|
name="file.txt",
|
|
type=ObjectType.FILE,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=100
|
|
)
|
|
file = await file.save(db_session)
|
|
|
|
assert file.is_file is True
|
|
assert file.is_folder is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_is_folder_property(db_session: AsyncSession):
|
|
"""测试 is_folder 属性"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="testuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
folder = Object(
|
|
name="folder",
|
|
type=ObjectType.FOLDER,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
folder = await folder.save(db_session)
|
|
|
|
assert folder.is_folder is True
|
|
assert folder.is_file is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_get_root(db_session: AsyncSession):
|
|
"""测试 get_root() 方法"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="rootuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建根目录
|
|
root = Object(
|
|
name=user.username,
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
root = await root.save(db_session)
|
|
|
|
# 获取根目录
|
|
fetched_root = await Object.get_root(db_session, user.id)
|
|
|
|
assert fetched_root is not None
|
|
assert fetched_root.id == root.id
|
|
assert fetched_root.parent_id is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_get_by_path_root(db_session: AsyncSession):
|
|
"""测试获取根目录"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="pathuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建根目录
|
|
root = Object(
|
|
name=user.username,
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
root = await root.save(db_session)
|
|
|
|
# 通过路径获取根目录
|
|
result = await Object.get_by_path(db_session, user.id, "/pathuser", user.username)
|
|
|
|
assert result is not None
|
|
assert result.id == root.id
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_get_by_path_nested(db_session: AsyncSession):
|
|
"""测试获取嵌套路径"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="nesteduser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建目录结构: root -> docs -> work -> project
|
|
root = Object(
|
|
name=user.username,
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
root = await root.save(db_session)
|
|
|
|
docs = Object(
|
|
name="docs",
|
|
type=ObjectType.FOLDER,
|
|
parent_id=root.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
docs = await docs.save(db_session)
|
|
|
|
work = Object(
|
|
name="work",
|
|
type=ObjectType.FOLDER,
|
|
parent_id=docs.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
work = await work.save(db_session)
|
|
|
|
project = Object(
|
|
name="project",
|
|
type=ObjectType.FOLDER,
|
|
parent_id=work.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
project = await project.save(db_session)
|
|
|
|
# 获取嵌套路径
|
|
result = await Object.get_by_path(
|
|
db_session,
|
|
user.id,
|
|
"/nesteduser/docs/work/project",
|
|
user.username
|
|
)
|
|
|
|
assert result is not None
|
|
assert result.id == project.id
|
|
assert result.name == "project"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_get_by_path_not_found(db_session: AsyncSession):
|
|
"""测试路径不存在"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="notfounduser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建根目录
|
|
root = Object(
|
|
name=user.username,
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
await root.save(db_session)
|
|
|
|
# 获取不存在的路径
|
|
result = await Object.get_by_path(
|
|
db_session,
|
|
user.id,
|
|
"/notfounduser/nonexistent",
|
|
user.username
|
|
)
|
|
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_get_children(db_session: AsyncSession):
|
|
"""测试 get_children() 方法"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="childrenuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建父目录
|
|
parent = Object(
|
|
name="parent",
|
|
type=ObjectType.FOLDER,
|
|
parent_id=None,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
parent = await parent.save(db_session)
|
|
|
|
# 创建子对象
|
|
child1 = Object(
|
|
name="child1.txt",
|
|
type=ObjectType.FILE,
|
|
parent_id=parent.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=100
|
|
)
|
|
await child1.save(db_session)
|
|
|
|
child2 = Object(
|
|
name="child2",
|
|
type=ObjectType.FOLDER,
|
|
parent_id=parent.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
await child2.save(db_session)
|
|
|
|
# 获取子对象
|
|
children = await Object.get_children(db_session, user.id, parent.id)
|
|
|
|
assert len(children) == 2
|
|
child_names = {c.name for c in children}
|
|
assert child_names == {"child1.txt", "child2"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_parent_child_relationship(db_session: AsyncSession):
|
|
"""测试父子关系"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="reluser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建父目录
|
|
parent = Object(
|
|
name="parent",
|
|
type=ObjectType.FOLDER,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
parent = await parent.save(db_session)
|
|
|
|
# 创建子文件
|
|
child = Object(
|
|
name="child.txt",
|
|
type=ObjectType.FILE,
|
|
parent_id=parent.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=50
|
|
)
|
|
child = await child.save(db_session)
|
|
|
|
# 加载关系
|
|
loaded_child = await Object.get(
|
|
db_session,
|
|
Object.id == child.id,
|
|
load=Object.parent
|
|
)
|
|
|
|
assert loaded_child.parent is not None
|
|
assert loaded_child.parent.id == parent.id
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_object_unique_constraint(db_session: AsyncSession):
|
|
"""测试同目录名称唯一约束"""
|
|
from models.policy import Policy, PolicyType
|
|
|
|
group = Group(name="测试组")
|
|
group = await group.save(db_session)
|
|
|
|
user = User(username="uniqueuser", password="password", group_id=group.id)
|
|
user = await user.save(db_session)
|
|
|
|
policy = Policy(name="本地策略", type=PolicyType.LOCAL, server="/tmp/test")
|
|
policy = await policy.save(db_session)
|
|
|
|
# 创建父目录
|
|
parent = Object(
|
|
name="parent",
|
|
type=ObjectType.FOLDER,
|
|
owner_id=user.id,
|
|
policy_id=policy.id
|
|
)
|
|
parent = await parent.save(db_session)
|
|
|
|
# 创建第一个文件
|
|
file1 = Object(
|
|
name="duplicate.txt",
|
|
type=ObjectType.FILE,
|
|
parent_id=parent.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=100
|
|
)
|
|
await file1.save(db_session)
|
|
|
|
# 尝试在同一目录创建同名文件
|
|
file2 = Object(
|
|
name="duplicate.txt",
|
|
type=ObjectType.FILE,
|
|
parent_id=parent.id,
|
|
owner_id=user.id,
|
|
policy_id=policy.id,
|
|
size=200
|
|
)
|
|
|
|
with pytest.raises(IntegrityError):
|
|
await file2.save(db_session)
|