diff --git a/jmclient/jmclient/client_protocol.py b/jmclient/jmclient/client_protocol.py index 8f5f60b..a183d52 100644 --- a/jmclient/jmclient/client_protocol.py +++ b/jmclient/jmclient/client_protocol.py @@ -83,7 +83,8 @@ class JMTakerClientProtocol(amp.AMP): #To avoid creating yet another config variable, we set the timeout #to 20 * maker_timeout_sec. - reactor.callLater(20*maker_timeout_sec, self.stallMonitor, + if not hasattr(self.taker, 'testflag'): #pragma: no cover + reactor.callLater(20*maker_timeout_sec, self.stallMonitor, self.taker.schedule_index+1) d = self.callRemote(commands.JMInit, diff --git a/jmclient/test/test_client_protocol.py b/jmclient/test/test_client_protocol.py index 5606a06..afd013d 100644 --- a/jmclient/test/test_client_protocol.py +++ b/jmclient/test/test_client_protocol.py @@ -226,6 +226,7 @@ class TrialTestJMClientProto(unittest.TestCase): for i, p in enumerate(params): takers[i].set_fail_init(p[0]) takers[i].set_fail_utxos(p[1]) + takers[i].testflag = True if i != 0: clientfactories.append(JMTakerClientProtocolFactory(takers[i])) clientconn = reactor.connectTCP("localhost", 27184, clientfactories[i])