Test responseImage on success and failure

krukawa · Jun 1, 2013 · 648b4fb · 648b4fb
1 parent 877a5a4
commit 648b4fb
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/Tests/AFImageRequestOperationTests.m b/Tests/AFImageRequestOperationTests.m
@@ -56,4 +56,22 @@ - (void)testThatImageRequestOperationDoesNotAcceptInvalidFormatTypes {
     expect(operation.error).willNot.beNil();
 }
 
+- (void)testThatImageResponseIsNotNilWhenRequestSucceeds {
+    NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:@"/response-headers?Content-Type=image/png" relativeToURL:self.baseURL]];
+    AFImageRequestOperation *operation = [[AFImageRequestOperation alloc] initWithRequest:request];
+    [operation start];
+
+    expect([operation isFinished]).will.beTruthy();
+    expect(operation.responseImage).willNot.beNil();
+}
+
+- (void)testThatImageResponseIsNilWhenRequestFails {
+    NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:@"/status/404" relativeToURL:self.baseURL]];
+    AFImageRequestOperation *operation = [[AFImageRequestOperation alloc] initWithRequest:request];
+    [operation start];
+
+    expect([operation isFinished]).will.beTruthy();
+    expect(operation.responseImage).will.beNil();
+}
+
 @end