/WebServiceDemo/Required_Libs/HTTPRequest/Classes/Tests/ASIS3RequestTests.m
Objective C | 860 lines | 615 code | 154 blank | 91 comment | 24 complexity | e74171308016dc8ccd3f4adfe2caf1ec MD5 | raw file
- //
- // ASIS3RequestTests.m
- // Part of ASIHTTPRequest -> http://allseeing-i.com/ASIHTTPRequest
- //
- // Created by Ben Copsey on 12/07/2009.
- // Copyright 2009 All-Seeing Interactive. All rights reserved.
- //
- #import "ASIS3RequestTests.h"
- #import "ASINetworkQueue.h"
- #import "ASIS3BucketObject.h"
- #import "ASIS3ObjectRequest.h"
- #import "ASIS3BucketRequest.h"
- #import "ASIS3ServiceRequest.h"
- // Fill in these to run the tests that actually connect and manipulate objects on S3
- static NSString *secretAccessKey = @"";
- static NSString *accessKey = @"";
- // You should run these tests on a bucket that does not yet exist
- static NSString *bucket = @"";
- // Used for subclass test
- @interface ASIS3ObjectRequestSubclass : ASIS3ObjectRequest {}
- @end
- @implementation ASIS3ObjectRequestSubclass;
- @end
- @interface ASIS3BucketRequestSubclass : ASIS3BucketRequest {}
- @end
- @implementation ASIS3BucketRequestSubclass;
- @end
- @interface ASIS3BucketObjectSubclass : ASIS3BucketObject {}
- @end
- @implementation ASIS3BucketObjectSubclass;
- @end
- // Stop clang complaining about undeclared selectors
- @interface ASIS3RequestTests ()
- - (void)GETRequestDone:(ASIHTTPRequest *)request;
- - (void)GETRequestFailed:(ASIHTTPRequest *)request;
- - (void)PUTRequestDone:(ASIHTTPRequest *)request;
- - (void)PUTRequestFailed:(ASIHTTPRequest *)request;
- - (void)DELETERequestDone:(ASIHTTPRequest *)request;
- - (void)DELETERequestFailed:(ASIHTTPRequest *)request;
- @end
- @implementation ASIS3RequestTests
- // All these tests are based on Amazon's examples at: http://docs.amazonwebservices.com/AmazonS3/2006-03-01/
- - (void)testAuthenticationHeaderGeneration
- {
- NSString *exampleSecretAccessKey = @"uV3F3YluFJax1cknvbcGwgjvx4QpvB+leU8dUj2o";
- NSString *exampleAccessKey = @"0PN5J17HBGZHT7JJ3X82";
- NSString *exampleBucket = @"johnsmith";
-
- // Test list all my buckets
- NSString *dateString = @"Wed, 28 Mar 2007 01:29:59 +0000";
- ASIS3ServiceRequest *serviceRequest = [ASIS3ServiceRequest serviceRequest];
- [serviceRequest setSecretAccessKey:exampleSecretAccessKey];
- [serviceRequest setAccessKey:exampleAccessKey];
- [serviceRequest setDateString:dateString];
- [serviceRequest buildRequestHeaders];
- BOOL success = [[[serviceRequest requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:Db+gepJSUbZKwpx1FR0DLtEYoZA="];
- GHAssertTrue(success,@"Failed to generate the correct authorisation header for a GET service request");
-
- // Test GET
- NSString *key = @"photos/puppy.jpg";
- dateString = @"Tue, 27 Mar 2007 19:36:42 +0000";
- ASIS3ObjectRequest *request = [ASIS3ObjectRequest requestWithBucket:exampleBucket key:key];
- [request setDateString:dateString];
- [request setSecretAccessKey:exampleSecretAccessKey];
- [request setAccessKey:exampleAccessKey];
- [request buildRequestHeaders];
- success = [[[request requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:xXjDGYUmKxnwqr5KXNPGldn5LbA="];
- GHAssertTrue(success,@"Failed to generate the correct authorisation header for a GET request");
-
- // Test PUT
- key = @"photos/puppy.jpg";
- dateString = @"Tue, 27 Mar 2007 21:15:45 +0000";
- request = [ASIS3ObjectRequest requestWithBucket:exampleBucket key:key];
- [request setRequestMethod:@"PUT"];
- [request setMimeType:@"image/jpeg"];
- [request setDateString:dateString];
- [request setSecretAccessKey:exampleSecretAccessKey];
- [request setAccessKey:exampleAccessKey];
- [request buildRequestHeaders];
- success = [[[request requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:hcicpDDvL9SsO6AkvxqmIWkmOuQ="];
- GHAssertTrue(success,@"Failed to generate the correct authorisation header for a PUT request");
-
- // Test List
- dateString = @"Tue, 27 Mar 2007 19:42:41 +0000";
- ASIS3BucketRequest *listRequest = [ASIS3BucketRequest requestWithBucket:exampleBucket];
- [listRequest setPrefix:@"photos"];
- [listRequest setMaxResultCount:50];
- [listRequest setMarker:@"puppy"];
- [listRequest setDateString:dateString];
- [listRequest setSecretAccessKey:exampleSecretAccessKey];
- [listRequest setAccessKey:exampleAccessKey];
- [listRequest buildRequestHeaders];
- success = [[[listRequest requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:jsRt/rhG+Vtp88HrYL706QhE4w4="];
- GHAssertTrue(success,@"Failed to generate the correct authorisation header for a list request");
-
- // Test fetch ACL
- dateString = @"Tue, 27 Mar 2007 19:44:46 +0000";
- listRequest = [ASIS3BucketRequest requestWithBucket:exampleBucket subResource:@"acl"];
- [listRequest setDateString:dateString];
- [listRequest setSecretAccessKey:exampleSecretAccessKey];
- [listRequest setAccessKey:exampleAccessKey];
- [listRequest buildRequestHeaders];
- success = [[[listRequest requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:thdUi9VAkzhkniLj96JIrOPGi0g="];
- GHAssertTrue(success,@"Failed to generate the correct authorisation header for a list request");
-
- // Test Unicode keys
- // Comment out this test for now, as the S3 example is relying on mixed-case hex-encoded characters in the url, which isn't going to be easy to replicate
- // exampleBucket = @"dictionary";
- // key = @"français/préfère";
- // dateString = @"Wed, 28 Mar 2007 01:49:49 +0000";
- // request = [ASIS3ObjectRequest requestWithBucket:exampleBucket key:key];
- // [request setDateString:dateString];
- // [request setSecretAccessKey:exampleSecretAccessKey];
- // [request setAccessKey:exampleAccessKey];
- // [request buildRequestHeaders];
- // success = [[[request requestHeaders] valueForKey:@"Authorization"] isEqualToString:@"AWS 0PN5J17HBGZHT7JJ3X82:dxhSBHoI6eVSPcXJqEghlUzZMnY="];
- //GHAssertTrue(success,@"Failed to generate the correct authorisation header for a list request");
- }
- - (void)testFailure
- {
- // Needs expanding to cover more failure states - this is just a test to ensure Amazon's error description is being added to the error
-
- // We're actually going to try with the Amazon example details, but the request will fail because the date is old
- NSString *exampleSecretAccessKey = @"uV3F3YluFJax1cknvbcGwgjvx4QpvB+leU8dUj2o";
- NSString *exampleAccessKey = @"0PN5J17HBGZHT7JJ3X82";
- NSString *exampleBucket = @"johnsmith";
- NSString *key = @"photos/puppy.jpg";
- NSString *dateString = @"Tue, 27 Mar 2007 19:36:42 +0000";
- ASIS3Request *request = [ASIS3ObjectRequest requestWithBucket:exampleBucket key:key];
- [request setDateString:dateString];
- [request setSecretAccessKey:exampleSecretAccessKey];
- [request setAccessKey:exampleAccessKey];
- [request startSynchronous];
- GHAssertNotNil([request error],@"Failed to generate an error when the request was not correctly signed");
-
- BOOL success = ([[request error] code] == ASIS3ResponseErrorType);
- GHAssertTrue(success,@"Generated error had the wrong error code");
-
- success = ([[[request error] localizedDescription] isEqualToString:@"The difference between the request time and the current time is too large."]);
- GHAssertTrue(success,@"Generated error had the wrong description");
-
- // Ensure a bucket request will correctly parse an error from S3
- request = [ASIS3BucketRequest requestWithBucket:exampleBucket];
- [request setDateString:dateString];
- [request setSecretAccessKey:exampleSecretAccessKey];
- [request setAccessKey:exampleAccessKey];
- [request startSynchronous];
- GHAssertNotNil([request error],@"Failed to generate an error when the request was not correctly signed");
-
- success = ([[request error] code] == ASIS3ResponseErrorType);
- GHAssertTrue(success,@"Generated error had the wrong error code");
-
- success = ([[[request error] localizedDescription] isEqualToString:@"The difference between the request time and the current time is too large."]);
- GHAssertTrue(success,@"Generated error had the wrong description");
-
- // Ensure a service request will correctly parse an error from S3
- request = [ASIS3ServiceRequest serviceRequest];
- [request setDateString:dateString];
- [request setSecretAccessKey:exampleSecretAccessKey];
- [request setAccessKey:exampleAccessKey];
- [request startSynchronous];
- GHAssertNotNil([request error],@"Failed to generate an error when the request was not correctly signed");
-
- success = ([[request error] code] == ASIS3ResponseErrorType);
- GHAssertTrue(success,@"Generated error had the wrong error code");
-
- success = ([[[request error] localizedDescription] isEqualToString:@"The difference between the request time and the current time is too large."]);
- GHAssertTrue(success,@"Generated error had the wrong description");
- }
- - (void)createTestBucket
- {
- // Test creating a bucket
- ASIS3BucketRequest *bucketRequest = [ASIS3BucketRequest PUTRequestWithBucket:bucket];
- [bucketRequest setSecretAccessKey:secretAccessKey];
- [bucketRequest setAccessKey:accessKey];
- [bucketRequest startSynchronous];
- GHAssertNil([bucketRequest error],@"Failed to create a bucket");
- }
- // To run this test, uncomment and fill in your S3 access details
- - (void)testREST
- {
- [self createTestBucket];
-
- BOOL success = (![secretAccessKey isEqualToString:@""] && ![accessKey isEqualToString:@""] && ![bucket isEqualToString:@""]);
- GHAssertTrue(success,@"You need to supply your S3 access details to run the REST test (see the top of ASIS3RequestTests.m)");
-
- // Test creating a bucket
- ASIS3BucketRequest *bucketRequest = [ASIS3BucketRequest PUTRequestWithBucket:bucket];
- [bucketRequest setSecretAccessKey:secretAccessKey];
- [bucketRequest setAccessKey:accessKey];
- [bucketRequest startSynchronous];
- GHAssertNil([bucketRequest error],@"Failed to create a bucket");
-
- // List buckets to make sure the bucket is there
- ASIS3ServiceRequest *serviceRequest = [ASIS3ServiceRequest serviceRequest];
- [serviceRequest setSecretAccessKey:secretAccessKey];
- [serviceRequest setAccessKey:accessKey];
- [serviceRequest startSynchronous];
- GHAssertNil([serviceRequest error],@"Failed to fetch the list of buckets from S3");
-
- BOOL foundBucket = NO;
- for (ASIS3Bucket *theBucket in [serviceRequest buckets]) {
- if ([[theBucket name] isEqualToString:bucket]) {
- foundBucket = YES;
- break;
- }
- }
- GHAssertTrue(foundBucket,@"Failed to retrive the newly-created bucket in a list of buckets");
-
- NSString *key = @"test";
-
- // Create the file
- NSString *text = @"This is my content";
- NSString *filePath = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:@"testfile.txt"];
- [[text dataUsingEncoding:NSUTF8StringEncoding] writeToFile:filePath atomically:NO];
-
- // PUT the file
- ASIS3ObjectRequest *request = [ASIS3ObjectRequest PUTRequestForFile:filePath withBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request setStorageClass:ASIS3StorageClassReducedRedundancy];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to PUT a file to S3");
- // GET the file
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@"This is my content"];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
- // Test fetch subresource
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key subResource:@"acl"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = ([[request responseString] rangeOfString:@"<AccessControlPolicy"].location != NSNotFound);
- GHAssertTrue(success,@"Failed to GET a subresource");
-
- // COPY the file
- request = [ASIS3ObjectRequest COPYRequestFromBucket:bucket key:key toBucket:bucket key:@"test-copy"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to COPY a file");
-
- // GET the copy
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:@"test-copy"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@"This is my content"];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
-
- // HEAD the copy
- request = [ASIS3ObjectRequest HEADRequestWithBucket:bucket key:@"test-copy"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Got a response body for a HEAD request");
-
- // Test listing the objects in this bucket
- ASIS3BucketRequest *listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest startSynchronous];
- GHAssertNil([listRequest error],@"Failed to download a list from S3");
- success = [[listRequest objects] count];
- GHAssertTrue(success,@"The file didn't show up in the list");
- // Test again with a prefix query
- listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setPrefix:@"test"];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest startSynchronous];
- GHAssertNil([listRequest error],@"Failed to download a list from S3");
- success = [[listRequest objects] count];
- GHAssertTrue(success,@"The file didn't show up in the list");
-
- // DELETE the file
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setRequestMethod:@"DELETE"];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to DELETE the file from S3");
-
- // (Also DELETE the copy we made)
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:@"test-copy"];
- [request setSecretAccessKey:secretAccessKey];
- [request setRequestMethod:@"DELETE"];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to DELETE the copy from S3");
-
- // Attempt to COPY the file, even though it is no longer there
- request = [ASIS3ObjectRequest COPYRequestFromBucket:bucket key:key toBucket:bucket key:@"test-copy"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- GHAssertNotNil([request error],@"Failed generate an error for what should have been a failed COPY");
-
- success = [[[request error] localizedDescription] isEqualToString:@"The specified key does not exist."];
- GHAssertTrue(success, @"Got the wrong error message");
-
- // PUT some data
- NSData *data = [@"Hello" dataUsingEncoding:NSUTF8StringEncoding];
- request = [ASIS3ObjectRequest PUTRequestForData:data withBucket:bucket key:key];
- [request setMimeType:@"text/plain"];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to PUT data to S3");
-
- // GET the data to check it uploaded properly
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@"Hello"];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
-
- // clean up (Delete it)
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setRequestMethod:@"DELETE"];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to DELETE the file from S3");
-
- // Delete the bucket
- bucketRequest = [ASIS3BucketRequest DELETERequestWithBucket:bucket];
- [bucketRequest setSecretAccessKey:secretAccessKey];
- [bucketRequest setAccessKey:accessKey];
- [bucketRequest startSynchronous];
- GHAssertNil([bucketRequest error],@"Failed to delete a bucket");
-
-
- }
- // Will upload a file to S3, gzipping it before uploading
- // The file will be stored deflate, and automatically inflated when downloaded
- // This means the file will take up less storage space, and will upload and download faster
- // The file should still be accessible by any HTTP client that supports gzipped responses (eg browsers, NSURLConnection, etc)
- - (void)testGZippedContent
- {
- [self createTestBucket];
-
- BOOL success = (![secretAccessKey isEqualToString:@""] && ![accessKey isEqualToString:@""] && ![bucket isEqualToString:@""]);
- GHAssertTrue(success,@"You need to supply your S3 access details to run the gzipped put test (see the top of ASIS3RequestTests.m)");
-
- // Create the file
- NSString *text = @"This is my content This is my content This is my content This is my content This is my content This is my content";
- NSString *filePath = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:@"testfile.txt"];
- [[text dataUsingEncoding:NSUTF8StringEncoding] writeToFile:filePath atomically:NO];
-
- NSString *key = @"gzipped-data";
- ASIS3ObjectRequest *request = [ASIS3ObjectRequest PUTRequestForFile:filePath withBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request setShouldCompressRequestBody:YES];
- [request setAccessPolicy:ASIS3AccessPolicyPublicRead]; // We'll make it public
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to PUT the gzipped file");
-
- // GET the file
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:text];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
-
- success = [[[request responseHeaders] valueForKey:@"Content-Encoding"] isEqualToString:@"gzip"];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
-
- // Now grab the data using something other than ASIHTTPRequest to ensure other HTTP clients can parse the gzipped content
- NSData *data = [NSURLConnection sendSynchronousRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:[NSString stringWithFormat:@"http://%@.s3.amazonaws.com/gzipped-data",bucket]]] returningResponse:NULL error:NULL];
- NSString *string = [[[NSString alloc] initWithBytes:[data bytes] length:[data length] encoding:NSUTF8StringEncoding] autorelease];
- success = [string isEqualToString:text];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
-
- // Cleanup
- request = [ASIS3ObjectRequest DELETERequestWithBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
-
- }
- - (void)testListRequest
- {
- [self createTestBucket];
-
- BOOL success = (![secretAccessKey isEqualToString:@""] && ![accessKey isEqualToString:@""] && ![bucket isEqualToString:@""]);
- GHAssertTrue(success,@"You need to supply your S3 access details to run the list test (see the top of ASIS3RequestTests.m)");
-
- // Firstly, create and upload 5 files
- int i;
- for (i=0; i<5; i++) {
- NSString *text = [NSString stringWithFormat:@"This is the content of file #%hi",i];
- NSString *filePath = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:[NSString stringWithFormat:@"%hi.txt",i]];
- [[text dataUsingEncoding:NSUTF8StringEncoding] writeToFile:filePath atomically:NO];
- NSString *key = [NSString stringWithFormat:@"test-file/%hi",i];
- ASIS3ObjectRequest *request = [ASIS3ObjectRequest PUTRequestForFile:filePath withBucket:bucket key:key];
- [request setSecretAccessKey:secretAccessKey];
- [request setAccessKey:accessKey];
- [request startSynchronous];
- GHAssertNil([request error],@"Give up on list request test - failed to upload a file");
- }
-
- // Test common prefixes
- ASIS3BucketRequest *listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest setDelimiter:@"/"];
- [listRequest startSynchronous];
- GHAssertNil([listRequest error],@"Failed to download a list from S3");
- success = NO;
- for (NSString *prefix in [listRequest commonPrefixes]) {
- if ([prefix isEqualToString:@"test-file/"]) {
- success = YES;
- }
- }
- GHAssertTrue(success,@"Failed to obtain a list of common prefixes");
-
-
- // Test truncation
- listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest setMaxResultCount:1];
- [listRequest startSynchronous];
- GHAssertTrue([listRequest isTruncated],@"Failed to identify what should be a truncated list of results");
-
- // Test urls are built correctly when requesting a subresource
- listRequest = [ASIS3BucketRequest requestWithBucket:bucket subResource:@"acl"];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest setDelimiter:@"/"];
- [listRequest setPrefix:@"foo"];
- [listRequest setMarker:@"bar"];
- [listRequest setMaxResultCount:5];
- [listRequest buildURL];
- NSString *expectedURL = [NSString stringWithFormat:@"http://%@.s3.amazonaws.com/?acl&prefix=foo&marker=bar&delimiter=/&max-keys=5",bucket];
- success = ([[[listRequest url] absoluteString] isEqualToString:expectedURL]);
- GHAssertTrue(success,@"Generated the wrong url when requesting a subresource");
-
-
- // Now get a list of the files
- listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setPrefix:@"test-file"];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest startSynchronous];
- GHAssertNil([listRequest error],@"Failed to download a list from S3");
- success = ([[listRequest objects] count] == 5);
- GHAssertTrue(success,@"List did not contain all files");
-
- // Please don't use an autoreleased operation queue with waitUntilAllOperationsAreFinished in your own code unless you're writing a test like this one
- // (The end result is no better than using synchronous requests) thx - Ben :)
- ASINetworkQueue *queue = [[[ASINetworkQueue alloc] init] autorelease];
-
- // Test fetching all the items
- [queue setRequestDidFinishSelector:@selector(GETRequestDone:)];
- [queue setRequestDidFailSelector:@selector(GETRequestFailed:)];
- [queue setDelegate:self];
- for (ASIS3BucketObject *object in [listRequest objects]) {
- ASIS3ObjectRequest *request = [object GETRequest];
- [request setAccessKey:accessKey];
- [request setSecretAccessKey:secretAccessKey];
- [queue addOperation:request];
- }
- [queue go];
- [queue waitUntilAllOperationsAreFinished];
-
-
- // Test uploading new files for all the items
- [queue setRequestDidFinishSelector:@selector(PUTRequestDone:)];
- [queue setRequestDidFailSelector:@selector(PUTRequestFailed:)];
- [queue setDelegate:self];
- i=0;
- // For each one, we'll just upload the same content again
- for (ASIS3BucketObject *object in [listRequest objects]) {
- NSString *oldFilePath = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:[NSString stringWithFormat:@"%hi.txt",i]];;
- ASIS3Request *request = [object PUTRequestWithFile:oldFilePath];
- [request setAccessKey:accessKey];
- [request setSecretAccessKey:secretAccessKey];
- [queue addOperation:request];
- i++;
- }
- [queue go];
- [queue waitUntilAllOperationsAreFinished];
-
-
- // Test deleting all the items
- [queue setRequestDidFinishSelector:@selector(DELETERequestDone:)];
- [queue setRequestDidFailSelector:@selector(DELETERequestFailed:)];
- [queue setDelegate:self];
- i=0;
- for (ASIS3BucketObject *object in [listRequest objects]) {
- ASIS3ObjectRequest *request = [object DELETERequest];
- [request setAccessKey:accessKey];
- [request setSecretAccessKey:secretAccessKey];
- [queue addOperation:request];
- i++;
- }
- [queue go];
- [queue waitUntilAllOperationsAreFinished];
-
- // Grab the list again, it should be empty now
- listRequest = [ASIS3BucketRequest requestWithBucket:bucket];
- [listRequest setPrefix:@"test-file"];
- [listRequest setSecretAccessKey:secretAccessKey];
- [listRequest setAccessKey:accessKey];
- [listRequest startSynchronous];
- GHAssertNil([listRequest error],@"Failed to download a list from S3");
- success = ([[listRequest objects] count] == 0);
- GHAssertTrue(success,@"List contained files that should have been deleted");
-
- }
- - (void)GETRequestDone:(ASIS3Request *)request
- {
- NSString *expectedContent = [NSString stringWithFormat:@"This is the content of file #%@",[[[request url] absoluteString] lastPathComponent]];
- BOOL success = ([[request responseString] isEqualToString:expectedContent]);
- GHAssertTrue(success,@"Got the wrong content when downloading one of the files");
-
- }
- - (void)GETRequestFailed:(ASIS3Request *)request
- {
- GHAssertTrue(NO,@"GET request failed for one of the items in the list");
- }
- - (void)PUTRequestDone:(ASIS3Request *)request
- {
- }
- - (void)PUTRequestFailed:(ASIS3Request *)request
- {
- GHAssertTrue(NO,@"PUT request failed for one of the items in the list");
- }
- - (void)DELETERequestDone:(ASIS3Request *)request
- {
- }
- - (void)DELETERequestFailed:(ASIS3Request *)request
- {
- GHAssertTrue(NO,@"DELETE request failed for one of the items in the list");
- }
- // Ensure class convenience constructors return an instance of our subclass
- - (void)testSubclasses
- {
- ASIS3ObjectRequestSubclass *instance = [ASIS3ObjectRequestSubclass requestWithBucket:@"bucket" key:@"key"];
- BOOL success = [instance isKindOfClass:[ASIS3ObjectRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
- instance = [ASIS3ObjectRequestSubclass PUTRequestForFile:@"/file" withBucket:@"bucket" key:@"key"];
- success = [instance isKindOfClass:[ASIS3ObjectRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
-
- instance = [ASIS3ObjectRequestSubclass DELETERequestWithBucket:@"bucket" key:@"key"];
- success = [instance isKindOfClass:[ASIS3ObjectRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
-
- instance = [ASIS3ObjectRequestSubclass COPYRequestFromBucket:@"bucket" key:@"key" toBucket:@"bucket" key:@"key2"];
- success = [instance isKindOfClass:[ASIS3ObjectRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
-
- instance = [ASIS3ObjectRequestSubclass HEADRequestWithBucket:@"bucket" key:@"key"];
- success = [instance isKindOfClass:[ASIS3ObjectRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
-
- ASIS3BucketRequestSubclass *instance2 = [ASIS3BucketRequestSubclass requestWithBucket:@"bucket"];
- success = [instance2 isKindOfClass:[ASIS3BucketRequestSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
-
- ASIS3BucketObjectSubclass *instance3 = [ASIS3BucketObjectSubclass objectWithBucket:@"bucket"];
- success = [instance3 isKindOfClass:[ASIS3BucketObjectSubclass class]];
- GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
- }
- - (void)s3RequestFailed:(ASIHTTPRequest *)request
- {
- GHFail(@"Request failed - cannot continue with test");
- [[self networkQueue] cancelAllOperations];
- }
- - (void)s3QueueFinished:(ASINetworkQueue *)queue
- {
- // BOOL success = (progress == 1.0);
- // GHAssertTrue(success,@"Failed to update progress properly");
- }
- - (void)testQueueProgress
- {
- ASIHTTPRequest *request = [ASIHTTPRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com/i/logo.png"]];
- [request startSynchronous];
- NSData *data = [request responseData];
-
- [self createTestBucket];
- // Upload objects
- progress = 0;
- [[self networkQueue] cancelAllOperations];
- [self setNetworkQueue:[ASINetworkQueue queue]];
- [[self networkQueue] setDelegate:self];
- [[self networkQueue] setRequestDidFailSelector:@selector(s3RequestFailed:)];
- [[self networkQueue] setQueueDidFinishSelector:@selector(s3QueueFinished:)];
- [[self networkQueue] setUploadProgressDelegate:self];
- [[self networkQueue] setShowAccurateProgress:YES];
- [[self networkQueue] setMaxConcurrentOperationCount:1];
-
- int i;
- for (i=0; i<5; i++) {
-
- NSString *key = [NSString stringWithFormat:@"stuff/file%hi.txt",i+1];
-
- ASIS3ObjectRequest *s3Request = [ASIS3ObjectRequest PUTRequestForData:data withBucket:bucket key:key];
- [s3Request setSecretAccessKey:secretAccessKey];
- [s3Request setAccessKey:accessKey];
- [s3Request setTimeOutSeconds:20];
- [s3Request setNumberOfTimesToRetryOnTimeout:3];
- [s3Request setMimeType:@"image/png"];
- [[self networkQueue] addOperation:s3Request];
- }
- [[self networkQueue] go];
- [[self networkQueue] waitUntilAllOperationsAreFinished];
-
- // Download objects
- progress = 0;
- [[self networkQueue] cancelAllOperations];
- [self setNetworkQueue:[ASINetworkQueue queue]];
- [[self networkQueue] setDelegate:self];
- [[self networkQueue] setRequestDidFailSelector:@selector(s3RequestFailed:)];
- [[self networkQueue] setQueueDidFinishSelector:@selector(s3QueueFinished:)];
- [[self networkQueue] setDownloadProgressDelegate:self];
- [[self networkQueue] setShowAccurateProgress:YES];
-
- for (i=0; i<5; i++) {
-
- NSString *key = [NSString stringWithFormat:@"stuff/file%hi.txt",i+1];
-
- ASIS3ObjectRequest *s3Request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [s3Request setSecretAccessKey:secretAccessKey];
- [s3Request setAccessKey:accessKey];
- NSString *downloadPath = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:[NSString stringWithFormat:@"%hi.jpg",i+1]];
- [s3Request setDownloadDestinationPath:downloadPath];
- [[self networkQueue] addOperation:s3Request];
- }
-
- [[self networkQueue] go];
- [[self networkQueue] waitUntilAllOperationsAreFinished];
- progress = 0;
-
- // Delete objects
- progress = 0;
-
- [[self networkQueue] cancelAllOperations];
- [self setNetworkQueue:[ASINetworkQueue queue]];
- [[self networkQueue] setDelegate:self];
- [[self networkQueue] setRequestDidFailSelector:@selector(s3RequestFailed:)];
- [[self networkQueue] setShowAccurateProgress:YES];
-
- for (i=0; i<5; i++) {
-
- NSString *key = [NSString stringWithFormat:@"stuff/file%hi.txt",i+1];
-
- ASIS3ObjectRequest *s3Request = [ASIS3ObjectRequest DELETERequestWithBucket:bucket key:key];
- [s3Request setSecretAccessKey:secretAccessKey];
- [s3Request setAccessKey:accessKey];
- [[self networkQueue] addOperation:s3Request];
- }
- [[self networkQueue] go];
- [[self networkQueue] waitUntilAllOperationsAreFinished];
- }
-
- // Will be called on Mac OS
- - (void)setDoubleValue:(double)newProgress;
- {
- progress = (float)newProgress;
- }
- - (void)setProgress:(float)newProgress;
- {
- progress = newProgress;
- }
- - (void)testCopy
- {
- NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
- ASIS3ObjectRequest *request = [ASIS3ObjectRequest requestWithBucket:@"foo" key:@"eep"];
- ASIS3ObjectRequest *request2 = [request copy];
- GHAssertNotNil(request2,@"Failed to create a copy");
-
- [pool release];
-
- BOOL success = ([request2 retainCount] > 0);
- GHAssertTrue(success,@"Failed to create a retained copy");
- success = ([request2 isKindOfClass:[ASIS3Request class]]);
- GHAssertTrue(success,@"Copy is of wrong class");
-
- [request2 release];
-
- pool = [[NSAutoreleasePool alloc] init];
-
- ASIS3BucketRequest *request3 = [ASIS3BucketRequest requestWithBucket:@"foo"];
- ASIS3BucketRequest *request4 = [request3 copy];
- GHAssertNotNil(request4,@"Failed to create a copy");
-
- [pool release];
-
- success = ([request4 retainCount] > 0);
- GHAssertTrue(success,@"Failed to create a retained copy");
- success = ([request4 isKindOfClass:[ASIS3BucketRequest class]]);
- GHAssertTrue(success,@"Copy is of wrong class");
-
- [request4 release];
-
- pool = [[NSAutoreleasePool alloc] init];
-
- ASIS3BucketObject *bucketObject = [ASIS3BucketObject objectWithBucket:@"foo"];
- ASIS3BucketObject *bucketObject2 = [bucketObject copy];
- GHAssertNotNil(bucketObject2,@"Failed to create a copy");
-
- [pool release];
-
- success = ([bucketObject2 retainCount] > 0);
- GHAssertTrue(success,@"Failed to create a retained copy");
-
- [bucketObject2 release];
- }
- - (void)testHTTPS
- {
- [ASIS3Request setSharedAccessKey:accessKey];
- [ASIS3Request setSharedSecretAccessKey:secretAccessKey];
- // Create a bucket
- ASIS3Request *request = [ASIS3BucketRequest PUTRequestWithBucket:bucket];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to create a bucket");
- // PUT something in it
- NSString *key = @"king";
- request = [ASIS3ObjectRequest PUTRequestForData:[@"fink" dataUsingEncoding:NSUTF8StringEncoding] withBucket:bucket key:key];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- BOOL success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to PUT some data into S3");
- // GET it
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@"fink"];
- GHAssertTrue(success,@"Failed to GET the correct data from S3");
- // DELETE it
- request = [ASIS3ObjectRequest DELETERequestWithBucket:bucket key:@"king"];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to DELETE the object from S3");
- // Delete the bucket
- request = [ASIS3BucketRequest DELETERequestWithBucket:bucket];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to delete a bucket");
- [ASIS3Request setSharedAccessKey:nil];
- [ASIS3Request setSharedSecretAccessKey:nil];
- }
- // Ideally this test would actually parse the ACL XML and check it, but for now it just makes sure S3 doesn't return an error
- - (void)testCannedACLs
- {
- [ASIS3Request setSharedAccessKey:accessKey];
- [ASIS3Request setSharedSecretAccessKey:secretAccessKey];
- // Create a bucket
- ASIS3Request *request = [ASIS3BucketRequest PUTRequestWithBucket:bucket];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to create a bucket");
- NSArray *ACLs = [NSArray arrayWithObjects:ASIS3AccessPolicyPrivate,ASIS3AccessPolicyPublicRead,ASIS3AccessPolicyPublicReadWrite,ASIS3AccessPolicyAuthenticatedRead,ASIS3AccessPolicyBucketOwnerRead,ASIS3AccessPolicyBucketOwnerFullControl,nil];
- for (NSString *cannedACL in ACLs) {
- // PUT object
- NSString *key = @"king";
- request = [ASIS3ObjectRequest PUTRequestForData:[@"fink" dataUsingEncoding:NSUTF8StringEncoding] withBucket:bucket key:key];
- [request setAccessPolicy:cannedACL];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to PUT some data into S3");
- // GET object ACL
- request = [ASIS3ObjectRequest requestWithBucket:bucket key:key subResource:@"acl"];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to fetch the object");
- }
- // DELETE it
- request = [ASIS3ObjectRequest DELETERequestWithBucket:bucket key:@"king"];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- BOOL success = [[request responseString] isEqualToString:@""];
- GHAssertTrue(success,@"Failed to DELETE the object from S3");
- // Delete the bucket
- request = [ASIS3BucketRequest DELETERequestWithBucket:bucket];
- [request setRequestScheme:ASIS3RequestSchemeHTTPS];
- [request startSynchronous];
- GHAssertNil([request error],@"Failed to delete a bucket");
- [ASIS3Request setSharedAccessKey:nil];
- [ASIS3Request setSharedSecretAccessKey:nil];
- }
- @synthesize networkQueue;
- @end