def test_simple_remove_love(self): from entity.models import Love target_loves = Love.objects.all()[:10] feed = LoveFeed(13) feed.delete() # slow implementation activities = [] for love in target_loves: # remove the items by key (id) activity = Activity(love.user, LoveVerb, love, love.user, time=love.created_at, extra_context=dict(hello='world')) activities.append(activity) feed.remove(activity) feed.add_many(activities) for activity in activities: assert feed.contains(activity) feed.remove_many(activities) assert feed.count() == 0 feed_loves = feed[:20]
def test_simple_add_love(self): loves = Love.objects.all()[:10] feed = LoveFeed(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) # close the feed feed.finish() feed_loves = feed[:20] assert isinstance(feed_loves[-1], FeedEndMarker) assert len(feed_loves) == 11 for activity in feed_loves: assert activity # now the fast version feed.delete() feed.add_many(activities) for activity in activities: assert feed.contains(activity)