def initdb():
"""Init/reset database."""
db.drop_all()
db.create_all()
user = User(name='demo', email='*****@*****.**', password='******', website='', location='', bio='')
db.session.add(user)
db.session.commit()
def setUp(self):
super(TestSearch, self).setUp()
for i in range(25):
name = 'user%d' % i
email = '*****@*****.**' % name
user = User(name=name, email=email, password='******')
db.session.add(user)
db.session.commit()
def test_get_current_time(self):
user = User(name='tester', email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
last_created_time = user.created_time
user.name = 'another_tester'
db.session.add(user)
db.session.commit()
assert user.created_time == last_created_time
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 signup():
form = SignupForm(next=request.args.get('next'))
if form.validate_on_submit():
user = User()
form.populate_obj(user)
db.session.add(user)
db.session.commit()
if login_user(user):
return redirect(form.next.data or url_for('user.index'))
return render_template('signup.html', form=form)
def _make_user(self):
user = User(name='tester', email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
assert user.id is not None
self.user = user
