Merge pull request AFNetworking#2432 from NSElvis/improve/UIRefreshCo…

…ntrol UIRefreshControl should react over task with completed state
jamesrusso · Nov 17, 2014 · 85f2bae · 85f2bae
2 parents 568f9fe + e2805f7
commit 85f2bae
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/UIKit+AFNetworking/UIRefreshControl+AFNetworking.m b/UIKit+AFNetworking/UIRefreshControl+AFNetworking.m
@@ -41,16 +41,14 @@ - (void)setRefreshingWithStateOfTask:(NSURLSessionTask *)task {
     [notificationCenter removeObserver:self name:AFNetworkingTaskDidCompleteNotification object:nil];
 
     if (task) {
-        if (task.state != NSURLSessionTaskStateCompleted) {
-            if (task.state == NSURLSessionTaskStateRunning) {
-                [self beginRefreshing];
-            } else {
-                [self endRefreshing];
-            }
+        if (task.state == NSURLSessionTaskStateRunning) {
+            [self beginRefreshing];
 
             [notificationCenter addObserver:self selector:@selector(af_beginRefreshing) name:AFNetworkingTaskDidResumeNotification object:task];
             [notificationCenter addObserver:self selector:@selector(af_endRefreshing) name:AFNetworkingTaskDidCompleteNotification object:task];
             [notificationCenter addObserver:self selector:@selector(af_endRefreshing) name:AFNetworkingTaskDidSuspendNotification object:task];
+        } else {
+            [self endRefreshing];
         }
     }
 }