tests/job.py
branchdevel
changeset 141 90c95fdd1e33
parent 140 75d0eaaa871d
child 142 7d0c1fa97123
--- a/tests/job.py	Thu Feb 09 17:36:31 2012 +0100
+++ b/tests/job.py	Thu Feb 09 18:27:58 2012 +0100
@@ -1,8 +1,9 @@
-import unittest
+from twisted.internet import reactor
+
 from Queue import deque
 
 from iro.controller.task import createJob, Task
-from iro.controller.pool  import taskPool
+from iro.controller.pool  import taskPool, dbPool
 
 from iro.model.job import exJobs, ExJob
 
@@ -14,11 +15,8 @@
 
 from iro.validate import vInteger
 
-
 from .dbtestcase import DBTestCase
 
-#activates all logging we can get.
-
 class DumpPool():
     def run(self, f,*a,**k):
         return f(*a,**k)
@@ -90,13 +88,11 @@
    
     def setUp(self):
         DBTestCase.setUp(self)
-        self.pool = data.pool
-        data.pool = DumpPool()
+        dbPool.start(reactor)
 
     def tearDown(self):
         exJobs.clear()
-        data.pool = self.pool
-        self.pool = None
+        dbPool.pool.stop()
         taskPool.pool.q.queue = deque()
         DBTestCase.tearDown(self)
     
@@ -104,27 +100,36 @@
         with self.session() as session:
             u = User(name='test',apikey='abcdef123456789')
             session.add(u)
-        exjob = createJob(u,[Telnumber('0123325456')],SMS('sms'),[])
-        self.assertEqual(taskPool.pool.q.qsize(),1)
-        self.assertEqual(exjob.tasks.keys(),[Telnumber('0123325456')])
-        self.assertIsInstance(exjob.tasks[Telnumber('0123325456')], Task)
+        
+        d = createJob(u,[Telnumber('0123325456')],SMS('sms'),[])
+        
+        def poolsize(result):
+            self.assertEqual(taskPool.pool.q.qsize(),1)
+            return result
+        d.addCallback(poolsize)
+        def keys(result):
+            self.assertEqual(result.tasks.keys(),[Telnumber('0123325456')])
+            return result
+        d.addCallback(keys)
+        def tasktest(result):
+            self.assertIsInstance(result.tasks[Telnumber('0123325456')], Task)
+            return result
+        d.addCallback(tasktest)
 
 
 
-    def testCreateFax(self):
-        with self.session() as session:
-            u = User(name='test',apikey='abcdef123456789')
-            session.add(u)
-        job = createJob(u,[Telnumber('0123325456')],Fax('header','fax',[]),[])
+        return d
 
-        self.assertEqual(taskPool.pool.q.qsize(),1)
+    #def testCreateFax(self):
+    #    with self.session() as session:
+    #        u = User(name='test',apikey='abcdef123456789')
+    #        session.add(u)
+    #    job = createJob(u,[Telnumber('0123325456')],Fax('header','fax',[]),[])
 
-    def testCreateMail(self):
-        with self.session() as session:
-            u = User(name='test',apikey='abcdef123456789')
-            session.add(u)
-        job = createJob(u,[],Mail('sub','body','t@t.de'),[])
-    
+    #    self.assertEqual(taskPool.pool.q.qsize(),1)
 
-if __name__ == '__main__':
-        unittest.main()
+    #def testCreateMail(self):
+    #    with self.session() as session:
+    #        u = User(name='test',apikey='abcdef123456789')
+    #        session.add(u)
+    #    job = createJob(u,[],Mail('sub','body','t@t.de'),[])