Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def test_import_anon_subcounts(self):
- def anon_subscribe():
- user = create_random_user()
- feed = FeedImporter().import_feed(self.feed, local=True)
- userprofile = UserProfile.objects.create(user=user)
- feed_type = djangolets.types.get_type("feed")
- box = Box(type=feed_type.name, config={"feed_id": feed.id})
- tab = userprofile.get_default_tab()
- tab[0].add_box(box)
- return user, feed, userprofile, box
- user1, feed1, profile1, box1 = anon_subscribe()
- request = MockRequest()
- import_anon_subcount(request, profile1)
- self.assertTrue(request.session.get("subscription_count_imported"))
- subs = AnonSubscriptions.objects.get(feed=feed1)
- self.assertEquals(subs.count, 1)
- user2, feed2, profile2, box2 = anon_subscribe()
- request = MockRequest()
- import_anon_subcount(request, profile2)
- self.assertTrue(request.session.get("subscription_count_imported"))
- subs = AnonSubscriptions.objects.get(feed=feed2)
- self.assertEquals(subs.count, 2)
Add Comment
Please, Sign In to add comment