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
     )
Beispiel #3
0
    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)