Fix NetworkInstrumentationFeatureTests assertions

DataDog · Dec 14, 2023 · f99a874 · f99a874
1 parent 894bf48
commit f99a874
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/DatadogInternal/Tests/NetworkInstrumentation/NetworkInstrumentationFeatureTests.swift b/DatadogInternal/Tests/NetworkInstrumentation/NetworkInstrumentationFeatureTests.swift
@@ -215,9 +215,9 @@ class NetworkInstrumentationFeatureTests: XCTestCase {
             .resume()
 
         // Then
-        waitForExpectations(timeout: 5, handler: nil)
-        _ = server.waitAndReturnRequests(count: 1)
+        _ = server.waitAndReturnRequests(count: 2)
 
+        waitForExpectations(timeout: 5, handler: nil)
         let dateAfterAllRequests = Date()
 
         XCTAssertEqual(handler.interceptions.count, 2, "Interceptor should record metrics for 2 tasks")
@@ -286,11 +286,10 @@ class NetworkInstrumentationFeatureTests: XCTestCase {
             .resume()
 
         // Then
-        waitForExpectations(timeout: 5, handler: nil)
-        _ = server.waitAndReturnRequests(count: 1)
+        _ = server.waitAndReturnRequests(count: 2)
 
+        waitForExpectations(timeout: 5, handler: nil)
         let dateAfterAllRequests = Date()
-
         XCTAssertEqual(handler.interceptions.count, 2, "Interceptor should record metrics for 2 tasks")
 
         try [url1, url2].forEach { url in