def test_onGotItem_noParams(self):
cdm = CorrelatedDefaultsManager(testjid)
cdm.broadcastStart = Mock()
cdm.broadcastStop = Mock()
cdm.broadcastLogs = Mock()
item = Element((pubsub.NS_PUBSUB_EVENT, 'item'))
item['id'] = str(1)
item.addElement('stuff', content='stuff that isnt parameters')
out = yield cdm.onGotItem(item)
self.assertFalse(cdm.broadcastStart.called)
self.assertFalse(cdm.broadcastStop.called)
self.assertFalse(cdm.broadcastLogs.called)
def test_broadcastStart(self):
cdm = CorrelatedDefaultsManager(testjid)
cdm.outputNode.onOutput = Mock(side_effect=utils.good_side_effect('lush'))
params = simulation.Parameters()
port = portfolio.Portfolio('p1')
out = yield cdm.broadcastStart(params, port)
self.assertEquals(cdm.outputNode.onOutput.call_count, 1)
expected = simulation.Parameters(cmd='start', timestamp=params.timestamp).toElement()
expected.addChild(portfolio.Portfolio('p1').toElement())
for (a, dic) in cdm.outputNode.onOutput.call_args_list:
self.assertEquals(expected.toXml(), dic['data'].toXml())
def test_onGotItem_startNoRunId(self):
cdm = CorrelatedDefaultsManager(testjid)
cdm.broadcastStart = Mock()
cdm.broadcastStop = Mock()
cdm.broadcastLogs = Mock(side_effect=utils.good_side_effect('lush'))
item = Element((pubsub.NS_PUBSUB_EVENT, 'item'))
params = simulation.Parameters(cmd='start')
p = item.addChild(params.toElement())
p.addChild(portfolio.Portfolio('p1').toElement())
expected = simulation.Parameters(cmd='start', run_id=None, timestamp=params.timestamp).toElement()
out = yield cdm.onGotItem(item)
self.assertFalse(cdm.broadcastStart.called)
self.assertEquals(cdm.broadcastLogs.call_count, 1)
self.assertFalse(cdm.broadcastStop.called)
for ((a,b), dic) in cdm.broadcastLogs.call_args_list:
self.assertEquals(b.toElement().toXml(), expected.toXml())
def test_onGotItem_stopping(self):
cdm = CorrelatedDefaultsManager(testjid)
cdm.broadcastStart = Mock()
cdm.broadcastStop = Mock(side_effect=utils.good_side_effect('lush'))
cdm.broadcastLogs = Mock()
item = Element((pubsub.NS_PUBSUB_EVENT, 'item'))
item['id'] = str(1)
params = simulation.Parameters(cmd='stop')
expected = params.toElement()
item.addChild(expected)
out = yield cdm.onGotItem(item)
self.assertFalse(cdm.broadcastStart.called)
self.assertEquals(cdm.broadcastStop.call_count, 1)
self.assertFalse(cdm.broadcastLogs.called)
for ((a,), dic) in cdm.broadcastStop.call_args_list:
self.assertEquals(a.toElement().toXml(), expected.toXml())
