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 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")