def test_capture_data_alt_callback(self):
def cb(*args, **kwargs):
return {'test': 'test'}
self.setupFakeStorage([capture.CaptureData('test', 'builder1', cb)])
self.setupBuild()
self.master.db.builds.finishBuild(buildid=1, results=0)
build_data = yield self.stats_service.master.data.get(('builds', 1))
msg = {
'data_name': 'test',
'post_data': {
'test': 'test'
},
'build_data': build_data
}
routingKey = ("stats-yieldMetricsValue", "stats-yield-data")
self.master.mq.callConsumer(routingKey, msg)
self.assertEqual([({
'test': 'test'
}, 'builder1-test', {
'build_number': '1',
'builder_name': 'builder1'
})], self.fake_storage_service.stored_data)
def test_capture_data_error(self):
def cb(*args, **kwargs):
raise TypeError
self.setupFakeStorage([capture.CaptureData('test', 'builder1', cb)])
self.setupBuild()
self.master.db.builds.finishBuild(buildid=1, results=0)
build_data = yield self.stats_service.master.data.get(('builds', 1))
msg = {
'data_name': 'test',
'post_data': {'test': 'test'},
'build_data': build_data
}
routingKey = ("stats-yieldMetricsValue", "stats-yield-data")
cap = self.fake_storage_service.captures[0]
self.assertFailure(cap.consume(routingKey, msg), CaptureCallbackError)
