def test_resolved(aws_ip_range_data, test_ip_range_cache, test_aws_ip, app):
ipresolver = IPResolver(app)
ipresolver.amazon_ranges = test_ip_range_cache["all_amazon"]
ipresolver.sync_token = test_ip_range_cache["sync_token"]
assert ipresolver.resolve_ip(test_aws_ip) == ResolvedLocation(
provider="aws",
service=None,
sync_token=123456789,
country_iso_code=None)
assert ipresolver.resolve_ip("10.0.0.2") == ResolvedLocation(
provider="aws",
service=None,
sync_token=123456789,
country_iso_code=None)
assert ipresolver.resolve_ip("6.0.0.2") == ResolvedLocation(
provider="aws",
service=None,
sync_token=123456789,
country_iso_code=u"US")
assert ipresolver.resolve_ip("1.2.3.4") == ResolvedLocation(
provider="internet",
service=u"US",
sync_token=123456789,
country_iso_code=u"US")
assert ipresolver.resolve_ip("127.0.0.1") == ResolvedLocation(
provider="internet",
service=None,
sync_token=123456789,
country_iso_code=None)
def test_resolved(aws_ip_range_data, test_ip_range_cache, test_aws_ip, app):
ipresolver = IPResolver(app)
ipresolver.amazon_ranges = test_ip_range_cache['all_amazon']
ipresolver.sync_token = test_ip_range_cache['sync_token']
assert ipresolver.resolve_ip(test_aws_ip) == ResolvedLocation(
provider='aws',
service=None,
sync_token=123456789,
country_iso_code=None)
assert ipresolver.resolve_ip('10.0.0.2') == ResolvedLocation(
provider='aws',
service=None,
sync_token=123456789,
country_iso_code=None)
assert ipresolver.resolve_ip('6.0.0.2') == ResolvedLocation(
provider='aws',
service=None,
sync_token=123456789,
country_iso_code=u'US')
assert ipresolver.resolve_ip('1.2.3.4') == ResolvedLocation(
provider='internet',
service=u'US',
sync_token=123456789,
country_iso_code=u'US')
assert ipresolver.resolve_ip('127.0.0.1') == ResolvedLocation(
provider='internet',
service=None,
sync_token=123456789,
country_iso_code=None)
def test_direct_download(
test_aws_ip,
test_empty_ip_range_cache,
test_ip_range_cache,
aws_ip_range_data,
ipranges_populated,
app,
):
ipresolver = IPResolver(app)
if ipranges_populated:
ipresolver.sync_token = (
test_ip_range_cache["sync_token"]
if ipranges_populated
else test_empty_ip_range_cache["sync_token"]
)
ipresolver.amazon_ranges = (
test_ip_range_cache["all_amazon"]
if ipranges_populated
else test_empty_ip_range_cache["all_amazon"]
)
context = StorageContext("nyc", None, config_provider, ipresolver)
# Create a test bucket and put some test content.
boto3.client("s3").create_bucket(Bucket=_TEST_BUCKET)
engine = CloudFrontedS3Storage(
context,
"cloudfrontdomain",
"keyid",
"test/data/test.pem",
"some/path",
_TEST_BUCKET,
_TEST_USER,
_TEST_PASSWORD,
)
engine.put_content(_TEST_PATH, _TEST_CONTENT)
assert engine.exists(_TEST_PATH)
# Request a direct download URL for a request from a known AWS IP, and ensure we are returned an S3 URL.
assert "s3.amazonaws.com" in engine.get_direct_download_url(_TEST_PATH, test_aws_ip)
if ipranges_populated:
# Request a direct download URL for a request from a non-AWS IP, and ensure we are returned a CloudFront URL.
assert "cloudfrontdomain" in engine.get_direct_download_url(_TEST_PATH, "1.2.3.4")
else:
# Request a direct download URL for a request from a non-AWS IP, but since IP Ranges isn't populated, we still
# get back an S3 URL.
assert "s3.amazonaws.com" in engine.get_direct_download_url(_TEST_PATH, "1.2.3.4")
def test_direct_download(test_aws_ip, test_empty_ip_range_cache,
test_ip_range_cache, aws_ip_range_data,
ipranges_populated, app):
ipresolver = IPResolver(app)
if ipranges_populated:
ipresolver.sync_token = test_ip_range_cache[
'sync_token'] if ipranges_populated else test_empty_ip_range_cache[
'sync_token']
ipresolver.amazon_ranges = test_ip_range_cache[
'all_amazon'] if ipranges_populated else test_empty_ip_range_cache[
'all_amazon']
context = StorageContext('nyc', None, None, config_provider,
ipresolver)
# Create a test bucket and put some test content.
boto.connect_s3().create_bucket(_TEST_BUCKET)
engine = CloudFrontedS3Storage(context, 'cloudfrontdomain', 'keyid',
'test/data/test.pem', 'some/path',
_TEST_BUCKET, _TEST_USER,
_TEST_PASSWORD)
engine.put_content(_TEST_PATH, _TEST_CONTENT)
assert engine.exists(_TEST_PATH)
# Request a direct download URL for a request from a known AWS IP, and ensure we are returned an S3 URL.
assert 's3.amazonaws.com' in engine.get_direct_download_url(
_TEST_PATH, test_aws_ip)
if ipranges_populated:
# Request a direct download URL for a request from a non-AWS IP, and ensure we are returned a CloudFront URL.
assert 'cloudfrontdomain' in engine.get_direct_download_url(
_TEST_PATH, '1.2.3.4')
else:
# Request a direct download URL for a request from a non-AWS IP, but since IP Ranges isn't populated, we still
# get back an S3 URL.
assert 's3.amazonaws.com' in engine.get_direct_download_url(
_TEST_PATH, '1.2.3.4')