def setUp(self):
self.u1 = User.objects.get_or_create(username='******')[0]
self.u2 = User.objects.get_or_create(username='******')[0]
self.u3 = User.objects.get_or_create(username='******')[0]
self.snippet = Snippet.objects.get_or_create(author=self.u1, title="Test")[0]
self.tag1 = Tag.objects.get_or_create(tag='python')[0]
self.tag2 = Tag.objects.get_or_create(tag='django')[0]
self.tag3 = Tag.objects.get_or_create(tag='c++')[0]
TaggedItem.objects.get_or_create(snippet=self.snippet, tag=self.tag1)
TaggedItem.objects.get_or_create(snippet=self.snippet, tag=self.tag2)
TaggedItem.objects.get_or_create(snippet=self.snippet, tag=self.tag3)
actions.follow(self.u1, self.u2)
actions.follow(self.u1, self.u3)
actions.follow(self.u2, self.u1)
actions.follow(self.u3, self.u1)
actions.follow(self.u3, self.u2)
actions.follow(self.u1, self.tag1)
actions.follow(self.u2, self.tag2)
actions.follow(self.u3, self.tag3)
actions.follow(self.u3, self.snippet)
actions.unfollow(self.u1, self.u3)
actions.unfollow(self.u3, self.tag3)
actions.unfollow(self.u3, self.snippet)
actions.unfollow(self.u3, self.u2)
def test_unfollow(self):
actions.unfollow(self.u1, self.u3)
actions.unfollow(self.u3, self.tag3)
actions.unfollow(self.u3, self.snippet)
actions.unfollow(self.u3, self.u2)
self.assertEqual(models.Follower.objects.followers(self.u3), [])
self.assertEqual(models.Follower.objects.following(self.u3), [self.u1])
