首先把默认的测试文件修改为符合自己的逻辑,比如这里
@pytest.fixture
def admin_user(db):
user = User(
username='admin',
email='admin@admin.com',
password='admin',
admin=True # 这里
)
db.session.add(user)
db.session.commit()
return user
def test_delete_user(client, db, user, admin_headers):
# test 404
rep = client.put("/api/v1/users/100000", headers=admin_headers)
assert rep.status_code == 404
db.session.add(user)
db.session.commit()
# test get_user
user_id = user.id
rep = client.delete(
'/api/v1/users/%d' % user_id,
headers=admin_headers
)
assert rep.status_code == 200
assert db.session.query(User).filter_by(id=user_id).first().active is False # 这里
测试Message
创建一个 test_message.py
详细查看代码
网友评论