def test_collector_auth_headers(self): collector = trace_exporter.Collector(http_transport=MockTransport, auth=('user', 'pass')) self.assertTrue(collector.http_transport.headers_set) collector = trace_exporter.Collector(http_transport=MockTransport) self.assertFalse(collector.http_transport.headers_set)
def test_collector_emit_succeeded(self, mock_logging): url = 'http://localhost:14268/api/traces?format=jaeger.thrift' collector = trace_exporter.Collector( thrift_url=url, http_transport=MockTransport, client=MockClient) collector.http_transport.is_open = True collector.http_transport.code = 200 collector.http_transport.message = 'success' collector.emit(MockBatch()) self.assertFalse(mock_logging.error.called) self.assertTrue(collector.http_transport.is_closed)
def test_export(self): exporter = trace_exporter.JaegerExporter(service_name='my_service', transport=MockTransport) exporter.export({}) collector = trace_exporter.Collector(transport=MockTransport, http_transport=MockTransport) collector.export({}) agent = trace_exporter.AgentClientUDP(transport=MockTransport) agent.export({}) self.assertTrue(exporter.transport.export_called) self.assertTrue(collector.transport.export_called) self.assertTrue(agent.transport.export_called)