5 

     5 

     6 from iro.controller.task import createJob, Task

     6 from iro.controller.task import createJob, Task

     7 from iro.controller.pool  import taskPool, dbPool

     7 from iro.controller.pool  import taskPool, dbPool

     8 

     8 

     9 from iro.model.job import exJobs, ExJob, Status

     9 from iro.model.job import exJobs, ExJob, Status

    11 from iro.model.pool import data

    11 from iro.model.pool import data

    12 

    12 

    13 from iro.model.message import SMS, Fax, Mail

    13 from iro.model.message import SMS, Fax, Mail

    14 from iro.model.schema import Job, User

    14 from iro.model.schema import Job, User

    15 from iro.telnumber import Telnumber

    15 from iro.telnumber import Telnumber

    16 

    16 

    17 from iro.validate import vInteger

    19 from iro.validate import vInteger

    18 from iro.error import NoRoute

    20 from iro.error import NoRoute

    19 

    21 

    20 from .dbtestcase import DBTestCase

    22 from .dbtestcase import DBTestCase

    21 

    23 

    92         DBTestCase.setUp(self)

    94         DBTestCase.setUp(self)

    93         dbPool.start(reactor)

    95         dbPool.start(reactor)

    94 

    96 

    95     def tearDown(self):

    97     def tearDown(self):

    96         exJobs.clear()

    98         exJobs.clear()

    97         dbPool.pool.stop()

   100         dbPool.pool.stop()

    98         taskPool.pool.q.queue = deque()

   101         taskPool.pool.q.queue = deque()

    99         DBTestCase.tearDown(self)

   102         DBTestCase.tearDown(self)

   100     

   103     

   101     @inlineCallbacks

   104     @inlineCallbacks

   114     @inlineCallbacks

   117     @inlineCallbacks

   115     def testRun(self):

   118     def testRun(self):

   116         with self.session() as session:

   119         with self.session() as session:

   117             u = User(name='test',apikey='abcdef123456789')

   120             u = User(name='test',apikey='abcdef123456789')

   118             session.add(u)

   121             session.add(u)

   119 

   124 

   120         exjob = yield exJobs.create(u, [Telnumber('123456789')], SMS('test'), ['test'])

   125         exjob = yield exJobs.create(u, [Telnumber('123456789')], SMS('test'), ['test'])

   121         

   126         

   122         task=Task(Telnumber('123456789'), exjob)

   127         task=Task(Telnumber('123456789'), exjob)

   123         ret = yield task._run()

   128         ret = yield task._run()