def init_data(self):
demo = User(name=u'demo',
email=u'*****@*****.**',
password=u'123456',
role_code=USER,
status_code=ACTIVE,
user_detail=UserDetail(
sex_code=MALE,
age=10,
url=u'http://demo.example.com',
deposit=100.00,
location=u'Hangzhou',
bio=u'admin Guy is ... hmm ... just a demo guy.'))
admin = User(name=u'admin',
email=u'*****@*****.**',
password=u'123456',
role_code=ADMIN,
status_code=ACTIVE,
user_detail=UserDetail(
sex_code=MALE,
age=10,
url=u'http://admin.example.com',
deposit=100.00,
location=u'Hangzhou',
bio=u'admin Guy is ... hmm ... just a admin guy.'))
db.session.add(demo)
db.session.add(admin)
db.session.commit()
def signup(self):
user = User()
user.user_detail = UserDetail()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
return user
def test_follow_unfollow(self):
user1 = User(name='tester1',
email='*****@*****.**',
password='******')
db.session.add(user1)
user2 = User(name='tester2',
email='*****@*****.**',
password='******')
db.session.add(user2)
db.session.commit()
user1.follow(user2)
assert user1.num_following == 1
assert user1.get_following_query().first().id == user2.id
assert user2.num_followers == 1
assert user2.get_followers_query().first().id == user1.id
user1.unfollow(user2)
assert user1.num_following == 0
assert user2.num_followers == 0
def initdb():
"""Init/reset database."""
db.drop_all()
db.create_all()
admin = User(
name=u'admin',
email=u'*****@*****.**',
password=u'123456',
role_code=ADMIN,
status_code=ACTIVE,
user_detail=UserDetail(
sex_code=MALE,
age=10,
url=u'http://admin.example.com',
deposit=100.00,
location=u'Hangzhou',
bio=u'admin Guy is ... hmm ... just a admin guy.'))
db.session.add(admin)
db.session.commit()
def create_profile(self):
user = User()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
