fix UI test

RevenueCat · Dec 4, 2024 · f1eeadd · f1eeadd
1 parent 9340e05
commit f1eeadd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Tests/RevenueCatUITests/CustomerCenter/CustomerCenterViewModelTests.swift b/Tests/RevenueCatUITests/CustomerCenter/CustomerCenterViewModelTests.swift
@@ -207,7 +207,7 @@ class CustomerCenterViewModelTests: TestCase {
         viewModel.trackImpression(darkMode: darkMode, displayMode: displayMode)
 
         expect(mockPurchases.trackedEvents.count) == 1
-        let trackedEvent = try XCTUnwrap(mockPurchases.trackedEvents.first)
+        let trackedEvent = try XCTUnwrap(mockPurchases.trackedEvents.first as? CustomerCenterEvent)
 
         expect(trackedEvent.data.darkMode) == darkMode
         expect(trackedEvent.data.displayMode) == displayMode

diff --git a/Tests/RevenueCatUITests/CustomerCenter/MockCustomerCenterPurchases.swift b/Tests/RevenueCatUITests/CustomerCenter/MockCustomerCenterPurchases.swift
@@ -55,8 +55,8 @@ final class MockCustomerCenterPurchases: @unchecked Sendable, CustomerCenterPurc
 
     var trackCallCount = 0
     var trackError: Error?
-    var trackedEvents: [CustomerCenterEvent] = []
-    func track(customerCenterEvent: CustomerCenterEvent) {
+    var trackedEvents: [CustomerCenterEventType] = []
+    func track(customerCenterEvent: any CustomerCenterEventType) {
         trackCallCount += 1
         trackedEvents.append(customerCenterEvent)
     }