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 setUp(self):
TestBase.setUp(self)
self.user_orig = User()
self.user_orig.email = u""
self.user_orig.name = u"test1"
self.user_orig.vote_weight = 1
self.user_orig.created_time = datetime.datetime.now()
self.user_orig.created_user = 0
self.user_orig.last_modified_time = datetime.datetime.now()
self.user_orig.last_modified_user = 0
self.user_orig.status = 1
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 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()
