|
@@ -62,7 +62,8 @@ class test_Persistent(StateResetCase):
|
|
self.assertTrue(self.p.db.closed)
|
|
self.assertTrue(self.p.db.closed)
|
|
|
|
|
|
def add_revoked(self, *ids):
|
|
def add_revoked(self, *ids):
|
|
- map(self.p.db.setdefault("revoked", LimitedSet()).add, ids)
|
|
|
|
|
|
+ for id in ids:
|
|
|
|
+ self.p.db.setdefault("revoked", LimitedSet()).add(id)
|
|
|
|
|
|
def test_merge(self, data=["foo", "bar", "baz"]):
|
|
def test_merge(self, data=["foo", "bar", "baz"]):
|
|
self.add_revoked(*data)
|
|
self.add_revoked(*data)
|
|
@@ -73,7 +74,8 @@ class test_Persistent(StateResetCase):
|
|
def test_sync(self, data1=["foo", "bar", "baz"],
|
|
def test_sync(self, data1=["foo", "bar", "baz"],
|
|
data2=["baz", "ini", "koz"]):
|
|
data2=["baz", "ini", "koz"]):
|
|
self.add_revoked(*data1)
|
|
self.add_revoked(*data1)
|
|
- map(state.revoked.add, data2)
|
|
|
|
|
|
+ for item in data2:
|
|
|
|
+ state.revoked.add(item)
|
|
self.p.sync(self.p.db)
|
|
self.p.sync(self.p.db)
|
|
|
|
|
|
for item in data2:
|
|
for item in data2:
|
|
@@ -92,7 +94,8 @@ class test_state(StateResetCase):
|
|
SimpleReq("bar"),
|
|
SimpleReq("bar"),
|
|
SimpleReq("baz"),
|
|
SimpleReq("baz"),
|
|
SimpleReq("baz")]):
|
|
SimpleReq("baz")]):
|
|
- map(state.task_accepted, requests)
|
|
|
|
|
|
+ for request in requests:
|
|
|
|
+ state.task_accepted(request)
|
|
for req in requests:
|
|
for req in requests:
|
|
self.assertIn(req, state.active_requests)
|
|
self.assertIn(req, state.active_requests)
|
|
self.assertEqual(state.total_count["foo"], 1)
|
|
self.assertEqual(state.total_count["foo"], 1)
|
|
@@ -101,7 +104,9 @@ class test_state(StateResetCase):
|
|
|
|
|
|
def test_ready(self, requests=[SimpleReq("foo"),
|
|
def test_ready(self, requests=[SimpleReq("foo"),
|
|
SimpleReq("bar")]):
|
|
SimpleReq("bar")]):
|
|
- map(state.task_accepted, requests)
|
|
|
|
|
|
+ for request in requests:
|
|
|
|
+ state.task_accepted(request)
|
|
self.assertEqual(len(state.active_requests), 2)
|
|
self.assertEqual(len(state.active_requests), 2)
|
|
- map(state.task_ready, requests)
|
|
|
|
|
|
+ for request in requests:
|
|
|
|
+ state.task_ready(requests)
|
|
self.assertEqual(len(state.active_requests), 0)
|
|
self.assertEqual(len(state.active_requests), 0)
|