SHOW:
|
|
- or go back to the newest paste.
| 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) |
