def test_buffers_runtime_info(self): """Make sure we send the buffers size info.""" stats_worker = StatsWorker(self.service, 10) # add some info the all the buffers we know mock = mocker.Mocker() producer = mock.mock() upload.MultipartUploadFactory.buffers_size = 0 mp = upload.MultipartUploadFactory(None, producer, 1024, 2048, 0, 1042, 10) mp.dataReceived('a' * 10) client = mock.mock() http_client.HTTPProducer.buffers_size = 0 http_prod = http_client.HTTPProducer(client) self.addCleanup(http_prod.timeout_call.cancel) http_prod.dataReceived('a' * 20) producers.S3Producer.buffers_size = 0 s3_prod = producers.S3Producer(1024) s3_prod.dataReceived('a' * 30) stats_worker.runtime_info() self.assertIn(('gauge', 'buffers_size.HTTPProducer', 20), self.metrics.calls) self.assertIn(('gauge', 'buffers_size.MultipartUploadFactory', 10), self.metrics.calls) self.assertIn(('gauge', 'buffers_size.S3Producer', 30), self.metrics.calls)
def test_runtime_info(self): """Make sure we add runtime info.""" stats_worker = StatsWorker(self.service, 10) # get the reactor from twisted.internet import reactor stats_worker.runtime_info() # check the reactor data self.assertIn(('gauge', 'reactor.readers', len(reactor.getReaders())), self.metrics.calls) self.assertIn(('gauge', 'reactor.writers', len(reactor.getWriters())), self.metrics.calls)