def test_insert(client):
mimetype = 'application/json'
headers = {
'Content-Type': mimetype,
'Accept': mimetype
}
data = {
"number": "3"
}
response = client.post(
url_for("api.insert"),
data=json.dumps(data),
headers=headers
)
assert response.status_code == HTTPStatus.OK
assert "inserted" in response.json
number = response.json["inserted"]
assert isinstance(number, int)
assert number == int(data["number"])
assert Number.query.filter_by(value=number).first()
Number.query.delete()
session.commit()
def test_get_issued_numbers(client):
session.add(Number(value=5))
session.add(Number(value=6))
session.add(Number(value=7))
session.commit()
response = client.get(url_for('api.get_numbers'))
assert response.status_code == HTTPStatus.OK
assert "numbers" in response.json
numbers = response.json["numbers"]
assert isinstance(numbers, list)
assert len(numbers)
assert numbers == [5, 6, 7]
Number.query.delete()
session.commit()
def authenticate(username, password):
"""
用户登录，登录成功返回token，写将登录时间写入数据库；登录失败返回失败原因
:param username
:param password
:return: true_return(data=token.decode(), msg='登录成功')
"""
user = session.query(admin_db).filter_by(username=username).first()
if user is None:
return false_return(msg='找不到用户')
else:
if user.check_password(password):
login_time = int(time.time())
user.login_time = login_time
session.commit()
user_info = {"login_time": login_time, "uid": user.uid}
token = Auth.encode_auth_token(user_info)
return true_return(data=token.decode(), msg='登录成功')
else:
return false_return(msg='密码不正确')
def reset_numbers():
Number.query.delete()
session.commit()
def insert_number_in_db(number):
number_to_insert = Number(value=number)
session.add(number_to_insert)
session.commit()
return number_to_insert
def commit_db_session(response):
session.commit()
return response