Tests for presigned url

Author: Alexander Vasin

poetry.lock

@@ -59,6 +59,7 @@ freezegun = "^1.5.0"
 pytest = "^7.4.4"
 setuptools = "^69.5.1"
 pytest-aiohttp = "^1.0.5"
+yarl = "^1.9.4"
 
 [build-system]
 requires = ["poetry-core"]

pyproject.toml

@@ -5,6 +5,8 @@
 from pathlib import Path
 
 import pytest
+from freezegun import freeze_time
+from yarl import URL
 
 from aiohttp_s3_client import S3Client
 
@@ -138,3 +140,64 @@ async def async_iterable(iterable: bytes):
     # FIXME: uncomment after update fakes3 image
     actual = gzip.GzipFile(fileobj=BytesIO(result)).read()
     assert actual == data
+
+
+@pytest.mark.parametrize('method,given_url', [
+    # Simple test
+    ('GET', 'test/object'),
+
+    # Check method name is ok in lowercase
+    ('get', 'test/object'),
+
+    # Absolute path
+    ('GET', '/test/object'),
+
+    # Check URL object with path is given
+    ('get', URL('./test/object')),
+
+    # Check URL object with path is given
+    ('get', URL('/test/object')),
+])
+def test_presign_non_absolute_url(s3_client, s3_url, method, given_url):
+    presigned = s3_client.presign_url('get', 'test/object')
+    assert presigned.is_absolute()
+    assert presigned.scheme == s3_url.scheme
+    assert presigned.host == s3_url.host
+    assert presigned.port == s3_url.port
+    assert presigned.path.startswith(s3_url.path)
+    assert presigned.path.endswith(str(given_url).lstrip('.'))
+
+
+@pytest.mark.parametrize('method,given_url', [
+    # String url
+    ('GET', 'https://absolute-url:123/path'),
+
+    # URL object
+    ('GET', URL('https://absolute-url:123/path')),
+])
+def test_presign_absolute_url(s3_client, method, given_url):
+    presigned = s3_client.presign_url(method, given_url)
+    assert presigned.is_absolute()
+
+    url_object = URL(given_url)
+    assert presigned.scheme == url_object.scheme
+    assert presigned.host == url_object.host
+    assert presigned.port == url_object.port
+    assert presigned.path == url_object.path
+
+
+@freeze_time("2024-01-01")
+def test_presign_url(s3_client, s3_url):
+    url = s3_client.presign_url('get', URL('./example'))
+    assert url.path == (s3_url / 'example').path
+    assert url.query == {
+        'X-Amz-Algorithm': 'AWS4-HMAC-SHA256',
+        'X-Amz-Content-Sha256': 'UNSIGNED-PAYLOAD',
+        'X-Amz-Credential': 'user/20240101/us-east-1/s3/aws4_request',
+        'X-Amz-Date': '20240101T000000Z',
+        'X-Amz-Expires': '86400',
+        'X-Amz-SignedHeaders': 'host',
+        'X-Amz-Signature': (
+            '7359f1286edf554b0eab363e3c93ee32855b8d429d975fdb0a5d2cb7ad5c0db0'
+        )
+    }