Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Creating server-safe anonymous clients for testing

Browse Source
This commit is contained in:
Jack Danger 2017-09-16 15:48:20 -07:00
parent e33702fbac
commit c8f6fb7738
1 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests/test_s3/test_s3.py
View File

@ -16,6 +16,7 @@ import boto3
from botocore.client import ClientError from botocore.client import ClientError
import botocore.exceptions import botocore.exceptions
from boto.exception import S3CreateError, S3ResponseError from boto.exception import S3CreateError, S3ResponseError
from botocore.handlers import disable_signing
from boto.s3.connection import S3Connection from boto.s3.connection import S3Connection
from boto.s3.key import Key from boto.s3.key import Key
from freezegun import freeze_time from freezegun import freeze_time
@ -864,36 +865,43 @@ def test_bucket_acl_switching():
g.permission == 'READ' for g in grants), grants g.permission == 'READ' for g in grants), grants
@mock_s3_deprecated
@mock_s3 @mock_s3
def test_s3_object_in_public_bucket(): def test_s3_object_in_public_bucket():
s3 = boto3.resource('s3') s3 = boto3.resource('s3')
bucket = s3.Bucket('test-bucket') bucket = s3.Bucket('test-bucket')
bucket.create(ACL='public-read') bucket.create(ACL='public-read')
bucket.put_object(ACL='public-read', Body=b'ABCD', Key='file.txt') bucket.put_object(ACL='public-read', Body=b'ABCD', Key='file.txt')
direct_url = 'https://test-bucket.s3.amazonaws.com/file.txt'
response = requests.get(direct_url) s3_anonymous = boto3.resource('s3')
response.status_code.should.equal(200) s3_anonymous.meta.client.meta.events.register('choose-signer.s3.*', disable_signing)
response.content.should.equal(b'ABCD')
contents = s3_anonymous.Object(key='file.txt', bucket_name='test-bucket').get()['Body'].read()
contents.should.equal(b'ABCD')
bucket.put_object(ACL='private', Body=b'ABCD', Key='file.txt') bucket.put_object(ACL='private', Body=b'ABCD', Key='file.txt')
response = requests.get(direct_url)
response.status_code.should.equal(403) with assert_raises(ClientError) as exc:
s3_anonymous.Object(key='file.txt', bucket_name='test-bucket').get()
exc.exception.response['Error']['Code'].should.equal('403')
@mock_s3_deprecated
@mock_s3 @mock_s3
def test_s3_object_in_private_bucket(): def test_s3_object_in_private_bucket():
s3 = boto3.resource('s3') s3 = boto3.resource('s3')
bucket = s3.Bucket('test-bucket') bucket = s3.Bucket('test-bucket')
bucket.create(ACL='private') bucket.create(ACL='private')
bucket.put_object(ACL='private', Body=b'ABCD', Key='file.txt') bucket.put_object(ACL='private', Body=b'ABCD', Key='file.txt')
direct_url = 'https://test-bucket.s3.amazonaws.com/file.txt'
response = requests.get(direct_url) s3_anonymous = boto3.resource('s3')
response.status_code.should.equal(403) s3_anonymous.meta.client.meta.events.register('choose-signer.s3.*', disable_signing)
with assert_raises(ClientError) as exc:
s3_anonymous.Object(key='file.txt', bucket_name='test-bucket').get()
exc.exception.response['Error']['Code'].should.equal('403')
bucket.put_object(ACL='public-read', Body=b'ABCD', Key='file.txt') bucket.put_object(ACL='public-read', Body=b'ABCD', Key='file.txt')
response = requests.get(direct_url) contents = s3_anonymous.Object(key='file.txt', bucket_name='test-bucket').get()['Body'].read()
response.status_code.should.equal(200) contents.should.equal(b'ABCD')
response.content.should.equal(b'ABCD')
@mock_s3_deprecated @mock_s3_deprecated