def test_get_user(self): user1 = User(username='******', email='*****@*****.**') user2 = User(username='******', email='*****@*****.**') db.session.add(user1) db.session.add(user2) db.session.commit() user = backend.get_user(user1.id) assert user['username'] == user1.username users = backend.get_user([user1.id, user2.id]) assert len(users) == 2
def test_get_user(self): user1 = User(username='******',email='*****@*****.**') user2 = User(username='******',email='*****@*****.**') db.session.add(user1) db.session.add(user2) db.session.commit() user = backend.get_user(user1.id) assert user['username'] == user1.username users = backend.get_user([user1.id,user2.id]) assert len(users) == 2
def cookie_auth(): if request.cookies.get('is_logined'): if authutil.is_logined(request): max_age = 3600*24*30*6 expires_time = int(time.time()) + max_age expires = strutil.cookie_date(expires_time) g.user = backend.get_user(int(request.cookie.get('ukey'))) @after_this_request def set_cookie(response): response.set_cookie('is_logined', 'True', max_age=max_age, expires=expires, path='/') return response else: g.user = {} @after_this_request def delete_cookie(response): response.delete_cookie('is_logined') return response else: g.user = {}