"""Test Datadog tags."""

from types import SimpleNamespace
from unittest.mock import Mock

from switchboard_consumer.utils.datadog import setup_datadog_tags


def test_no_current_span():
    """Test no current span."""
    tracer = Mock()
    tracer.current_span.return_value = None
    message = {}
    logger = Mock()

    assert setup_datadog_tags(tracer, message, logger) is None


def test_bad_message():
    """Test bad message format."""
    tracer = Mock()
    message = {}
    logger = Mock()

    assert setup_datadog_tags(tracer, message, logger) is None


def test_valid_message():
    """Test with a valid message."""
    tracer = Mock()
    current_span = Mock()
    tracer.current_span.return_value = current_span
    message = SimpleNamespace(correlation_id='1',
                              entity_type='a',
                              message_type='b')
    logger = Mock()

    result = setup_datadog_tags(tracer, message, logger)

    assert result is None
    assert current_span.set_tag.call_count == 3