def test_store_user(self): from columns.model import User tmp = User.from_dict(dict( id=1, name=u'test_user', open_id=None, fb_id=None, twitter_id=None, type=1, profile=None, )) store_user(session,tmp) self.assertEquals(session['user_id'], tmp.id) self.assertEquals(session['user_name'], tmp.name) self.assertEquals(session['user_type'], tmp.type) self.assertEquals(session['user_profile'], tmp.profile) self.assertEquals(session['user_openid'], tmp.open_id) self.assertEquals(session['user_fbid'], tmp.fb_id) self.assertEquals(session['user_twitterid'], tmp.twitter_id)
def test_retrieve_user(self): from columns.model import User res = retrieve_user(session) self.assertEquals(res,None) tmp = User.from_dict(dict( id=1, name=u'test_user', open_id=None, fb_id=None, twitter_id=None, type=1, profile=None, )) store_user(session,tmp) res = retrieve_user(session) self.assertEquals(res['id'], tmp.id) self.assertEquals(res['name'], tmp.name) self.assertEquals(res['type'], tmp.type) self.assertEquals(res['profile'], tmp.profile) self.assertEquals(res['open_id'], tmp.open_id) self.assertEquals(res['fb_id'], tmp.fb_id) self.assertEquals(res['twitter_id'], tmp.twitter_id)