def test_output_with_exception( self, c ): options = MagicMock( in_test_mode=False ) device = 'device' port = 'port' data = {Constants.DataPacket.device: device, Constants.DataPacket.port: port} cs = COSMSend( options ) cs.createDataStream = Mock() cs.createJSONReport = Mock() cs.report_data = Mock( side_effect=Exception( 'Test' ) ) cs.output( data ) cs.createDataStream.called_once_with( device, port, data ) cs.createJSONReport.called_once_with( device, port, data ) cs.report_data.called_once_with( device, port, data )
def test_output_with_exception(self, c): options = MagicMock(in_test_mode=False) device = 'device' port = 'port' data = { Constants.DataPacket.device: device, Constants.DataPacket.port: port } cs = COSMSend(options) cs.createDataStream = Mock() cs.createJSONReport = Mock() cs.report_data = Mock(side_effect=Exception('Test')) cs.output(data) cs.createDataStream.called_once_with(device, port, data) cs.createJSONReport.called_once_with(device, port, data) cs.report_data.called_once_with(device, port, data)
def test_createFeed_with_two_datestreams(self, config): options = None cs = COSMSend(options) device = 'device 1' port = 'port 1' config.assert_called_once_with() cs.config = self.config_data data = { 'device': device, 'port': port, Constants.DataPacket.action: Constants.DataPacket.accumulate, Constants.DataPacket.arrival_time: datetime.datetime(2012, 1, 2, 3, 4, 5), Constants.DataPacket.current_value: 10 } cs.createDataStream(device, port, data) data[Constants.DataPacket.current_value] = 545454 cs.createDataStream(device, port, data) data['device'] = device = 'device 2' data[Constants.DataPacket.current_value] = 999 cs.createDataStream(device, port, data) data['device'] = device = 'device 1' data[Constants.DataPacket.action] = Constants.DataPacket.send cs.report_data = MagicMock() cs.output(data) pprint.pprint(cs.json) pprint.pprint(cs.datapoints) # self.assertEqual( cs.[Constants.Cosm.title], Constants.Cosm.title ) # self.assertEqual( cs.json[Constants.Cosm.status], Constants.Cosm.status ) # self.assertEqual( cs.json[Constants.Cosm.creator], Constants.Cosm.creator ) # self.assertEqual( cs.json[Constants.Cosm.created], Constants.Cosm.created ) # self.assertEqual( cs.json[Constants.Cosm.feed], 'url' ) # self.assertEqual( cs.json[Constants.Cosm.email], Constants.Cosm.email ) # self.assertEqual( cs.json[Constants.Cosm.id], Constants.Cosm.id ) # self.assertEqual( cs.json[Constants.Cosm.auto_feed_url], ( 'url', ) ) # self.assertEqual( cs.json[Constants.Cosm.version], Constants.Cosm.version ) cs = None
def test_createFeed_with_two_datestreams( self, config ): options = None cs = COSMSend( options ) device = 'device 1' port = 'port 1' config.assert_called_once_with() cs.config = self.config_data data = {'device': device, 'port': port, Constants.DataPacket.action: Constants.DataPacket.accumulate, Constants.DataPacket.arrival_time: datetime.datetime( 2012, 1, 2, 3, 4, 5 ), Constants.DataPacket.current_value: 10} cs.createDataStream( device, port, data ) data[Constants.DataPacket.current_value] = 545454 cs.createDataStream( device, port, data ) data['device'] = device = 'device 2' data[Constants.DataPacket.current_value] = 999 cs.createDataStream( device, port, data ) data['device'] = device = 'device 1' data[Constants.DataPacket.action] = Constants.DataPacket.send cs.report_data = MagicMock() cs.output( data ) pprint.pprint( cs.json ) pprint.pprint( cs.datapoints ) # self.assertEqual( cs.[Constants.Cosm.title], Constants.Cosm.title ) # self.assertEqual( cs.json[Constants.Cosm.status], Constants.Cosm.status ) # self.assertEqual( cs.json[Constants.Cosm.creator], Constants.Cosm.creator ) # self.assertEqual( cs.json[Constants.Cosm.created], Constants.Cosm.created ) # self.assertEqual( cs.json[Constants.Cosm.feed], 'url' ) # self.assertEqual( cs.json[Constants.Cosm.email], Constants.Cosm.email ) # self.assertEqual( cs.json[Constants.Cosm.id], Constants.Cosm.id ) # self.assertEqual( cs.json[Constants.Cosm.auto_feed_url], ( 'url', ) ) # self.assertEqual( cs.json[Constants.Cosm.version], Constants.Cosm.version ) cs = None