class BaseTest(unittest.TestCase): def setUp(self): self.a = ActivityFeed(redis='redis://:@localhost:6379/15') self._empty() def _empty(self): a = self.a keys = a.redis.keys('{}*'.format(a.namespace)) if keys: a.redis.delete(*keys) def add_items_to_feed(self, user_id, items_to_add=5, aggregate=None): """Helper method to add items to a given feed. :param items_to_add: [int] Number of items to add to the feed. """ if aggregate is None: aggregate = self.a.aggregate now = timestamp_utcnow() for i in range(1, items_to_add + 1): self.a.update_item(user_id, i, now, aggregate) now += 5
class BaseTest(unittest.TestCase): def setUp(self): self.a = ActivityFeed(redis="redis://:@localhost:6379/15") self._empty() def _empty(self): a = self.a keys = a.redis.keys("{}*".format(a.namespace)) if keys: a.redis.delete(*keys) def add_items_to_feed(self, user_id, items_to_add=5, aggregate=None): """Helper method to add items to a given feed. :param items_to_add: [int] Number of items to add to the feed. """ if aggregate is None: aggregate = self.a.aggregate now = timestamp_utcnow() for i in range(1, items_to_add + 1): self.a.update_item(user_id, i, now, aggregate) now += 5
def test_feederboard_for(self): 'should create a leaderboard using an existing Redis connection' a = ActivityFeed() feederboard_david = a.feederboard_for('david') feederboard_person = a.feederboard_for('person') self.assertEqual(feederboard_david is None, False) self.assertEqual(feederboard_person is None, False)
def example_setup(): global a a = ActivityFeed(redis='redis://:@localhost:6379/15', items_loader=items_loader, aggregate=True) _empty(a) #now = timestamp_utcnow() #for i in range(1, 1200, 2): # create_item(a, 'foo', i, now) # now += 5 # create_item(a, 'bar', i+1, now) # a.aggregate_item('foo', i+1, now) # now += 5 return a
def test_key(self): 'should return the correct key for the non-aggregate feed' a = ActivityFeed() self.assertEquals(a.feed_key('david'), 'activity_feed:david')
def test_key_aggregate(self): 'should return the correct key for an aggregate feed' a = ActivityFeed() self.assertEqual(a.feed_key('david', True), 'activity_feed:aggregate:david')
def setUp(self): self.a = ActivityFeed(redis='redis://:@localhost:6379/15') self._empty()
def setUp(self): self.a = ActivityFeed(redis="redis://:@localhost:6379/15") self._empty()