def test_response_attributes(self): OpenTelemetryMiddleware._add_response_attributes( # noqa pylint: disable=protected-access self.span, "404 Not Found") expected = ( mock.call("http.status_code", 404), mock.call("http.status_text", "Not Found"), ) self.assertEqual(self.span.set_attribute.call_count, len(expected)) self.span.set_attribute.assert_has_calls(expected, any_order=True)
def test_response_attributes_invalid_status_code(self): OpenTelemetryMiddleware._add_response_attributes( # noqa pylint: disable=protected-access self.span, "Invalid Status Code") self.assertEqual(self.span.set_attribute.call_count, 1) self.span.set_attribute.assert_called_with("http.status_text", "Status Code")