"""Tests for the Sentry Connector."""

import os
from unittest.mock import patch

import auditlogger

from collaborator.connectors import audit_logger


def test_logger_decorator():
    """Test that the decorator adds a logger."""

    @audit_logger.logger
    def test_func(logger):
        assert isinstance(logger, auditlogger.logger.AuditLogger)

    test_func()


def test_logger_decorator_logger_present():
    """Test that the decorator doesn't add a logger."""

    @audit_logger.logger
    def test_func(logger):
        assert not isinstance(logger, auditlogger.logger.AuditLogger)
        assert logger == "fakelogger"

    test_func(logger="fakelogger")


def test_logger_decorator_logger_present_no_kwargs():
    """Test that the decorator doesn't add a logger."""
    test_host = "test_host"
    test_user = "test_user"
    test_pass = "test_pass"
    os.environ["AUDIT_LOG_ES"] = test_host
    os.environ["AUDIT_LOG_USER"] = test_user
    os.environ["AUDIT_LOG_PASS"] = test_pass
    mock_audit_logger_hosts = [
        {
            "host": test_host,
            "port": 443,
            "username": test_user,
            "password": test_pass,
        }
    ]

    with patch("collaborator.config.AUDIT_LOGGER_HOSTS", new=mock_audit_logger_hosts):

        @audit_logger.logger
        def log_event(logger=None):
            assert logger.hosts == mock_audit_logger_hosts