122 lines
3.2 KiB
Python
122 lines
3.2 KiB
Python
from __future__ import unicode_literals
|
|
import os
|
|
from sure import expect
|
|
from moto.s3.utils import (
|
|
bucket_name_from_url,
|
|
_VersionedKeyStore,
|
|
parse_region_from_url,
|
|
clean_key_name,
|
|
undo_clean_key_name,
|
|
)
|
|
from parameterized import parameterized
|
|
|
|
|
|
def test_base_url():
|
|
expect(bucket_name_from_url("https://s3.amazonaws.com/")).should.equal(None)
|
|
|
|
|
|
def test_localhost_bucket():
|
|
expect(bucket_name_from_url("https://wfoobar.localhost:5000/abc")).should.equal(
|
|
"wfoobar"
|
|
)
|
|
|
|
|
|
def test_localhost_without_bucket():
|
|
expect(bucket_name_from_url("https://www.localhost:5000/def")).should.equal(None)
|
|
|
|
|
|
def test_force_ignore_subdomain_for_bucketnames():
|
|
os.environ["S3_IGNORE_SUBDOMAIN_BUCKETNAME"] = "1"
|
|
expect(
|
|
bucket_name_from_url("https://subdomain.localhost:5000/abc/resource")
|
|
).should.equal(None)
|
|
del os.environ["S3_IGNORE_SUBDOMAIN_BUCKETNAME"]
|
|
|
|
|
|
def test_versioned_key_store():
|
|
d = _VersionedKeyStore()
|
|
|
|
d.should.have.length_of(0)
|
|
|
|
d["key"] = [1]
|
|
|
|
d.should.have.length_of(1)
|
|
|
|
d["key"] = 2
|
|
d.should.have.length_of(1)
|
|
|
|
d.should.have.key("key").being.equal(2)
|
|
|
|
d.get.when.called_with("key").should.return_value(2)
|
|
d.get.when.called_with("badkey").should.return_value(None)
|
|
d.get.when.called_with("badkey", "HELLO").should.return_value("HELLO")
|
|
|
|
# Tests key[
|
|
d.shouldnt.have.key("badkey")
|
|
d.__getitem__.when.called_with("badkey").should.throw(KeyError)
|
|
|
|
d.getlist("key").should.have.length_of(2)
|
|
d.getlist("key").should.be.equal([[1], 2])
|
|
d.getlist("badkey").should.be.none
|
|
|
|
d.setlist("key", 1)
|
|
d.getlist("key").should.be.equal([1])
|
|
|
|
d.setlist("key", (1, 2))
|
|
d.getlist("key").shouldnt.be.equal((1, 2))
|
|
d.getlist("key").should.be.equal([1, 2])
|
|
|
|
d.setlist("key", [[1], [2]])
|
|
d["key"].should.have.length_of(1)
|
|
d.getlist("key").should.be.equal([[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",
|
|
]:
|
|
parse_region_from_url(url).should.equal(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",
|
|
]:
|
|
parse_region_from_url(url).should.equal(expected)
|
|
|
|
|
|
@parameterized(
|
|
[
|
|
("foo/bar/baz", "foo/bar/baz"),
|
|
("foo", "foo"),
|
|
(
|
|
"foo/run_dt%3D2019-01-01%252012%253A30%253A00",
|
|
"foo/run_dt=2019-01-01%2012%3A30%3A00",
|
|
),
|
|
]
|
|
)
|
|
def test_clean_key_name(key, expected):
|
|
clean_key_name(key).should.equal(expected)
|
|
|
|
|
|
@parameterized(
|
|
[
|
|
("foo/bar/baz", "foo/bar/baz"),
|
|
("foo", "foo"),
|
|
(
|
|
"foo/run_dt%3D2019-01-01%252012%253A30%253A00",
|
|
"foo/run_dt%253D2019-01-01%25252012%25253A30%25253A00",
|
|
),
|
|
]
|
|
)
|
|
def test_undo_clean_key_name(key, expected):
|
|
undo_clean_key_name(key).should.equal(expected)
|