mdb1 · August 12, 2023 23:20 · Aug 12, 2023
diff --git a/NotificationCenterMock.swift b/NotificationCenterMock.swift
@@ -0,0 +1,42 @@
+/// Mock instance of NotificationCenter, to be used in tests' targets
+public final class NotificationCenterMock: NotificationCenter {
+    public var postCalls = 0
+    public var postedNotifications: [NSNotification.Name] = []
+    public var postReceivedObject: Any?
+    public var postReceivedUserInfos: [[AnyHashable: Any]] = []
+
+    public var addObserverCalls = 0
+    public var addObserverReceivedSelector: Selector?
+    public var addObserverReceivedName: NSNotification.Name?
+    public var addObserverReceivedObject: Any?
+
+    /// Initializer
+    override public init() {}
+}
+
+public extension NotificationCenterMock {
+    override func post(name aName: NSNotification.Name, object anObject: Any?, userInfo: [AnyHashable: Any]?) {
+        postCalls += 1
+        postedNotifications.append(aName)
+        postReceivedObject = anObject
+        if let userInfo {
+            postReceivedUserInfos.append(userInfo)
+        }
+        super.post(name: aName, object: anObject, userInfo: userInfo)
+    }
+}
+
+public extension NotificationCenterMock {
+    override func addObserver(
+        _ observer: Any,
+        selector aSelector: Selector,
+        name aName: NSNotification.Name?,
+        object anObject: Any?
+    ) {
+        addObserverCalls += 1
+        addObserverReceivedSelector = aSelector
+        addObserverReceivedName = aName
+        addObserverReceivedObject = anObject
+        super.addObserver(observer, selector: aSelector, name: aName, object: anObject)
+    }
+}