add tests

Author: lesnik512

tests/test_faststream_bootstrap.py

@@ -1,6 +1,7 @@
 import pytest
 import structlog
 from faststream.redis import RedisBroker, TestRedisBroker
+from faststream.redis.opentelemetry import RedisTelemetryMiddleware
 from faststream.redis.prometheus import RedisPrometheusMiddleware
 from opentelemetry.sdk.trace.export import ConsoleSpanExporter
 from starlette import status
@@ -20,6 +21,7 @@ def broker() -> RedisBroker:
 
 async def test_faststream_bootstrap(broker: RedisBroker) -> None:
     prometheus_metrics_path = "/test-metrics-path"
+    health_check_path = "/custom-health-check-path"
     bootstrapper = FastStreamBootstrapper(
         bootstrap_config=FastStreamConfig(
             broker=broker,
@@ -30,16 +32,47 @@ async def test_faststream_bootstrap(broker: RedisBroker) -> None:
             opentelemetry_endpoint="otl",
             opentelemetry_instrumentors=[CustomInstrumentor()],
             opentelemetry_span_exporter=ConsoleSpanExporter(),
+            opentelemetry_middleware_cls=RedisTelemetryMiddleware,
             prometheus_metrics_path=prometheus_metrics_path,
             prometheus_middleware_cls=RedisPrometheusMiddleware,
             sentry_dsn="https://testdsn@localhost/1",
-            health_checks_path="/health/",
+            health_checks_path=health_check_path,
             logging_buffer_capacity=0,
         ),
     )
     application = bootstrapper.bootstrap()
     logger.info("testing logging", key="value")
+    test_client = TestClient(app=application)
 
     async with TestRedisBroker(broker):
-        response = TestClient(app=application).get(prometheus_metrics_path)
+        response = test_client.get(prometheus_metrics_path)
         assert response.status_code == status.HTTP_200_OK
+
+        response = test_client.get(health_check_path)
+        assert response.status_code == status.HTTP_200_OK
+        assert response.json() == {"health_status": True, "service_name": "microservice", "service_version": "2.0.0"}
+
+
+async def test_faststream_bootstrap_health_check_wo_broker() -> None:
+    health_check_path = "/custom-health-check-path"
+    bootstrapper = FastStreamBootstrapper(
+        bootstrap_config=FastStreamConfig(
+            service_name="microservice",
+            service_version="2.0.0",
+            service_environment="test",
+            service_debug=False,
+            opentelemetry_endpoint="otl",
+            opentelemetry_instrumentors=[CustomInstrumentor()],
+            opentelemetry_span_exporter=ConsoleSpanExporter(),
+            sentry_dsn="https://testdsn@localhost/1",
+            health_checks_path=health_check_path,
+            logging_buffer_capacity=0,
+        ),
+    )
+    application = bootstrapper.bootstrap()
+    logger.info("testing logging", key="value")
+    test_client = TestClient(app=application)
+
+    response = test_client.get(health_check_path)
+    assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
+    assert response.text == "Service is unhealthy"