Merge 61594d8 into d3d350e

button · Aug 12, 2020 · 5b9b9ca · 5b9b9ca
2 parents d3d350e + 61594d8
commit 5b9b9ca
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/Source/Core.swift b/Source/Core.swift
@@ -177,7 +177,7 @@ final internal class Core: CoreType {
                     completion(nil, nil)
                     return
             }
-            self?.buttonDefaults.attributionToken = token
+            self?.updateAttributionIfNeeded(token: token)
             completion(url, nil)
         }
     }

diff --git a/Tests/UnitTests/CoreTests.swift b/Tests/UnitTests/CoreTests.swift
@@ -360,10 +360,11 @@ class CoreTests: XCTestCase {
         let expectation = XCTestExpectation(description: "handle post install url")
         let testDefaults = TestButtonDefaults(userDefaults: TestUserDefaults())
         let testSystem = TestSystem()
+        let testNotificationCenter = TestNotificationCenter()
         let core = Core(buttonDefaults: testDefaults,
                         client: testClient,
                         system: testSystem,
-                        notificationCenter: TestNotificationCenter())
+                        notificationCenter: testNotificationCenter)
         testDefaults.testHasFetchedPostInstallURL = false
         testSystem.testIsNewInstall = true
         core.applicationId = ApplicationId("app-123")
@@ -378,6 +379,13 @@ class CoreTests: XCTestCase {
             XCTAssertTrue(testDefaults.didStoreToken)
             XCTAssertNotNil(testDefaults.attributionToken)
             XCTAssertEqual(testDefaults.attributionToken, expectedToken)
+            XCTAssertEqual(core.attributionToken, expectedToken)
+            XCTAssertTrue(testNotificationCenter.didCallPostNotification)
+            XCTAssertNil(testNotificationCenter.testObject)
+            XCTAssertEqual(testNotificationCenter.testNotificationName,
+                           Notification.Name.Button.AttributionTokenDidChange)
+            XCTAssertEqual(testNotificationCenter.testUserInfo! as NSDictionary,
+                           [Notification.Key.NewToken: expectedToken])
             expectation.fulfill()
         }
         XCTAssertEqual(testClient.testParameters as NSDictionary,