From 648b4fb77fcf4e876fe6a9182365dba941989b35 Mon Sep 17 00:00:00 2001
From: Adam Fraser <adamfraser0@gmail.com>
Date: Sat, 1 Jun 2013 20:49:07 +1000
Subject: [PATCH] Test responseImage on success and failure

---
 Tests/AFImageRequestOperationTests.m | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/Tests/AFImageRequestOperationTests.m b/Tests/AFImageRequestOperationTests.m
index 459cf7f..ba0c6a4 100644
--- a/Tests/AFImageRequestOperationTests.m
+++ b/Tests/AFImageRequestOperationTests.m
@@ -56,4 +56,22 @@
     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