1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
from unittest.mock import patch
import pytest
from moto.s3.utils import (
_VersionedKeyStore,
bucket_name_from_url,
compute_checksum,
cors_matches_origin,
parse_region_from_url,
)
def test_base_url():
assert bucket_name_from_url("https://s3.amazonaws.com/") is None
def test_localhost_bucket():
assert bucket_name_from_url("https://wfoobar.localhost:5000/abc") == "wfoobar"
def test_localhost_without_bucket():
assert bucket_name_from_url("https://www.localhost:5000/def") is None
def test_force_ignore_subdomain_for_bucketnames():
with patch("moto.s3.utils.S3_IGNORE_SUBDOMAIN_BUCKETNAME", True):
assert (
bucket_name_from_url("https://subdomain.localhost:5000/abc/resource")
is None
)
def test_versioned_key_store():
key_store = _VersionedKeyStore()
assert not key_store
key_store["key"] = [1]
assert len(key_store) == 1
key_store["key"] = 2
assert len(key_store) == 1
assert key_store["key"] == 2
assert key_store.get("key") == 2
assert key_store.get("badkey") is None
assert key_store.get("badkey", "HELLO") == "HELLO"
# Tests key[
assert "badkey" not in key_store
with pytest.raises(KeyError):
_ = key_store["badkey"]
assert len(key_store.getlist("key")) == 2
assert key_store.getlist("key") == [[1], 2]
assert key_store.getlist("badkey") is None
key_store.setlist("key", 1)
assert key_store.getlist("key") == [1]
key_store.setlist("key", (1, 2))
assert key_store.getlist("key") != (1, 2)
assert key_store.getlist("key") == [1, 2]
key_store.setlist("key", [[1], [2]])
assert len(key_store["key"]) == 1
assert key_store.getlist("key") == [[1], [2]]
def test_parse_region_from_url():
expected = "us-west-2"
for url in [
"http://s3-us-west-2.amazonaws.com/bucket",
"http://s3.us-west-2.amazonaws.com/bucket",
"http://bucket.s3-us-west-2.amazonaws.com",
"https://s3-us-west-2.amazonaws.com/bucket",
"https://s3.us-west-2.amazonaws.com/bucket",
"https://bucket.s3-us-west-2.amazonaws.com",
]:
assert parse_region_from_url(url) == expected
expected = "us-east-1"
for url in [
"http://s3.amazonaws.com/bucket",
"http://bucket.s3.amazonaws.com",
"https://s3.amazonaws.com/bucket",
"https://bucket.s3.amazonaws.com",
]:
assert parse_region_from_url(url) == expected
def test_checksum_sha256():
checksum = b"h9FJy0JMA4dlbyEdJYn7Wx4WIpkhMJ6YWIQZzMqKc2I="
assert compute_checksum(b"somedata", "SHA256") == checksum
# Unknown algorithms fallback to SHA256 for now
assert compute_checksum(b"somedata", algorithm="unknown") == checksum
def test_checksum_sha1():
assert compute_checksum(b"somedata", "SHA1") == b"76oxGuRIpzdMEiBhv+2VLZQOnjc="
def test_checksum_crc32():
assert compute_checksum(b"somedata", "CRC32") == b"Uwy90A=="
def test_checksum_crc32c():
try:
assert compute_checksum(b"somedata", "CRC32C") == b"dB9qBQ=="
except: # noqa: E722 Do not use bare except
# Optional library Can't be found - just revert to CRC32
assert compute_checksum(b"somedata", "CRC32C") == b"Uwy90A=="
def test_cors_utils():
"Fancy string matching"
assert cors_matches_origin("a", ["a"])
assert cors_matches_origin("b", ["a", "b"])
assert not cors_matches_origin("c", [])
assert not cors_matches_origin("c", ["a", "b"])
assert cors_matches_origin("http://www.google.com", ["http://*.google.com"])
assert cors_matches_origin("http://www.google.com", ["http://www.*.com"])
assert cors_matches_origin("http://www.google.com", ["http://*"])
assert cors_matches_origin("http://www.google.com", ["*"])
assert not cors_matches_origin("http://www.google.com", ["http://www.*.org"])
assert not cors_matches_origin("http://www.google.com", ["https://*"])
|
