Пример #1
0
  def test_user_follow(self):
    user1 = User(first_name="John", last_name="Test User 1", email="*****@*****.**", password="******")
    user2 = User(first_name="Joe", last_name="Test User 2", email="*****@*****.**", password="******")

    self.assertEquals(0, len(user1.followers))
    self.assertEquals(0, len(user1.followees))
    self.assertEquals(0, len(user2.followers))
    self.assertEquals(0, len(user2.followees))

    user1.follow(user2)

    self.assert_(user1.is_following(user2))
    self.assertEquals(1, len(user2.followers))
    self.assertEquals(1, len(user1.followees))
    self.assertEquals(0, len(user2.followees))
    self.assertEquals(0, len(user1.followers))
    self.assert_(user2 in user1.followees)

    user1.unfollow(user2)

    self.assert_(not user1.is_following(user2))
    self.assertEquals(0, len(user1.followers))
    self.assertEquals(0, len(user1.followees))
    self.assertEquals(0, len(user2.followers))
    self.assertEquals(0, len(user2.followees))
    self.assert_(user2 not in user1.followers)