def testBlacklistSet(self):
blset = utils.BlacklistSet(0.1)
blset.update([1, 2])
time.sleep(0.3)
blset.add(3)
with self.assertRaises(KeyError):
blset.remove(2)
self.assertNotIn(1, blset)
self.assertNotIn(2, blset)
self.assertIn(3, blset)
blset.add(4)
time.sleep(0.3)
blset.add(5)
blset.add(6)
self.assertSetEqual({5, 6}, set(blset))
def test_blacklist_set():
blset = utils.BlacklistSet(0.1)
blset.update([1, 2])
time.sleep(0.3)
blset.add(3)
with pytest.raises(KeyError):
blset.remove(2)
assert 1 not in blset
assert 2 not in blset
assert 3 in blset
blset.add(4)
time.sleep(0.3)
blset.add(5)
blset.add(6)
assert {5, 6} == set(blset)
