diff --git a/shared/observability/src/airflow_shared/observability/metrics/statsd_logger.py b/shared/observability/src/airflow_shared/observability/metrics/statsd_logger.py index 3500a04dc7be3..b5f3977724126 100644 --- a/shared/observability/src/airflow_shared/observability/metrics/statsd_logger.py +++ b/shared/observability/src/airflow_shared/observability/metrics/statsd_logger.py @@ -170,7 +170,12 @@ def get_statsd_logger( statsd_influxdb_enabled: bool = False, ) -> SafeStatsdLogger: """Return logger for StatsD.""" - statsd = stats_class(host, port, prefix, ipv6) + statsd = stats_class( + host=host, + port=port, + prefix=prefix, + ipv6=ipv6, + ) metric_tags_validator = PatternBlockListValidator(statsd_disabled_tags) validator = get_validator(metrics_allow_list, metrics_block_list) diff --git a/shared/observability/tests/observability/metrics/test_stats.py b/shared/observability/tests/observability/metrics/test_stats.py index 4121ffe9eee85..04f2731248a35 100644 --- a/shared/observability/tests/observability/metrics/test_stats.py +++ b/shared/observability/tests/observability/metrics/test_stats.py @@ -64,6 +64,13 @@ def get_statsd_logger_factory( ) +class TestGetStatsdLogger: + @mock.patch("statsd.StatsClient") + def test_passes_ipv6_as_keyword(self, mock_stats_client): + statsd_logger.get_statsd_logger(stats_class=mock_stats_client, ipv6=True) + assert mock_stats_client.call_args.kwargs["ipv6"] is True + + class TestStats: def setup_method(self): self.statsd_client = Mock(spec=statsd.StatsClient)