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('初始化完毕')