Spaces:
Build error
Build error
Validify-testbot-1
/
botbuilder-python
/libraries
/botbuilder-integration-applicationinsights-aiohttp
/tests
/test_aiohttp_processor.py
from unittest.mock import Mock | |
from aiounittest import AsyncTestCase | |
import aiohttp # pylint: disable=unused-import | |
from botbuilder.integration.applicationinsights.aiohttp import ( | |
aiohttp_telemetry_middleware, | |
AiohttpTelemetryProcessor, | |
) | |
class TestAiohttpTelemetryProcessor(AsyncTestCase): | |
# pylint: disable=protected-access | |
def test_can_process(self): | |
assert AiohttpTelemetryProcessor.detect_aiohttp() | |
assert AiohttpTelemetryProcessor().can_process() | |
def test_retrieve_aiohttp_body(self): | |
aiohttp_telemetry_middleware._REQUEST_BODIES = Mock() | |
aiohttp_telemetry_middleware._REQUEST_BODIES.pop = Mock( | |
return_value="test body" | |
) | |
assert aiohttp_telemetry_middleware.retrieve_aiohttp_body() == "test body" | |
assert AiohttpTelemetryProcessor().get_request_body() == "test body" | |
aiohttp_telemetry_middleware._REQUEST_BODIES = {} | |