BlocksTests.m
3.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
//
// BlocksTests.m
// Mac
//
// Created by Ben Copsey on 18/10/2010.
// Copyright 2010 All-Seeing Interactive. All rights reserved.
//
#import "BlocksTests.h"
#import "ASIHTTPRequest.h"
@implementation BlocksTests
// ASIHTTPRequest always calls blocks on the main thread (just like it does with delegate methods)
// So, we'll force this request to run on the main thread so we can rely on blocks having been called before the request returns
- (BOOL)shouldRunOnMainThread { return YES; }
#if NS_BLOCKS_AVAILABLE
#if TARGET_OS_IPHONE
// It isn't safe to allow the view to deallocate on a thread other than the main thread / web thread, so this test is designed to cause a crash semi-reliably
- (void)testBlockMainThreadSafety
{
NSURL *url = [NSURL URLWithString:@"http://allseeing-i.com"];
UIWebView *webView = [[[UIWebView alloc] initWithFrame:CGRectMake(0,0,200,200)] autorelease];
__block ASIHTTPRequest *request = [ASIHTTPRequest requestWithURL:url];
[request setCompletionBlock:^ {[webView loadHTMLString:[request responseString] baseURL:url]; }];
[request startAsynchronous];
}
#endif
- (void)testBlocks
{
NSData *dataToSend = [@"This is my post body" dataUsingEncoding:NSUTF8StringEncoding];
__block BOOL started = NO;
__block BOOL receivedHeaders = NO;
__block BOOL complete = NO;
__block BOOL failed = NO;
__block unsigned long long totalBytesReceived = 0;
__block unsigned long long totalDownloadSize = 0;
__block unsigned long long totalBytesSent = 0;
__block unsigned long long totalUploadSize = 0;
NSMutableData *dataReceived = [NSMutableData data];
// There's actually no need for us to use '__block' here, because we aren't using the request inside any of our blocks, but it's good to get into the habit of doing this anyway.
__block ASIHTTPRequest *request = [ASIHTTPRequest requestWithURL:[NSURL URLWithString:@"http://allseeing-i.com/ASIHTTPRequest/tests/blocks"]];
[request setStartedBlock:^{
started = YES;
}];
[request setHeadersReceivedBlock:^(NSDictionary *headers) {
receivedHeaders = YES;
}];
[request setCompletionBlock:^{
complete = YES;
}];
[request setFailedBlock:^{
failed = YES;
}];
[request setBytesReceivedBlock:^(unsigned long long length, unsigned long long total) {
totalBytesReceived += length;
}];
[request setDownloadSizeIncrementedBlock:^(long long length){
totalDownloadSize += length;
}];
[request setBytesSentBlock:^(unsigned long long length, unsigned long long total) {
totalBytesSent += length;
}];
[request setUploadSizeIncrementedBlock:^(long long length){
totalUploadSize += length;
}];
[request setDataReceivedBlock:^(NSData *data){
[dataReceived appendData:data];
}];
[request setRequestMethod:@"PUT"];
[request appendPostData:dataToSend];
[request startSynchronous];
GHAssertFalse(failed,@"Request failed, cannot proceed with test");
GHAssertTrue(started,@"Failed to call started block");
GHAssertTrue(receivedHeaders,@"Failed to call received headers block");
GHAssertTrue(complete,@"Failed to call completed block");
BOOL success = (totalBytesReceived == 457);
GHAssertTrue(success,@"Failed to call bytes received block, or got wrong amount of data");
success = (totalDownloadSize == 457);
GHAssertTrue(success,@"Failed to call download size increment block");
success = (totalBytesSent == [dataToSend length]);
GHAssertTrue(success,@"Failed to call bytes sent block");
success = (totalUploadSize == [dataToSend length]);
GHAssertTrue(success,@"Failed to call upload size increment block");
request = [ASIHTTPRequest requestWithURL:nil];
[request setFailedBlock:^{
failed = YES;
}];
[request startSynchronous];
GHAssertTrue(failed,@"Failed to call request failure block");
}
#endif
@end