from unittest.mock import MagicMock, patch

from configs import dify_config
from controllers.web import site as site_module
from extensions.storage.storage_type import StorageType
from models.model import IconType, Site


def test_build_site_icon_url_uses_s3_presigned_url() -> None:
    site = MagicMock(spec=Site)
    site.icon_type = IconType.IMAGE
    site.icon = "11111111-1111-4111-8111-111111111111"

    with (
        patch.object(dify_config, "EDITION", "CLOUD"),
        patch.object(dify_config, "STORAGE_TYPE", StorageType.S3),
        patch.object(site_module, "db") as mock_db,
        patch.object(site_module, "FileService") as mock_file_service,
        patch.object(site_module, "build_icon_url") as mock_build_icon_url,
    ):
        mock_file_service.return_value.get_file_presigned_url.return_value = (
            "https://s3.example.com/icon.png?signature=test"
        )

        result = site_module._build_site_icon_url(site=site, tenant_id="tenant-id")

    assert result == "https://s3.example.com/icon.png?signature=test"
    mock_file_service.assert_called_once_with(mock_db.engine)
    mock_file_service.return_value.get_file_presigned_url.assert_called_once_with(
        file_id="11111111-1111-4111-8111-111111111111",
        tenant_id="tenant-id",
    )
    mock_build_icon_url.assert_not_called()


def test_build_site_icon_url_keeps_preview_url_for_self_hosted_s3() -> None:
    site = MagicMock(spec=Site)
    site.icon_type = IconType.IMAGE
    site.icon = "11111111-1111-4111-8111-111111111111"

    with (
        patch.object(dify_config, "EDITION", "SELF_HOSTED"),
        patch.object(dify_config, "STORAGE_TYPE", StorageType.S3),
        patch.object(site_module, "FileService") as mock_file_service,
        patch.object(site_module, "build_icon_url", return_value="https://api.example.com/files/icon/file-preview"),
    ):
        result = site_module._build_site_icon_url(site=site, tenant_id="tenant-id")

    assert result == "https://api.example.com/files/icon/file-preview"
    mock_file_service.assert_not_called()


def test_build_site_icon_url_keeps_preview_url_for_non_s3_storage() -> None:
    site = MagicMock(spec=Site)
    site.icon_type = IconType.IMAGE
    site.icon = "11111111-1111-4111-8111-111111111111"

    with (
        patch.object(dify_config, "EDITION", "CLOUD"),
        patch.object(dify_config, "STORAGE_TYPE", StorageType.LOCAL),
        patch.object(site_module, "FileService") as mock_file_service,
        patch.object(site_module, "build_icon_url", return_value="https://api.example.com/files/icon/file-preview"),
    ):
        result = site_module._build_site_icon_url(site=site, tenant_id="tenant-id")

    assert result == "https://api.example.com/files/icon/file-preview"
    mock_file_service.assert_not_called()