コード例 #1
0
ファイル: test_concurrency.py プロジェクト: percolate/redset
    def _make_ss(self):
        class Serializer(object):
            loads = int
            dumps = str

        return ScheduledSet(
            redis.Redis(),
            self.set_name,
            serializer=Serializer(),
        )
コード例 #2
0
    def setUp(self):
        self.key = 'scheduled_set_test'
        self.now = time.time() - 1  # offset to avoid having to sleep

        self.ss = ScheduledSet(redis.Redis(), self.key)
コード例 #3
0
ファイル: test_sets.py プロジェクト: indatalabs/redset
    def setUp(self):
        self.key = 'scheduled_set_test'
        self.now = time.time() - 1  # offset to avoid having to sleep

        self.ss = ScheduledSet(redis.Redis(), self.key)
コード例 #4
0
class ScheduledSetTest(unittest.TestCase):
    def setUp(self):
        self.key = 'scheduled_set_test'
        self.now = time.time() - 1  # offset to avoid having to sleep

        self.ss = ScheduledSet(redis.Redis(), self.key)

    def tearDown(self):
        self.ss.clear()

    def test_schedule(self):
        self.ss.add(1, self.now)
        self.ss.add(2, self.now + 1000)

        next_item = self.ss.pop()
        self.assertEquals(next_item, '1')

        with self.assertRaises(KeyError):
            self.ss.pop()

        self.assertEquals(len(self.ss), 1)

    def test_peek(self):
        with self.assertRaises(KeyError):
            self.ss.peek()

        self.ss.add(1, self.now - 1000)
        self.ss.add(2, self.now)
        self.ss.add(3, self.now + 1000)

        self.assertEquals(
            self.ss.peek(),
            '1',
        )

        self.assertEquals(
            self.ss.peek(position=1),
            '2',
        )

        with self.assertRaises(KeyError):
            self.ss.peek(position=2)

        self.ss.pop()
        self.ss.pop()

        with self.assertRaises(KeyError):
            self.ss.peek()

        self.assertEquals(len(self.ss), 1)

    def test_take(self):
        self.ss.add('1', self.now - 3)
        self.ss.add('2', self.now - 2)
        self.ss.add('3', self.now - 1)

        items = self.ss.take(2)
        self.assertEquals(len(items), 2)
        self.assertEquals(['1', '2'], items)

        self.assertEquals(self.ss.pop(), '3')

        self.assertEquals(
            len(self.ss),
            0,
        )

        self.assertEquals(
            self.ss.take(0),
            [],
        )

        self.assertEquals(
            self.ss.take(-1),
            [],
        )

    def test_length(self):
        for i in range(2):
            self.ss.add(i, self.now + 50)

        for i in range(3):
            self.ss.add(i + 2, self.now - 50)

        self.assertEquals(
            len(self.ss),
            5,
        )

    def test_length_available(self):
        for i in range(2):
            self.ss.add(i, self.now + 50)

        for i in range(3):
            self.ss.add(i + 2, self.now - 50)

        self.assertEquals(
            self.ss.available(),
            3,
        )

    def test_contains(self):
        for i in range(5):
            self.ss.add(i)

        self.assertTrue(0 in self.ss)

        self.assertFalse(-1 in self.ss)

    def test_add_dup(self):
        for i in range(5):
            self.ss.add(i)

        dup_added_at = 10
        self.ss.add(0, score=dup_added_at)

        self.assertEquals(
            len(self.ss),
            5,
        )

        self.assertEquals(
            int(self.ss.score(0)),
            int(dup_added_at),
        )

    def test_clear(self):
        self.assertFalse(self.ss.clear())

        for i in range(5):
            self.ss.add(i)

        self.assertTrue(self.ss.clear())
        self.assertEquals(
            len(self.ss),
            0,
        )

    def test_discard(self):
        self.ss.add(0)
        self.ss.add(1, self.now + 50)

        self.assertTrue(self.ss.discard(0))
        self.assertFalse(self.ss.discard(0))

        self.assertTrue(self.ss.discard(1))
        self.assertFalse(self.ss.discard(1))

    def test_peek_score(self):
        self.assertEquals(
            None,
            self.ss.peek_score(),
        )

        for i in range(3):
            self.ss.add(i, self.now - i)

        self.assertEquals(
            int(self.now - 2),
            int(self.ss.peek_score()),
        )

        self.ss.pop()

        self.assertEquals(
            int(self.now - 1),
            int(self.ss.peek_score()),
        )
コード例 #5
0
ファイル: test_sets.py プロジェクト: indatalabs/redset
class ScheduledSetTest(unittest.TestCase):

    def setUp(self):
        self.key = 'scheduled_set_test'
        self.now = time.time() - 1  # offset to avoid having to sleep

        self.ss = ScheduledSet(redis.Redis(), self.key)

    def tearDown(self):
        self.ss.clear()

    def test_schedule(self):
        self.ss.add(1, self.now)
        self.ss.add(2, self.now + 1000)

        next_item = self.ss.pop()
        self.assertEquals(next_item, '1')

        with self.assertRaises(KeyError):
            self.ss.pop()

        self.assertEquals(len(self.ss), 1)

    def test_peek(self):
        with self.assertRaises(KeyError):
            self.ss.peek()

        self.ss.add(1, self.now - 1000)
        self.ss.add(2, self.now)
        self.ss.add(3, self.now + 1000)

        self.assertEquals(
            self.ss.peek(),
            '1',
        )

        self.assertEquals(
            self.ss.peek(position=1),
            '2',
        )

        with self.assertRaises(KeyError):
            self.ss.peek(position=2)

        self.ss.pop()
        self.ss.pop()

        with self.assertRaises(KeyError):
            self.ss.peek()

        self.assertEquals(len(self.ss), 1)

    def test_take(self):
        self.ss.add('1', self.now - 3)
        self.ss.add('2', self.now - 2)
        self.ss.add('3', self.now - 1)

        items = self.ss.take(2)
        self.assertEquals(len(items), 2)
        self.assertEquals(['1', '2'], items)

        self.assertEquals(self.ss.pop(), '3')

        self.assertEquals(
            len(self.ss),
            0,
        )

        self.assertEquals(
            self.ss.take(0),
            [],
        )

        self.assertEquals(
            self.ss.take(-1),
            [],
        )

    def test_length(self):
        for i in range(2):
            self.ss.add(i, self.now + 50)

        for i in range(3):
            self.ss.add(i + 2, self.now - 50)

        self.assertEquals(
            len(self.ss),
            5,
        )

    def test_length_available(self):
        for i in range(2):
            self.ss.add(i, self.now + 50)

        for i in range(3):
            self.ss.add(i + 2, self.now - 50)

        self.assertEquals(
            self.ss.available(),
            3,
        )

    def test_contains(self):
        for i in range(5):
            self.ss.add(i)

        self.assertTrue(
            0 in self.ss
        )

        self.assertFalse(
            -1 in self.ss
        )

    def test_add_dup(self):
        for i in range(5):
            self.ss.add(i)

        dup_added_at = 10
        self.ss.add(0, score=dup_added_at)

        self.assertEquals(
            len(self.ss),
            5,
        )

        self.assertEquals(
            int(self.ss.score(0)),
            int(dup_added_at),
        )

    def test_clear(self):
        self.assertFalse(self.ss.clear())

        for i in range(5):
            self.ss.add(i)

        self.assertTrue(self.ss.clear())
        self.assertEquals(
            len(self.ss),
            0,
        )

    def test_discard(self):
        self.ss.add(0)
        self.ss.add(1, self.now + 50)

        self.assertTrue(self.ss.discard(0))
        self.assertFalse(self.ss.discard(0))

        self.assertTrue(self.ss.discard(1))
        self.assertFalse(self.ss.discard(1))

    def test_peek_score(self):
        self.assertEquals(
            None,
            self.ss.peek_score(),
        )

        for i in range(3):
            self.ss.add(i, self.now - i)

        self.assertEquals(
            int(self.now - 2),
            int(self.ss.peek_score()),
        )

        self.ss.pop()

        self.assertEquals(
            int(self.now - 1),
            int(self.ss.peek_score()),
        )