예제 #1
0
    def test_pop_random(self):
        bounded_map = BoundedDecayingMap(self.io_loop, concurrency=4, window=Amount(5, Time.SECONDS))

        with pytest.raises(BoundedDecayingMap.Empty):
            bounded_map.pop_random()

        yield bounded_map.add("foo1", "peer1")
        yield bounded_map.add("foo2", "peer2")
        yield bounded_map.add("foo2", "peer3")
        assert len(bounded_map) == 3

        random_element = bounded_map.pop_random()
        assert random_element in frozenset([("foo1", "peer1"), ("foo2", "peer2"), ("foo2", "peer3")])
        assert len(bounded_map) == 2

        bounded_map = BoundedDecayingMap(self.io_loop, concurrency=4, window=Amount(5, Time.SECONDS))
        yield bounded_map.add("foo2", "peer1")
        yield bounded_map.add("foo2", "peer2")
        assert len(bounded_map) == 2
        bounded_map.pop_random()
        assert len(bounded_map) == 1
        bounded_map.pop_random()
        assert len(bounded_map) == 0
        with pytest.raises(BoundedDecayingMap.Empty):
            bounded_map.pop_random()
        assert len(bounded_map) == 0
예제 #2
0
    def test_concurrency(self):
        bounded_map = BoundedDecayingMap(self.io_loop, concurrency=1, window=Amount(5, Time.SECONDS))

        yield bounded_map.add("foo1", "peer1")
        assert bounded_map.contains("foo1", "peer1")

        self.io_loop.add_future(bounded_map.add("foo1", "peer2"), self.stop)
        self.io_loop.add_callback(bounded_map.remove, "foo1")
        self.wait()

        assert not bounded_map.contains("foo1", "peer1")
        assert bounded_map.contains("foo1", "peer2")