~tribaal/txaws/xss-hardening

« back to all changes in this revision

Viewing changes to txaws/s3/tests/test_client.py

  • Committer: Drew Smathers
  • Date: 2012-05-16 02:44:47 UTC
  • mfrom: (133.5.1 921419-uploadpart)
  • Revision ID: drew.smathers@gmail.com-20120516024447-dq47lkfsjyvrow3c
Merged 921419-uploadpart [r=oubiwann][f=921419]

This branch adds Upload Part method to s3 client.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1022
1022
            amz_headers={"acl": "public"})
1023
1023
        return deferred.addCallback(check_result)
1024
1024
 
 
1025
    def test_upload_part(self):
 
1026
 
 
1027
        class StubQuery(client.Query):
 
1028
 
 
1029
            def __init__(query, action, creds, endpoint, bucket=None,
 
1030
                         object_name=None, data="", body_producer=None,
 
1031
                         content_type=None, receiver_factory=None, metadata={}):
 
1032
                super(StubQuery, query).__init__(action=action, creds=creds,
 
1033
                                                 bucket=bucket,
 
1034
                                                 object_name=object_name,
 
1035
                                                 data=data)
 
1036
                self.assertEquals(action, "PUT")
 
1037
                self.assertEqual(creds.access_key, "foo")
 
1038
                self.assertEqual(creds.secret_key, "bar")
 
1039
                self.assertEqual(query.bucket, "example-bucket")
 
1040
                self.assertEqual(query.object_name,
 
1041
                    "example-object?partNumber=3&uploadId=testid")
 
1042
                self.assertEqual(query.data, "some data")
 
1043
                self.assertEqual(query.metadata, {})
 
1044
 
 
1045
            def submit(query, url_context=None):
 
1046
                return succeed(None)
 
1047
 
 
1048
        creds = AWSCredentials("foo", "bar")
 
1049
        s3 = client.S3Client(creds, query_factory=StubQuery)
 
1050
        return s3.upload_part("example-bucket", "example-object", "testid", 3,
 
1051
                              "some data")
 
1052
 
1025
1053
S3ClientTestCase.skip = s3clientSkip
1026
1054
 
1027
1055