"""Test s3_file check."""

from unittest.mock import MagicMock

from pytest import MonkeyPatch

from assets.connectors import s3
from assets.models import s3_file


def test_create_presigned_url(monkeypatch: MonkeyPatch) -> None:
    """Test create_presigned_url."""
    s3_client_mock = MagicMock(autospec=True)
    s3_client_mock.generate_presigned_url.return_value = "URL"

    monkeypatch.setattr(s3, "get_s3_client", MagicMock(return_value=s3_client_mock))

    bucket = "bucket"
    key = "key"

    result = s3_file.create_presigned_url(bucket, key)

    s3_client_mock.generate_presigned_url.assert_called_with(
        "get_object", Params={"Bucket": bucket, "Key": key}, ExpiresIn=3600
    )

    assert result == "URL"