def test_constructor(self): patch_worker = mock.patch( 'opencensus.common.transports.async_._Worker', autospec=True) exporter = mock.Mock() with patch_worker: transport = async_.AsyncTransport(exporter) self.assertTrue(transport.worker.start.called) self.assertEqual(transport.exporter, exporter)
def test_flush(self): patch_worker = mock.patch( 'opencensus.common.transports.async_._Worker', autospec=True) exporter = mock.Mock() with patch_worker: transport = async_.AsyncTransport(exporter) transport.flush() self.assertTrue(transport.worker.flush.called)
def test_export(self): patch_worker = mock.patch( 'opencensus.common.transports.async_._Worker', autospec=True) exporter = mock.Mock() with patch_worker: transport = async_.AsyncTransport(exporter) trace = { 'traceId': 'test', 'spans': [{}, {}], } transport.export(trace) self.assertTrue(transport.worker.enqueue.called)