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])