def __test_save(self):
user = User(**self.test_data)
user.password = util.md5(user.password)
user.email = 'email'
# user.save(
with tools.assert_raises(mongoengine.errors.ValidationError):
user = user.save()
tools.assert_is_none(user.id)
user = User(**self.test_data)
user.password = util.md5(user.password)
user.save()
tools.assert_is_not_none(user.id)
self.id = str(user.id)
self.username = user.username
self.password = self.test_data['password']
def test_login(cls, test_data=None):
"""
测试登录是否成功，返回access_token
"""
if test_data is None:
test_data = user_data
user = User(**test_data)
user.password = util.md5(user.password)
user.save()
cls.user = user
test_user = {
'username': user.username,
'password': test_data.get('password', '')
}
data = json.dumps(test_user)
response = test_app.post('/api/v1/login',
data=data,
content_type='application/json')
json_resp = json.loads(response.data)
cls.id = str(user.id)
cls.token = f'JWT {json_resp["data"]["access_token"]}'