def test_success(self): add_TraceServiceServicer_to_server( TraceServiceServicerSUCCESS(), self.server ) self.assertEqual( self.exporter.export([self.span]), SpanExportResult.SUCCESS )
def test_failure(self): add_TraceServiceServicer_to_server( TraceServiceServicerALREADY_EXISTS(), self.server ) self.assertEqual( self.exporter.export([self.span]), SpanExportResult.FAILURE )
def test_unavailable_delay(self, mock_sleep, mock_expo): mock_expo.configure_mock(**{"return_value": [1]}) add_TraceServiceServicer_to_server( TraceServiceServicerUNAVAILABLEDelay(), self.server) self.assertEqual(self.exporter.export([self.span]), SpanExportResult.FAILURE) mock_sleep.assert_called_with(4)