def test_get(self, db):
users = Users(db['mongodb'])
result = users.get(db['user_id'])
_users = result.to_json()
assert result is not None
assert isinstance(_users, str)
def get_user(self, email):
"""Get user from DB and return User object"""
users = Users(self.db, self.uri)
result = users.get_by_email(email)
user = User.user(result)
return user
def test_update(self, db):
users = Users(db['mongodb'])
user = {}
user['first_name'] = "Aegon"
user['last_name'] = "Targaryen"
user['id'] = db['user_id']
result = users.update(user)
assert result == 1
def test_insert(self, db):
users = Users(db['mongodb'])
user = {
"first_name": "John",
"last_name": "Snow",
"email": "*****@*****.**",
"gender": "male",
"dob": "1984-07-31T00:38:21.702Z"
}
result = users.insert(user)
assert result is not None
def authenticate(self, username, password):
"""Authentication handler for JWT token"""
users = Users(self.db, self.uri)
user = users.get_by_email(username)
if user == {}:
return None
if util.verify_password(user['password'], password):
u = User.user(user)
del u.__dict__['password']
g.current_user = u # set current user in app context
return u
def test_delete(self, db):
users = Users(db['mongodb'])
result = users.delete(db['user_id'])
assert result is 1
def test_get_all(self, db):
users = Users(db['mongodb'])
result = users.get(db['user_id'])
assert result is not None
def test_by_email(self, db):
users = Users(db['mongodb'])
result = users.get_by_email(db['user']['email'])
assert result is not None
def identity(self, payload):
"""Identity handler for JWT token"""
users = Users(self.uri)
user_id = payload['identity']
return users.get(user_id)