def test_export(self):
        exporter = jaeger_exporter.JaegerExporter(service_name='my_service',
                                                  transport=MockTransport)
        exporter.export({})

        collector = jaeger_exporter.Collector(transport=MockTransport,
                                              http_transport=MockTransport)
        collector.export({})

        agent = jaeger_exporter.AgentClientUDP(transport=MockTransport)
        agent.export({})

        self.assertTrue(exporter.transport.export_called)
        self.assertTrue(collector.transport.export_called)
        self.assertTrue(agent.transport.export_called)
 def test_packet_capacity_exceeded(self, mock_logging):
     agent_client = jaeger_exporter.AgentClientUDP(client=MockClient,
                                                   max_packet_size=-1)
     agent_client.emit({})
     self.assertTrue(mock_logging.warn.called)
    def test_agent_emit_succeeded(self, mock_logging):
        agent_client = jaeger_exporter.AgentClientUDP(client=MockClient)

        agent_client.emit({})
        self.assertTrue(agent_client.client.emit_called)
        self.assertFalse(mock_logging.warn.called)