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 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 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 reset_db(noinput=False):
"""Reset database"""
if not noinput:
print "WARNING: This will reset the database and may cause data loss."
response = raw_input("Are you sure you want to continue? (Yes/No) ")
if not response == "Yes":
print "Aborted."
sys.exit()
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(gender_code=MALE,
age=25,
url=u'http://example.com',
location=u'Kampala',
bio=u''))
db.session.add(admin)
english = Language(name="English",
iso639_1="en",
iso639_2="eng",
locale_code="en_UG")
db.session.add(english)
luganda = Language(name="Luganda",
iso639_1="lg",
iso639_2="lug",
locale_code="lg_UG")
db.session.add(luganda)
db.session.commit()
alembic_cfg = Config("alembic.ini")
command.stamp(alembic_cfg, "head")