From ea1b24550db19a119dac423f8aee21f690197ce8 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Wed, 25 May 2011 17:20:39 +0100
Subject: Remove stuff after iterating over it.

---
 bgproc.py       | 3 ++-
 bgproc_tests.py | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/bgproc.py b/bgproc.py
index 949fb63..690880c 100644
--- a/bgproc.py
+++ b/bgproc.py
@@ -53,6 +53,7 @@ class BackgroundProcessing(object):
 
     def __iter__(self):
         '''Iterate over immediately available results.'''
-        for result in self.results:
+        while self.results:
+            result = self.results.pop(0)
             yield result
 
diff --git a/bgproc_tests.py b/bgproc_tests.py
index 885a223..dc6c79e 100644
--- a/bgproc_tests.py
+++ b/bgproc_tests.py
@@ -49,4 +49,5 @@ class BackgroundProcessingTests(unittest.TestCase):
         self.bg.enqueue_request(0)
         self.assertEqual(self.bg.wait_for_results(), True)
         self.assertEqual(list(self.bg), [1])
+        self.assertEqual(list(self.bg), [])
 
-- 
cgit v1.2.1