Improve test

element-hq · Jul 13, 2023 · 2630a82 · 2630a82
1 parent 6245f9a
commit 2630a82
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/UnitTests/Sources/NotificationManager/NotificationManagerTests.swift b/UnitTests/Sources/NotificationManager/NotificationManagerTests.swift
@@ -223,10 +223,12 @@ final class NotificationManagerTests: XCTestCase {
     func test_InvitesNotificationsRemoval() async throws {
         let notificationPublisher = NotificationCenter.default.publisher(for: .invitesScreenAppeared).first()
         let expectation = expectation(description: #function)
-        let subscription = notificationPublisher
+        var cancellables: Set<AnyCancellable> = .init()
+        notificationPublisher
             .sink { _ in
                 expectation.fulfill()
             }
+            .store(in: &cancellables)
 
         NotificationCenter.default.post(name: .invitesScreenAppeared, object: nil)
         await fulfillment(of: [expectation])