96 exJobs.clear() |
96 exJobs.clear() |
97 dbPool.pool.stop() |
97 dbPool.pool.stop() |
98 taskPool.pool.q.queue = deque() |
98 taskPool.pool.q.queue = deque() |
99 DBTestCase.tearDown(self) |
99 DBTestCase.tearDown(self) |
100 |
100 |
|
101 @inlineCallbacks |
101 def testCreateSMS(self): |
102 def testCreateSMS(self): |
102 with self.session() as session: |
103 with self.session() as session: |
103 u = User(name='test',apikey='abcdef123456789') |
104 u = User(name='test',apikey='abcdef123456789') |
104 session.add(u) |
105 session.add(u) |
105 |
106 |
106 d = createJob(u,[Telnumber('0123325456')],SMS('sms'),[]) |
107 job = yield createJob(u,[Telnumber('0123325456')],SMS('sms'),[]) |
107 |
108 |
108 def poolsize(result): |
109 self.assertEqual(taskPool.pool.q.qsize(),1) |
109 self.assertEqual(taskPool.pool.q.qsize(),1) |
110 |
110 return result |
111 self.assertEqual(job.tasks.keys(),[Telnumber('0123325456')]) |
111 d.addCallback(poolsize) |
112 self.assertIsInstance(job.tasks[Telnumber('0123325456')], Task) |
112 |
113 |
113 def keys(result): |
|
114 self.assertEqual(result.tasks.keys(),[Telnumber('0123325456')]) |
|
115 return result |
|
116 d.addCallback(keys) |
|
117 return d |
|
118 @inlineCallbacks |
114 @inlineCallbacks |
119 def testRun(self): |
115 def testRun(self): |
120 with self.session() as session: |
116 with self.session() as session: |
121 u = User(name='test',apikey='abcdef123456789') |
117 u = User(name='test',apikey='abcdef123456789') |
122 session.add(u) |
118 session.add(u) |
123 |
|
124 def tasktest(result): |
|
125 self.assertIsInstance(result.tasks[Telnumber('0123325456')], Task) |
|
126 return result |
|
127 d.addCallback(tasktest) |
|
128 |
|
129 |
|
130 |
119 |
131 exjob = yield exJobs.create(u, [Telnumber('123456789')], SMS('test'), ['test']) |
120 exjob = yield exJobs.create(u, [Telnumber('123456789')], SMS('test'), ['test']) |
132 |
121 |
133 task=Task(Telnumber('123456789'), exjob) |
122 task=Task(Telnumber('123456789'), exjob) |
134 ret = yield task._run() |
123 ret = yield task._run() |