Exemple #1
0
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
Exemple #2
0
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