75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
# test_integration_username.py
|
||
|
||
import pytest
|
||
import sys
|
||
import os
|
||
|
||
# 获取文件所在目录的上两级目录（EnglishPal）
|
||
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
|
||
sys.path.append(project_root)
|
||
|
||
from app.Login import add_user, check_username_availability
|
||
|
||
|
||
# 模拟数据库行为
|
||
class MockUserDB:
|
||
def __init__(self):
|
||
self.users = {}
|
||
|
||
def insert_user(self, username, password, start_date, expiry_date):
|
||
if username in self.users:
|
||
return False
|
||
self.users[username] = {
|
||
'password': password,
|
||
'start_date': start_date,
|
||
'expiry_date': expiry_date
|
||
}
|
||
return True
|
||
|
||
def get_user_by_username(self, username):
|
||
return self.users.get(username)
|
||
|
||
|
||
original_insert_user = None
|
||
original_get_user_by_username = None
|
||
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def setup_and_teardown():
|
||
global original_insert_user, original_get_user_by_username
|
||
|
||
from app.Login import insert_user as original_iu, get_user_by_username as original_gubu
|
||
original_insert_user = original_iu
|
||
original_get_user_by_username = original_gubu
|
||
|
||
mock_db = MockUserDB()
|
||
|
||
def mock_insert_user(**kwargs):
|
||
return mock_db.insert_user(**kwargs)
|
||
|
||
def mock_get_user_by_username(username):
|
||
return mock_db.get_user_by_username(username)
|
||
|
||
from app.Login import insert_user, get_user_by_username
|
||
insert_user = mock_insert_user
|
||
get_user_by_username = mock_get_user_by_username
|
||
|
||
yield
|
||
|
||
from app.Login import insert_user, get_user_by_username
|
||
insert_user = original_insert_user
|
||
get_user_by_username = original_get_user_by_username
|
||
|
||
|
||
def test_add_user_with_empty_username_fails():
|
||
with pytest.raises(ValueError) as exc_info:
|
||
add_user('', 'Password123')
|
||
assert "Attribute User.name is required" in str(exc_info.value)
|
||
|
||
|
||
def test_add_user_with_valid_username_succeeds():
|
||
result = add_user('validuser', 'Password123')
|
||
assert result is None # 如果 add_user 不返回值，则断言无异常即可
|
||
assert check_username_availability('validuser') is False # 用户已存在
|
||
|