Tests for notification and block use

ElfSundae · Jun 29, 2018 · c4caddc · c4caddc
1 parent aed00d3
commit c4caddc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Tests/Tests/AFURLSessionManagerTests.m b/Tests/Tests/AFURLSessionManagerTests.m
@@ -142,6 +142,12 @@ - (void)testSessionTaskDoesReportMetrics {
 #endif
         return YES;
     }];
+
+    __weak XCTestExpectation *metricsBlock = [self expectationWithDescription:@"Metrics completion block is called"];
+    [self.localManager setTaskDidFinishCollectingMetricsBlock:^(NSURLSession * _Nonnull session, NSURLSessionTask * _Nonnull task, NSURLSessionTaskMetrics * _Nullable metrics) {
+        [metricsBlock fulfill];
+    }];
+
     NSURLSessionTask *task = [self.localManager downloadTaskWithRequest:[self bigImageURLRequest]
                                                                progress:nil
                                                             destination:nil