def test_follow(self):
user1 = User(username='******', email='*****@*****.**')
user2 = User(username='******', email='*****@*****.**')
db.session.add(user1)
db.session.add(user2)
db.session.commit()
self.assertEqual(user1.followed.all(), [])
self.assertEqual(user1.followers.all(), [])
user1.follow(user2)
db.session.commit()
self.assertTrue(user1.is_following(user2))
self.assertEqual(user1.followed.count(), 1)
self.assertEqual(user1.followed.first().username, 'Susan')
self.assertEqual(user2.followers.count(), 1)
self.assertEqual(user2.followers.first().username, 'John')
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):
u1 = User(username='******', email='*****@*****.**')
u2 = User(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, 'B')
self.assertEqual(u2.followers.count(), 1)
self.assertEqual(u2.followers.first().username, 'A')
u1.unfollow(u2)
db.session.commit()
self.assertFalse(u1.is_following(u2))
self.assertEqual(u1.followed.count(), 0)
self.assertEqual(u2.followers.count(), 0)