def logar(params): usuario = params.get('usuario') email = params.get('email') senha = params.get('senha') if senha and (usuario or email): hash = SHA256.new() hash.update(senha) query = {'senha': hash.hexdigest()} if usuario: query['username'] = usuario elif email: query['email'] = email else: abort(403) u_dict = Usuario.collection().find_one(query) u = Usuario() u = Usuario.make_usuario_from_dict(u, u_dict) if u: token_dict = Token.collection().find_one({'usuario': u.get_id(), 'expired': False}) if token_dict: t = Token().make_from_dict(token_dict) return t.token else: t = Token() t.token = Token.generate_token() t.usuario = u.get_id() t.expired = False Token.collection().insert(t.make_dict()) return t.token else: abort(403) else: abort(500)
def test_validate_token_expired(self, mock_from_json): token = Token() token.expired = True mock_from_json.return_value = token validate_func = mock.Mock() self.assertFalse(validate_token(token, "GET", None, validate_func)[0])
def test_token_expires_is_not_expired(self): token = Token() token.expired = False self.assertFalse(_is_expired_token(token))
def test_token_expires_is_expired(self): token = Token() token.expired = True self.assertTrue(_is_expired_token(token))