1 files changed, 10 insertions, 13 deletions
diff --git a/bgproc_tests.py b/bgproc_tests.py
index 5915321..3f3634d 100644
--- a/bgproc_tests.py
+++ b/bgproc_tests.py
@@ -19,27 +19,24 @@ import unittest
 import bgproc
 
 
-requests = []
-
 def callback(request):
-    requests.append(request)
-    return len(requests)
+    return request + 1
 
 
 class BackgroundProcessingTests(unittest.TestCase):
 
     def setUp(self):
-        del requests[:]
         self.bg = bgproc.BackgroundProcessing(callback)
+        
+    def tearDown(self):
+        self.bg.close_requests()
+        self.bg.finish()
     
-    def test_wait_for_results_returns_false_initially(self):
-        self.assertEqual(self.bg.wait_for_results(), False)
-
-    def test_iterates_to_empty_list_initially(self):
-        self.assertEqual(list(self.bg), [])
+    def test_get_results_returns_nothing_initially(self):
+        self.assertEqual(self.bg.get_results(), [])
 
     def test_processes_stuff(self):
         self.bg.enqueue_request(0)
-        self.assertEqual(self.bg.wait_for_results(), True)
-        self.assertEqual(list(self.bg), [1])
-        self.assertEqual(list(self.bg), [])
+        self.assertEqual(self.bg.get_results(block=True), [1])
+        self.assertEqual(self.bg.get_results(block=True), [])
+