15 |
15 |
16 from iro.offer import Smstrade, SMTP, Offer |
16 from iro.offer import Smstrade, SMTP, Offer |
17 from iro.offer.smstrade import SmstradeException, StatusCode |
17 from iro.offer.smstrade import SmstradeException, StatusCode |
18 |
18 |
19 from ..test_helpers.dbtestcase import DBTestCase |
19 from ..test_helpers.dbtestcase import DBTestCase |
20 |
20 from ..test_helpers.utils import DummyPool |
21 class DummyPool(): |
|
22 def run(self, f,*a,**k): |
|
23 return f(*a,**k) |
|
24 |
21 |
25 def run( f,*a,**k): |
22 def run( f,*a,**k): |
26 return f(*a,**k) |
23 return f(*a,**k) |
27 |
|
28 from twisted.python import log |
|
29 |
|
30 class DummyObserver(object): |
|
31 def __init__(self): |
|
32 self.e=[] |
|
33 |
|
34 def start(self): |
|
35 log.addObserver(self.emit) |
|
36 |
|
37 def stop(self): |
|
38 log.removeObserver(self.emit) |
|
39 |
|
40 def emit(self, eventDict): |
|
41 self.e.append(eventDict) |
|
42 |
|
43 |
|
44 |
24 |
45 class IntegratedOfferTests(DBTestCase): |
25 class IntegratedOfferTests(DBTestCase): |
46 def setUp(self): |
26 def setUp(self): |
47 DBTestCase.setUp(self) |
27 DBTestCase.setUp(self) |
48 self.pool = data.pool |
28 self.pool = data.pool |
49 data.pool = DummyPool() |
29 data.pool = DummyPool() |
50 |
30 |
51 self.taskPool = taskPool.run |
31 self.taskPool = taskPool.run |
52 taskPool.run = run |
32 taskPool.run = run |
53 |
33 |
54 self.log = DummyObserver() |
|
55 self.log.start() |
|
56 |
|
57 def tearDown(self): |
34 def tearDown(self): |
58 self.log.stop() |
|
59 exJobs.clear() |
35 exJobs.clear() |
60 data.pool = self.pool |
36 data.pool = self.pool |
61 self.pool = None |
37 self.pool = None |
62 taskPool.run = self.taskPool |
38 taskPool.run = self.taskPool |
63 self.taskPool = None |
39 self.taskPool = None |