Explicitly set the expectation to nil to avoid calling fulfill twice

xiaoyongLi-letcook · Oct 12, 2016 · 025c82f · 025c82f
1 parent 83c33fb
commit 025c82f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Tests/Tests/AFNetworkReachabilityManagerTests.m b/Tests/Tests/AFNetworkReachabilityManagerTests.m
@@ -87,7 +87,7 @@ - (void)testDomainReachabilityNotification {
 
 - (void)verifyReachabilityStatusBlockGetsCalledWithManager:(AFNetworkReachabilityManager *)manager
 {
-    __weak XCTestExpectation *expectation = [self expectationWithDescription:@"reachability status change block gets called"];
+    __weak __block XCTestExpectation *expectation = [self expectationWithDescription:@"reachability status change block gets called"];
 
     typeof(manager) __weak weakManager = manager;
     [manager setReachabilityStatusChangeBlock:^(AFNetworkReachabilityStatus status) {
@@ -98,6 +98,7 @@ - (void)verifyReachabilityStatusBlockGetsCalledWithManager:(AFNetworkReachabilit
 
         if (reachable) {
             [expectation fulfill];
+            expectation = nil;
         }
     }];