thorpedosg

Untitled

Jul 24th, 2018
32
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. def test_import_anon_subcounts(self):
  2.  
  3. def anon_subscribe():
  4. user = create_random_user()
  5. feed = FeedImporter().import_feed(self.feed, local=True)
  6. userprofile = UserProfile.objects.create(user=user)
  7. feed_type = djangolets.types.get_type("feed")
  8.  
  9. box = Box(type=feed_type.name, config={"feed_id": feed.id})
  10. tab = userprofile.get_default_tab()
  11. tab[0].add_box(box)
  12.  
  13. return user, feed, userprofile, box
  14.  
  15. user1, feed1, profile1, box1 = anon_subscribe()
  16. request = MockRequest()
  17. import_anon_subcount(request, profile1)
  18. self.assertTrue(request.session.get("subscription_count_imported"))
  19. subs = AnonSubscriptions.objects.get(feed=feed1)
  20. self.assertEquals(subs.count, 1)
  21.  
  22. user2, feed2, profile2, box2 = anon_subscribe()
  23. request = MockRequest()
  24. import_anon_subcount(request, profile2)
  25. self.assertTrue(request.session.get("subscription_count_imported"))
  26. subs = AnonSubscriptions.objects.get(feed=feed2)
  27. self.assertEquals(subs.count, 2)
Add Comment
Please, Sign In to add comment