From 3ecc2afc23d699387ae2784bb729a7a762bb28b9 Mon Sep 17 00:00:00 2001 From: Adam Fraser Date: Sat, 1 Jun 2013 20:50:03 +1000 Subject: [PATCH] Test image processing block --- Tests/AFImageRequestOperationTests.m | 34 ++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/Tests/AFImageRequestOperationTests.m b/Tests/AFImageRequestOperationTests.m index ba0c6a4..52a3649 100644 --- a/Tests/AFImageRequestOperationTests.m +++ b/Tests/AFImageRequestOperationTests.m @@ -74,4 +74,38 @@ expect(operation.responseImage).will.beNil(); } +- (void)testImageProcessingBlockIsRunOnSuccess { + __block UIImage *blockImage = nil; + NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:@"/response-headers?Content-Type=image/png" relativeToURL:self.baseURL]]; + + AFImageRequestOperation *operation = [AFImageRequestOperation imageRequestOperationWithRequest:request imageProcessingBlock:^UIImage *(UIImage *image) { + blockImage = [[UIImage alloc] init]; + return blockImage; + } success:^(NSURLRequest *request, NSHTTPURLResponse *response, UIImage *image) { + } failure:^(NSURLRequest *request, NSHTTPURLResponse *response, NSError *error) { + }]; + + [operation start]; + expect([operation isFinished]).will.beTruthy(); + expect(operation.error).will.beNil(); + expect(blockImage).willNot.beNil(); +} + +- (void)testImageProcessingBlockIsNotRunOnFailure { + __block UIImage *blockImage = nil; + NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:@"/status/404" relativeToURL:self.baseURL]]; + + AFImageRequestOperation *operation = [AFImageRequestOperation imageRequestOperationWithRequest:request imageProcessingBlock:^UIImage *(UIImage *image) { + blockImage = [[UIImage alloc] init]; + return blockImage; + } success:^(NSURLRequest *request, NSHTTPURLResponse *response, UIImage *image) { + } failure:^(NSURLRequest *request, NSHTTPURLResponse *response, NSError *error) { + }]; + [operation start]; + + expect([operation isFinished]).will.beTruthy(); + expect(operation.error).willNot.beNil(); + expect(blockImage).will.beNil(); +} + @end