def test_user_model(self): from motionman.models import User user_orig = User() user_orig.id = 888 user_orig.email = u"*****@*****.**" user_orig.name = u"test1" user_orig.vote_weight = 1 user_orig.created_time = datetime.datetime.now() user_orig.created_user = 3 user_orig.last_modified_time = datetime.datetime.now() user_orig.last_modified_user = 3 user_orig.status = 2 user_new = copy.deepcopy(user_orig) session = DBSession() session.add(user_new) session.flush() transaction.commit() user_load = session.query(User).filter(User.id==888).first() self.assertEqual(user_load.email, user_orig.email) self.assertEqual(user_load.name, user_orig.name) self.assertEqual(user_load.vote_weight, user_orig.vote_weight) self.assertEqual(user_load.created_time, user_orig.created_time) self.assertEqual(user_load.created_user, user_orig.created_user) self.assertEqual(user_load.last_modified_time, user_orig.last_modified_time) self.assertEqual(user_load.last_modified_user, user_orig.last_modified_user) self.assertEqual(user_load.status, user_orig.status) session.delete(user_load) session.flush() transaction.commit()
def add_user(name, email, vote_weight): session = DBSession() dup_check = session.query(User).filter(and_(User.status==1,or_(User.name==name, User.email==email))).all() if len(dup_check) != 0: for i in dup_check: if i.name == name: raise DuplicatedName() if i.email == email: raise DuplicatedEmail() user = User() user.email = email user.name = name user.vote_weight = vote_weight try: session.add(user) session.flush() transaction.commit() except IntegrityError: transaction.abort()
def test_user_model(self): from motionman.models import User user_orig = User() user_orig.id = 888 user_orig.email = u"*****@*****.**" user_orig.name = u"test1" user_orig.vote_weight = 1 user_orig.created_time = datetime.datetime.now() user_orig.created_user = 3 user_orig.last_modified_time = datetime.datetime.now() user_orig.last_modified_user = 3 user_orig.status = 2 user_new = copy.deepcopy(user_orig) session = DBSession() session.add(user_new) session.flush() transaction.commit() user_load = session.query(User).filter(User.id == 888).first() self.assertEqual(user_load.email, user_orig.email) self.assertEqual(user_load.name, user_orig.name) self.assertEqual(user_load.vote_weight, user_orig.vote_weight) self.assertEqual(user_load.created_time, user_orig.created_time) self.assertEqual(user_load.created_user, user_orig.created_user) self.assertEqual(user_load.last_modified_time, user_orig.last_modified_time) self.assertEqual(user_load.last_modified_user, user_orig.last_modified_user) self.assertEqual(user_load.status, user_orig.status) session.delete(user_load) session.flush() transaction.commit()
def add_user(name, email, vote_weight): session = DBSession() dup_check = session.query(User).filter( and_(User.status == 1, or_(User.name == name, User.email == email))).all() if len(dup_check) != 0: for i in dup_check: if i.name == name: raise DuplicatedName() if i.email == email: raise DuplicatedEmail() user = User() user.email = email user.name = name user.vote_weight = vote_weight try: session.add(user) session.flush() transaction.commit() except IntegrityError: transaction.abort()