Add tests for verifing reachability manager consistence for notificat…

…ion posting and getting
ElfSundae · Jul 15, 2018 · ffbcabe · ffbcabe
1 parent 06079ad
commit ffbcabe
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Tests/Tests/AFNetworkReachabilityManagerTests.m b/Tests/Tests/AFNetworkReachabilityManagerTests.m
@@ -125,4 +125,28 @@ - (void)testDomainReachabilityBlock {
     [self verifyReachabilityStatusBlockGetsCalledWithManager:self.domainReachability];
 }
 
+- (void)testDifferentReachabilityManagerNotificationPostsAndGets {
+    [self verifyReachabilityNotificationPostsAndGetsConsistenceWithManager:self.domainReachability notificationManager:self.addressReachability];
+}
+
+- (void)testSameReachabilityManagerNotificationPostsAndGets {
+    [self verifyReachabilityNotificationPostsAndGetsConsistenceWithManager:self.addressReachability notificationManager:self.addressReachability];
+}
+
+- (void)verifyReachabilityNotificationPostsAndGetsConsistenceWithManager:(AFNetworkReachabilityManager *)currentManager notificationManager:(AFNetworkReachabilityManager *)noteManager {
+    BOOL isSameReachabilityManager = [currentManager isEqual:noteManager];
+    [self expectationForNotification:AFNetworkingReachabilityDidChangeNotification
+                              object:nil
+                             handler:^BOOL(NSNotification *note) {
+                                 id currentReachabilityRef = (__bridge id)currentManager.currentNetworkReachability;
+                                 id noteReachabilityRef = note.object;
+                                 BOOL isSameReachabilityRef = [currentReachabilityRef isEqual:noteReachabilityRef];
+                                 return isSameReachabilityManager == isSameReachabilityRef;
+                             }];
+
+    [noteManager startMonitoring];
+
+    [self waitForExpectationsWithCommonTimeout];
+}
+
 @end