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
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")
