def test_aggregated_feed(self):
loves = Love.objects.all()[:10]
feed = AggregatedFeed(13)
# slow version
activities = []
feed.delete()
for love in loves:
activity = Activity(love.user,
LoveVerb,
love,
love.user,
time=love.created_at,
extra_context=dict(hello='world'))
activities.append(activity)
feed.add(activity)
assert feed.contains(activity)
# so we have something to compare to
aggregator = RecentVerbAggregator()
aggregated_activities = aggregator.aggregate(activities)
# check the feed
feed_loves = feed[:20]
self.assertEqual(len(aggregated_activities), len(feed_loves))
# now the fast version
feed.delete()
self.assertEqual(int(feed.count()), 0)
feed.add_many(activities)
for activity in activities:
assert feed.contains(activity)
def test_add_remove(self):
'''
Try to remove an aggregated activity
'''
loves = Love.objects.all()[:1]
feed = AggregatedFeed(13)
# slow version
activities = []
feed.delete()
for love in loves:
activity = love.create_activity()
activities.append(activity)
feed.add(activity)
assert feed.contains(activity)
aggregated_activity = feed[:10][0]
feed.remove(aggregated_activity)
assert not feed.contains(activity)
