From 77bb2d41e4ad04773df0bc7ad661aaecde39846f Mon Sep 17 00:00:00 2001
From: Jin Xu <jin.xu@amplitude.com>
Date: Tue, 17 Dec 2024 13:37:29 -0800
Subject: [PATCH] fix: fix likely-to-fail test case

---
 Tests/AmplitudeTests/Utilities/EventPipelineTests.swift | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift b/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift
index 99bd19c..e2da40e 100644
--- a/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift
+++ b/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift
@@ -252,11 +252,8 @@ final class EventPipelineTests: XCTestCase {
         pipeline.flush {
             flushExpectation.fulfill()
         }
-        wait(for: [uploadExpectations[0], uploadExpectations[1]], timeout: 1)
-        XCTAssertEqual(httpClient.uploadCount, 2)
-        XCTAssertEqual(pipeline.configuration.offline, false)
 
-        wait(for: [uploadExpectations[2], flushExpectation], timeout: 1)
+        wait(for: uploadExpectations + [flushExpectation], timeout: 1)
         XCTAssertEqual(httpClient.uploadCount, 3)
         XCTAssertEqual(pipeline.configuration.offline, false)
     }