tests: Updates tests for async start+updateContext

Unleash · Oct 4, 2024 · 6190b51 · 6190b51
1 parent 66a298d
commit 6190b51
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/Tests/UnleashProxyClientSwiftTests/MockPoller.swift b/Tests/UnleashProxyClientSwiftTests/MockPoller.swift
@@ -49,6 +49,7 @@ public class MockDictionaryStorageProvider: StorageProvider {
 
 class MockPoller: Poller {
     var dataGenerator: () -> [String: Toggle];
+    var stubCompletionError: PollerError?
 
     init(callback: @escaping () -> [String: Toggle], unleashUrl: URL, apiKey: String, session: PollerSession) {
         self.dataGenerator = callback
@@ -57,5 +58,6 @@ class MockPoller: Poller {
 
     override func getFeatures(context: Context, completionHandler: ((PollerError?) -> Void)? = nil) -> Void {
         self.storageProvider = MockDictionaryStorageProvider(storage: dataGenerator())
+        completionHandler?(stubCompletionError)
     }
 }
diff --git a/Tests/UnleashProxyClientSwiftTests/UnleashProxyClientSwiftTests.swift b/Tests/UnleashProxyClientSwiftTests/UnleashProxyClientSwiftTests.swift
@@ -46,6 +46,21 @@
             XCTAssert(unleash.poller.timer != nil)
         }
 
+        @MainActor
+        func testTimerWithAsyncStart() async throws {
+            func dataGenerator() -> [String: UnleashProxyClientSwift.Toggle] {
+                return generateTestToggleMapWithVariant()
+            }
+
+            let mockedPoller = MockPollerSession()
+            let unleash = try await setup(
+                dataGenerator: dataGenerator,
+                session: mockedPoller
+            )
+
+            XCTAssert(unleash.poller.timer != nil)
+        }
+
         func testUpdateContext() {
             func dataGenerator() -> [String: UnleashProxyClientSwift.Toggle] {
                 return generateTestToggleMapWithVariant()
@@ -62,6 +77,24 @@
 
             XCTAssert(url.contains("appName=test") && url.contains("sessionId=uuid-234-test") && url.contains("userId=uuid-123-test") && url.contains("environment=dev"))
         }
+
+        @MainActor
+        func testUpdateContextAsync() async throws {
+            func dataGenerator() -> [String: UnleashProxyClientSwift.Toggle] {
+                return generateTestToggleMapWithVariant()
+            }
+
+            let unleash = try await setup(dataGenerator: dataGenerator)
+
+            var context: [String: String] = [:]
+            context["userId"] = "uuid-123-test"
+            context["sessionId"] = "uuid-234-test"
+            try await unleash.updateContext(context: context)
+
+            let url = unleash.poller.formatURL(context: unleash.context)!.absoluteString
+
+            XCTAssert(url.contains("appName=test") && url.contains("sessionId=uuid-234-test") && url.contains("userId=uuid-123-test") && url.contains("environment=dev"))
+        }
     }
 
     final class unleash_proxy_client_base_swiftTests: XCTestCase {

diff --git a/Tests/UnleashProxyClientSwiftTests/testUtils.swift b/Tests/UnleashProxyClientSwiftTests/testUtils.swift
@@ -42,6 +42,17 @@ func setup(dataGenerator: @escaping () -> [String: Toggle], session: PollerSessi
     return unleash
 }
 
+@available(iOS 13, *)
+func setup(dataGenerator: @escaping () -> [String: Toggle], session: PollerSession = MockPollerSession()) async throws -> UnleashClient {
+    let poller = MockPoller(callback: dataGenerator, unleashUrl: URL(string: "https://app.unleash-hosted.com/hosted/api/proxy")!, apiKey: "SECRET", session: session)
+    let metrics = MockMetrics(appName: "test")
+
+    let unleash = UnleashProxyClientSwift.UnleashClient(unleashUrl: "https://app.unleash-hosted.com/hosted/api/proxy", clientKey: "dss22d", refreshInterval: 15, appName: "test", environment: "dev", poller: poller, metrics: metrics)
+
+    try await unleash.start()
+    return unleash
+}
+
 func setupBase(dataGenerator: @escaping () -> [String: Toggle], session: PollerSession = MockPollerSession()) -> UnleashClientBase {
     let poller = MockPoller(callback: dataGenerator, unleashUrl: URL(string: "https://app.unleash-hosted.com/hosted/api/proxy")!, apiKey: "SECRET", session: session)
     let metrics = MockMetrics(appName: "test")
@@ -51,3 +62,14 @@ func setupBase(dataGenerator: @escaping () -> [String: Toggle], session: PollerS
     unleash.start()
     return unleash
 }
+
+@available(iOS 13, *)
+func setupBase(dataGenerator: @escaping () -> [String: Toggle], session: PollerSession = MockPollerSession()) async throws -> UnleashClientBase {
+    let poller = MockPoller(callback: dataGenerator, unleashUrl: URL(string: "https://app.unleash-hosted.com/hosted/api/proxy")!, apiKey: "SECRET", session: session)
+    let metrics = MockMetrics(appName: "test")
+
+    let unleash = UnleashProxyClientSwift.UnleashClientBase(unleashUrl: "https://app.unleash-hosted.com/hosted/api/proxy", clientKey: "dss22d", refreshInterval: 15, appName: "test", environment: "dev", poller: poller, metrics: metrics)
+
+    try await unleash.start()
+    return unleash
+}