diff options
-rw-r--r-- | bgproc.py | 1 | ||||
-rw-r--r-- | bgproc_tests.py | 4 |
2 files changed, 5 insertions, 0 deletions
@@ -23,6 +23,7 @@ class BackgroundProcessing(object): def enqueue_request(self, request): '''Put a request into queue, to be processed by workers whenever.''' + self.pending_requests += 1 def close_requests(self): '''Signal workers that they can retire. diff --git a/bgproc_tests.py b/bgproc_tests.py index 17bc4fc..9709636 100644 --- a/bgproc_tests.py +++ b/bgproc_tests.py @@ -41,3 +41,7 @@ class BackgroundProcessingTests(unittest.TestCase): def test_iterates_to_empty_list_initially(self): self.assertEqual(list(self.bg), []) + def test_enqueue_increments_pending_requests(self): + self.bg.enqueue_request(0) + self.assertEqual(self.bg.pending_requests, 1) + |