From c4caddc5e98b0fbab5937d97c75977dee9035e51 Mon Sep 17 00:00:00 2001
From: Caelin Jackson-King <caelin@yelp.com>
Date: Fri, 29 Jun 2018 10:50:58 -0700
Subject: [PATCH] Tests for notification and block use

---
 Tests/Tests/AFURLSessionManagerTests.m | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Tests/Tests/AFURLSessionManagerTests.m b/Tests/Tests/AFURLSessionManagerTests.m
index c43e673b9c..deb067dfe2 100644
--- 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