class CreateLocationTestCase(unittest.TestCase): def setUp(self) -> None: """清空数据库,注册用户。""" self.location_repository = DatabaseLocationRepository( connection=mysql_connection, ) self.user_repository = DatabaseUserRepository( connection=mysql_connection, ) self.location_repository.clear() self.user_repository.clear() register_user(self.location_repository, self.user_repository) def tearDown(self) -> None: self.location_repository.clear() self.user_repository.clear() def test_create_location(self): """测试创建一个地点。""" with main.app.test_client() as client: client.post('/user/login', json={ 'email': '*****@*****.**', 'password': '******', }) name = '工地' rv = client.post('/location', json={ 'name': name, }) json_data = rv.get_json() assert json_data['result']['id'] assert json_data['result']['name'] == name
def client(): with main.app.test_client() as client: yield client # 在这里把数据库中插入的数据删掉 if _user_id: user_repository = DatabaseUserRepository(mysql_connection) user_repository.remove(_user_id)
def setUp(self) -> None: """清空数据库,注册用户。""" self.location_repository = DatabaseLocationRepository( connection=mysql_connection, ) self.user_repository = DatabaseUserRepository( connection=mysql_connection, ) self.location_repository.clear() self.user_repository.clear() register_user(self.location_repository, self.user_repository)
def user(self): return DatabaseUserRepository(self.mysql_connection)
from nest.repository.location import DatabaseLocationRepository from nest.repository.task import DatabaseTaskRepository from nest.repository.user import DatabaseUserRepository from nest.web import main from .user_helper import register_user from tests.web.helper import mysql_connection _task_id = None location_repository = DatabaseLocationRepository( connection=mysql_connection, ) task_repository = DatabaseTaskRepository( connection=mysql_connection, ) user_repository = DatabaseUserRepository( connection=mysql_connection, ) def clear_database(): task_repository.clear() user_repository.clear() # 写法来自这里:https://docs.pytest.org/en/stable/xunit_setup.html def setup_module(): clear_database() register_user(location_repository, user_repository) print('初始化完毕')