感谢分享
// ASIFormDataRequestTests.m
// Part of ASIHTTPRequest -> http://allseeing-i.com/ASIHTTPRequest
//
// Created by Ben Copsey on 08/11/2008.
// Copyright 2008 All-Seeing Interactive. All rights reserved.
//
#import "ASIFormDataRequestTests.h"
#import "ASIFormDataRequest.h"
// Used for subclass test
@interface ASIFormDataRequestSubclass : ASIFormDataRequest {}
@end
@implementation ASIFormDataRequestSubclass;
@end
@implementation ASIFormDataRequestTests
-(void)testDefaultMethod
{
ASIFormDataRequest *request = [[[ASIFormDataRequest alloc] initWithURL:[NSURLURLWithString:@"http://wedontcare.com"]] autorelease];
GHAssertTrue([[request requestMethod] isEqualToString:@"POST"], @"Default request method should be POST");
}
- (void)testAddNilKeysAndValues
{
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/empty-post-value"]];
[request setPostValue:nil forKey:@"key1"];
[request setPostValue:@"value2" forKey:@"key2"];
[request setData:nil forKey:@"file1"];
[request setData:[@"hello" dataUsingEncoding:NSUTF8StringEncoding] forKey:@"file2"];
[request startSynchronous];
BOOL success = ([[request responseString] isEqualToString:@"key1: key2: value2 file1: file2: hello"]);
GHAssertTrue(success, @"Sent wrong data");
// Test nil key (no key or value should be sent to the server)
request = [ASIFormDataRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com"]];
[request addPostValue:@"value1" forKey:nil];
[request addPostValue:@"value2" forKey:@"key2"];
[request buildPostBody];
NSString *postBody = [[[NSString alloc] initWithData:[request postBody] encoding:NSUTF8StringEncoding] autorelease];
success = ([postBody isEqualToString:@"key2=value2"]);
GHAssertTrue(success, @"Sent wrong data");
}
- (void)testPostWithFileUpload
{
NSURL *url = [NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/post"];
//Create a 32kb file
unsigned int size = 1024*32;
NSMutableData *data = [NSMutableData dataWithLength:size];
NSString *path = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:@"bigfile"];
[data writeToFile:path atomically:NO];
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
NSDate *d = [NSDate date];
#if TARGET_OS_IPHONE
NSValue *v = [NSValue valueWithCGRect:CGRectMake(0, 0, 200, 200)];
#else
NSValue *v = [NSValue valueWithRect:NSMakeRect(0, 0, 200, 200)];
#endif
[request setPostValue:@"foo" forKey:@"post_var"];
[request setPostValue:d forKey:@"post_var2"];
[request setPostValue:v forKey:@"post_var3"];
[request setFile:path forKey:@"file"];
[request startSynchronous];
BOOL success = ([[request responseString] isEqualToString:[NSStringstringWithFormat:@"post_var: %@ post_var2: %@ post_var3: %@ file_name: %@ file_size: %hu content_type: %@",@"foo",d,v,@"bigfile",size,@"application/octet-stream"]]);
GHAssertTrue(success,@"Failed to upload the correct data (using local file)");
//Try the same with the raw data
request = [[[ASIFormDataRequest alloc] initWithURL:url] autorelease];
[request setPostValue:@"foo" forKey:@"post_var"];
[request setPostValue:d forKey:@"post_var2"];
[request setPostValue:v forKey:@"post_var3"];
[request setData:data forKey:@"file"];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"post_var: %@ post_var2: %@ post_var3: %@ file_name: %@ file_size: %hu content_type: %@",@"foo",d,v,@"file",size,@"application/octet-stream"]]);
GHAssertTrue(success,@"Failed to upload the correct data (using NSData)");
//Post with custom content-type and file name
request = [[[ASIFormDataRequest alloc] initWithURL:url] autorelease];
[request setPostValue:@"foo" forKey:@"post_var"];
[request setPostValue:d forKey:@"post_var2"];
[request setPostValue:v forKey:@"post_var3"];
[request setFile:path withFileName:@"myfile" andContentType:@"text/plain" forKey:@"file"];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"post_var: %@ post_var2: %@ post_var3: %@ file_name: %@ file_size: %hu content_type: %@",@"foo",d,v,@"myfile",size,@"text/plain"]]);
GHAssertTrue(success,@"Failed to send the correct content-type / file name");
//Post raw data with custom content-type and file name
request = [[[ASIFormDataRequest alloc] initWithURL:url] autorelease];
[request setPostValue:@"foo" forKey:@"post_var"];
[request setPostValue:d forKey:@"post_var2"];
[request setPostValue:v forKey:@"post_var3"];
[request setData:data withFileName:@"myfile" andContentType:@"text/plain" forKey:@"file"];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"post_var: %@ post_var2: %@ post_var3: %@ file_name: %@ file_size: %hu content_type: %@",@"foo",d,v,@"myfile",size,@"text/plain"]]);
GHAssertTrue(success,@"Failed to send the correct content-type / file name");
}
// Test fix for bug where setting an empty string for a form post value would cause the rest of the post body to be ignored (because an NSOutputStream won't like it if you try to write 0 bytes)
- (void)testEmptyData
{
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/post-empty"]];
[request setPostValue:@"hello" forKey:@"a_non_empty_string"];
[request setPostValue:@"" forKey:@"zzz_empty_string"];
[request setPostValue:@"there" forKey:@"xxx_non_empty_string"];
[request setShouldStreamPostDataFromDisk:YES];
[request buildPostBody];
[request startSynchronous];
BOOL success = ([[request responseString] isEqualToString:@"a_non_empty_string: hello zzz_empty_string: xxx_non_empty_string: there"]);
GHAssertTrue(success,@"Failed to send the correct post data");
}
// Ensure class convenience constructor returns an instance of our subclass
- (void)testSubclass
{
ASIFormDataRequestSubclass *instance = [ASIFormDataRequestSubclass requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com"]];
BOOL success = [instance isKindOfClass:[ASIFormDataRequestSubclass class]];
GHAssertTrue(success,@"Convenience constructor failed to return an instance of the correct class");
}
- (void)testURLEncodedPost
{
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/url-encoded-post"]];
[request setPostValue:@"value1" forKey:@"value1"];
[request setPostValue:@"(%20 ? =)" forKey:@"value2"];
[request setPostValue:@"£100.00" forKey:@"value3"];
[request setPostValue:@"" forKey:@"value4"];
[request setPostValue:@"&??aaa=//ciaoèèè" forKey:@"teskey&aa"];
[request setShouldStreamPostDataFromDisk:YES];
[request setPostFormat:ASIURLEncodedPostFormat];
[request startSynchronous];
BOOL success = ([[request responseString] isEqualToString:@"value1: value1 value2: (%20 ? =) value3: £100.00 value4: teskey&aa: &??aaa=//ciaoèèè"]);
GHAssertTrue(success,@"Failed to send the correct post data");
}
- (void)testCharset
{
NSURL *url = [NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/formdata-charset"];
NSString *testString = @"£££s don't seem to buy me many €€€s these days";
// Test the default (UTF-8) with a url-encoded request
NSString *charset = @"utf-8";
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request startSynchronous];
BOOL success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
// Test the default (UTF-8) with a multipart/form-data request
request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request setPostFormat:ASIMultipartFormDataPostFormat];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
// Test a different charset
testString = @"£££s don't seem to buy me many $$$s these days";
charset = @"iso-8859-1";
request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request setStringEncoding:NSISOLatin1StringEncoding];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
// And again with multipart/form-data request
request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request setPostFormat:ASIMultipartFormDataPostFormat];
[request setStringEncoding:NSISOLatin1StringEncoding];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
// Once more for luck
charset = @"windows-1252";
request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request setStringEncoding:NSWindowsCP1252StringEncoding];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
request = [ASIFormDataRequest requestWithURL:url];
[request setPostValue:testString forKey:@"value"];
[request setPostFormat:ASIMultipartFormDataPostFormat];
[request setStringEncoding:NSWindowsCP1252StringEncoding];
[request startSynchronous];
success = ([[request responseString] isEqualToString:[NSString stringWithFormat:@"Got data in %@: %@",charset,testString]]);
GHAssertTrue(success,@"Failed to correctly encode the data");
// Ensure charset isn't added to file post (GH issue 36)
request = [ASIFormDataRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/Tests/return-raw-request"]];
[request setData:[@"test 123" dataUsingEncoding:NSUTF8StringEncoding] forKey:@"file"];
[request setRequestMethod:@"PUT"];
[request startSynchronous];
success = ([[request responseString] rangeOfString:@"charset=utf-8"].location == NSNotFound);
GHAssertTrue(success,@"Sent a charset header for an uploaded file");
}
- (void)testPUT
{
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com/ASIHTTPRequest/Tests/put_form_data"]];
[request setRequestMethod:@"PUT"];
[request setPostValue:@"cheep cheep" forKey:@"hello"];
[request startSynchronous];
NSString *expectedResponse = [[[NSString alloc] initWithBytes:[[request postBody] bytes] length:[[request postBody] length] encoding:[request stringEncoding]] autorelease];
BOOL success = ([[request responseString] isEqualToString:expectedResponse]);
GHAssertTrue(success,@"Failed to send form data using PUT");
// Ensure that other methods still default to POST
request = [ASIFormDataRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/Tests/put_form_data"]];
[request setRequestMethod:@"DELETE"];
[request setPostValue:@"cheep cheep" forKey:@"hello"];
[request startSynchronous];
success = ([[request responseString] isEqualToString:@"Got POST instead"]);
GHAssertTrue(success,@"Failed to send form data using PUT");
}
- (void)testCopy
{
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com"]];
ASIFormDataRequest *request2 = [request copy];
GHAssertNotNil(request2,@"Failed to create a copy");
[pool release];
BOOL success = ([request2 retainCount] == 1);
GHAssertTrue(success,@"Failed to create a retained copy");
success = ([request2 isKindOfClass:[ASIFormDataRequest class]]);
GHAssertTrue(success,@"Copy is of wrong class");
[request2 release];
}
- (void)testMultipleValuesForASingleKey
{
ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:[NSURLURLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/multiple-values"]];
[request addPostValue:@"here" forKey:@"test_value[]"];
[request addPostValue:@"are" forKey:@"test_value[]"];
[request addPostValue:@"some" forKey:@"test_value[]"];
[request addPostValue:@"values" forKey:@"test_value[]"];
NSString *path1 = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:@"file1.txt"];
NSString *path2 = [[self filePathForTemporaryTestFiles] stringByAppendingPathComponent:@"file2.txt"];
[@"hello" writeToFile:path1 atomically:NO encoding:NSUTF8StringEncoding error:nil];
[@"there" writeToFile:path2 atomically:NO encoding:NSUTF8StringEncoding error:nil];
[request addFile:path1 forKey:@"test_file[]"];
[request addFile:path2 forKey:@"test_file[]"];
[request startSynchronous];
NSString *expectedOutput = @"here are some values file1.txt file2.txt ";
BOOL success = [[request responseString] isEqualToString:expectedOutput];
GHAssertTrue(success,@"Failed to send the correct data");
// Check data replaces older data
request = [ASIFormDataRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/single-values"]];
[request addPostValue:@"here" forKey:@"test_value[]"];
[request addPostValue:@"are" forKey:@"test_value[]"];
[request addPostValue:@"some" forKey:@"test_value[]"];
[request addPostValue:@"values" forKey:@"test_value[]"];
[request setPostValue:@"this is new data" forKey:@"test_value[]"];
[request addFile:path1 forKey:@"test_file[]"];
[request addFile:path2 forKey:@"test_file[]"];
[request setData:[@"this is new data" dataUsingEncoding:NSUTF8StringEncoding] forKey:@"test_file[]"];
[request startSynchronous];
expectedOutput = @"this is new data file ";
success = [[request responseString] isEqualToString:expectedOutput];
GHAssertTrue(success,@"Failed to send the correct data");
}
@end