def test_follow(self):
u1 = Users(username='******', email='*****@*****.**')
u2 = Users(username='******', email='*****@*****.**')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
self.assertEqual(u1.followed.all(), [])
self.assertEqual(u1.followers.all(), [])
u1.follow(u2)
db.session.commit()
self.assertTrue(u1.is_following(u2))
self.assertEqual(u1.followed.count(), 1)
self.assertEqual(u1.followed.first().username, 'susan')
self.assertEqual(u2.followers.count(), 1)
self.assertEqual(u2.followers.first().username, 'john')
u1.unfollow(u2)
db.session.commit()
self.assertFalse(u1.is_following(u2))
self.assertEqual(u1.followed.count(), 0)
self.assertEqual(u2.followers.count(), 0)
def test_follow(self):
user1 = Users(username='******')
user2 = Users(username='******')
db.session.add_all([user1, user2])
db.session.commit()
self.assertEqual(user1.followed.all(), [])
self.assertEqual(user2.followed.all(), [])
user1.follow(user2)
db.session.commit()
self.assertEqual(user1.followed.all(), [user2])
self.assertTrue(user1.is_following(user2))
self.assertEqual(user1.followed.first().username, 'Darcia')
self.assertEqual(user1.followed.count(), 1)
self.assertEqual(user2.followers.count(), 1)
self.assertEqual(user2.followers.first().username, 'Tylor')
user1.unfollow(user2)
db.session.commit()
self.assertFalse(user1.is_following(user2))
self.assertEqual(user1.followed.count(), 0)
self.assertEqual(user2.followers.count(), 0)
def test_follow(self):
# create a friend
u1 = Users(username='******', email='*****@*****.**', password= '******')
u2 = Users(username='******', email='*****@*****.**', password = '******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
assert u1.unfollow(u2) == None
u = u1.follow(u2)
db.session.add(u)
db.session.commit()
assert u1.follow(u2) == None
assert u1.is_following(u2)
assert u1.followed.count() == 1
assert u1.followed.first().username == 'susan'
assert u2.followers.count() == 1
assert u2.followers.first().username == 'john'
u = u1.unfollow(u2)
assert u != None
db.session.add(u)
db.session.commit()
assert u1.is_following(u2) == False
assert u1.followed.count() == 0
assert u2.followers.count() == 0
