test: fix failures on watchOS (#144)

amplitude · Mar 22, 2024 · f14fe5d · f14fe5d
1 parent a6f1559
commit f14fe5d
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift b/Tests/AmplitudeTests/Utilities/EventPipelineTests.swift
@@ -26,12 +26,15 @@ final class EventPipelineTests: XCTestCase {
         configuration = Configuration(
             apiKey: "testApiKey",
             flushIntervalMillis: Int(Self.FLUSH_INTERVAL_SECONDS * 1000),
-            storageProvider: storage
+            storageProvider: storage,
+            offline: NetworkConnectivityCheckerPlugin.Disabled
         )
         let amplitude = Amplitude(configuration: configuration)
         httpClient = FakeHttpClient(configuration: configuration, diagnostics: configuration.diagonostics)
         pipeline = EventPipeline(amplitude: amplitude)
         pipeline.httpClient = httpClient
+        pipeline.flushTimer?.suspend()
+
     }
 
     override func tearDown() {
@@ -74,6 +77,8 @@ final class EventPipelineTests: XCTestCase {
     }
 
     func testFlushWhenOffline() {
+        pipeline.amplitude.configuration.offline = false
+
         let testEvent = BaseEvent(userId: "unit-test", deviceId: "unit-test-machine", eventType: "testEvent")
         try? pipeline.storage?.write(key: StorageKey.EVENTS, value: testEvent)
 
@@ -91,6 +96,9 @@ final class EventPipelineTests: XCTestCase {
         let testEvent = BaseEvent(userId: "unit-test", deviceId: "unit-test-machine", eventType: "testEvent")
         try? pipeline.storage?.write(key: StorageKey.EVENTS, value: testEvent)
 
+        let httpResponseExpectation = expectation(description: "httpresponse")
+        httpClient.uploadExpectations = [httpResponseExpectation]
+
         let flushExpectations = (0..<2).map { _ in
             let expectation = expectation(description: "flush")
             pipeline.flush {
@@ -99,8 +107,9 @@ final class EventPipelineTests: XCTestCase {
             return expectation
         }
 
-        let waitResult = XCTWaiter.wait(for: flushExpectations, timeout: 1)
-        XCTAssertNotEqual(waitResult, .timedOut)
+        wait(for: flushExpectations, timeout: 1)
+        wait(for: [httpResponseExpectation], timeout: 1)
+
         XCTAssertEqual(httpClient.uploadCount, 1)
         let uploadedEvents = BaseEvent.fromArrayString(jsonString: httpClient.uploadedEvents[0])
         XCTAssertEqual(uploadedEvents?.count, 1)