test: add test for configured user provider

amplitude · Aug 10, 2021 · 401a0f3 · 401a0f3
1 parent 1e4a05c
commit 401a0f3
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/Tests/ExperimentTests/ExperimentClientTests.swift b/Tests/ExperimentTests/ExperimentClientTests.swift
@@ -144,6 +144,32 @@ class ExperimentClientTests: XCTestCase {
         XCTAssertEqual(expectedUserAfterMerge, mergedUser)
     }
 
+    func testMergeUserWithConfiguredProvider() {
+        let client = DefaultExperimentClient(
+            apiKey: API_KEY,
+            config: ExperimentConfig.Builder()
+                .userProvider(TestUserProvider())
+                .build(),
+            storage: InMemoryStorage()
+        )
+        let user = ExperimentUser.Builder()
+            .deviceId("device_id")
+            .userId(nil)
+            .version("version")
+            .build()
+        client.setUser(user)
+        let mergedUser = client.mergeUserWithProvider()
+        let expectedUserAfterMerge = ExperimentUser.Builder()
+            .deviceId("device_id")
+            .userId(nil)
+            .version("version")
+            .language("")
+            .library("\(ExperimentConfig.Constants.Library)/\(ExperimentConfig.Constants.Version)")
+            .build()
+        XCTAssertEqual(expectedUserAfterMerge, mergedUser)
+    }
+
+
     func testInitialVariantsSourceOverridesFetch() {
         var variant = initialVariantSourceClient.variant(KEY, fallback: nil)
         XCTAssertNotNil(variant)