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)