"""Test datadog connector."""

from unittest.mock import patch

from collaborator.connectors import datadog


@patch("collaborator.connectors.datadog.api")
@patch("collaborator.connectors.datadog.initialize")
def test_datadog_connection(mock_datadog_initialize, mock_datadog_api):
    """Test datadog_connection context manager."""
    with datadog.datadog_connection(app_key="1", api_key="2") as conn:
        conn.Event.create(title="title", text="text", tags=["tags"])

        mock_datadog_initialize.assert_called_with(app_key="1", api_key="2")
        mock_datadog_api.Event.create.assert_called_with(
            title="title", text="text", tags=["tags"]
        )


@patch("collaborator.connectors.datadog.api")
@patch("collaborator.connectors.datadog.initialize")
def test_call_datadog_with_event(mock_datadog_initialize, mock_datadog_api):
    """Test call datadog with event function."""
    datadog.call_datadog_with_event(title="title", text="text", tags=["tags"])

    mock_datadog_initialize.assert_called_with(
        app_key="DATADOG_APP_KEY", api_key="DATADOG_API_KEY"
    )
    mock_datadog_api.Event.create.assert_called_with(
        title="title", text="text", tags=["tags"]
    )