def test_follow(self):
"""
Unit test for following who i want to follow.
"""
u1 = User(nickname='foll1', email='*****@*****.**')
u2 = User(nickname='foll2', email='*****@*****.**')
u1.make_a_hash('foll1')
u2.make_a_hash('foll2')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
assert u1.unfollow(u2) is None
u = u1.follow(u2)
db.session.add(u)
db.session.commit()
assert u1.follow(u2) is None
assert u1.is_following(u2)
assert u1.followed.count() == 1
assert u1.followed.first().nickname == 'foll2'
assert u2.followers.count() == 1
assert u2.followers.first().nickname == 'foll1'
u = u1.unfollow(u2)
assert u is not None
db.session.add(u)
db.session.commit()
assert not u1.is_following(u2)
assert u1.followed.count() == 0
assert u2.followers.count() == 0
def test_password_hash(self):
"""
Unit test for password hash with bcrypt.
"""
u = User(nickname='pass', email='*****@*****.**')
u.make_a_hash('passwordofpass')
assert u.check_password('passwordofpass')
def test_avatar(self):
u = User(nickname='john', email='*****@*****.**')
u.make_a_hash('john')
avatar = u.avatar(128)
expected = 'http://www.gravatar.com/avatar/' \
'd4c74594d841139328695756648b6bd6'
assert avatar[0:len(expected)] == expected
def test_register_with_blog_option(self):
"""
Unit test for blog publishing.
"""
u1 = User(nickname='john', email='*****@*****.**')
u2 = User(nickname='susan', email='*****@*****.**')
u1.make_a_hash('john')
u2.make_a_hash('susan')
db.session.add(u1)
db.session.add(u2)
opt1 = Option(user=u1)
opt2 = Option(user=u2)
db.session.add(opt1)
db.session.add(opt2)
db.session.commit()
# Test in setting side.
assert opt1.is_blog_publishing() is True
assert opt2.is_blog_publishing() is True
opt1.set_blog_status(False)
opt2.set_blog_status(False)
assert opt1.is_blog_publishing() is False
assert opt2.is_blog_publishing() is False
# Test in user side.
u1.opts.set_blog_status(True)
assert u1.opts.is_blog_publishing() is True
assert u2.opts.is_blog_publishing() is False
u2.opts.set_blog_status(True)
assert u2.opts.is_blog_publishing() is True
def test_followed_blog_posts(self):
"""
Unit test to get a list of followed people' blog post.
"""
utcnow = datetime.utcnow()
u1 = User(nickname='foll1', email='*****@*****.**')
u2 = User(nickname='foll2', email='*****@*****.**')
u1.make_a_hash('foll1')
u2.make_a_hash('foll2')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
bp1 = BlogPost(body='i\'m user 1',
timestamp=utcnow + timedelta(seconds=1),
blog_author=u1)
bp2 = BlogPost(body='i\'m user 2',
timestamp=utcnow + timedelta(seconds=2),
blog_author=u2)
u1.follow(u2)
db.session.add(bp1)
db.session.add(bp2)
db.session.commit()
f1 = u1.followed_blog_posts().all()
assert len(f1) == 1
assert f1 == [bp2]
def test_follow_posts(self):
"""
Unit test to get a list of followed people' timeline post.
"""
# make four users
u1 = User(nickname='john', email='*****@*****.**')
u2 = User(nickname='susan', email='*****@*****.**')
u3 = User(nickname='mary', email='*****@*****.**')
u4 = User(nickname='david', email='*****@*****.**')
u1.make_a_hash('john')
u2.make_a_hash('susan')
u3.make_a_hash('mary')
u4.make_a_hash('david')
db.session.add(u1)
db.session.add(u2)
db.session.add(u3)
db.session.add(u4)
# make four posts
utcnow = datetime.utcnow()
p1 = Post(body='post from john', author=u1,
timestamp=utcnow + timedelta(seconds=1))
p2 = Post(body='post from susan', author=u2,
timestamp=utcnow + timedelta(seconds=2))
p3 = Post(body='post from mary', author=u3,
timestamp=utcnow + timedelta(seconds=3))
p4 = Post(body='post from david', author=u4,
timestamp=utcnow + timedelta(seconds=4))
db.session.add(p1)
db.session.add(p2)
db.session.add(p3)
db.session.add(p4)
db.session.commit()
# setup the followers
u1.follow(u1) # john follows himself
u1.follow(u2) # john follows susan
u1.follow(u4) # john follows david
u2.follow(u2) # susan follows herself
u2.follow(u3) # susan follows mary
u3.follow(u3) # mary follows herself
u3.follow(u4) # mary follows david
u4.follow(u4) # david follows himself
db.session.add(u1)
db.session.add(u2)
db.session.add(u3)
db.session.add(u4)
db.session.commit()
# check the followed posts of each user
f1 = u1.followed_posts().all()
f2 = u2.followed_posts().all()
f3 = u3.followed_posts().all()
f4 = u4.followed_posts().all()
assert len(f1) == 3
assert len(f2) == 2
assert len(f3) == 2
assert len(f4) == 1
assert f1 == [p4, p2, p1]
assert f2 == [p3, p2]
assert f3 == [p4, p3]
assert f4 == [p4]
