def test_following(self): u1 = User.create(username='******') u2 = User.create(username='******') u3 = User.create(username='******') f1 = Follow.create(b=u2) u1.following.append(f1) f2 = Follow.create(b=u3) u1.following.append(f2) exp = [u2, u3] self.assertEqual(u1.get_following(), exp)
def test_followers(self): u1 = User.create(username='******') u2 = User.create(username='******') u3 = User.create(username='******') f2 = Follow.create(b=u2) u1.following.append(f2) f2 = Follow.create(b=u2) u3.following.append(f2) u1.save() u3.save() exp = [u1, u3] self.assertEqual(u2.get_followers(), exp)