fix messaging breakage (#6139)

charlotteliang · web-flow · commit 1c3abf1afe89 · 2020-07-28T09:56:36.000-07:00
diff --git a/FirebaseMessaging/Tests/UnitTests/FIRMessagingRemoteNotificationsProxyTest.m b/FirebaseMessaging/Tests/UnitTests/FIRMessagingRemoteNotificationsProxyTest.m
@@ -159,8 +159,9 @@ - (void)setUp {
   // Update +sharedProxy to always return our test instance
   OCMStub([_mockProxyClass sharedProxy]).andReturn(self.proxy);
   if (@available(macOS 10.14, iOS 10.0, *)) {
-    _mockUserNotificationCenter =
-        OCMPartialMock([UNUserNotificationCenter currentNotificationCenter]);
+    _mockUserNotificationCenter = OCMClassMock([UNUserNotificationCenter class]);
+    OCMStub([_mockUserNotificationCenter currentNotificationCenter])
+        .andReturn(_mockUserNotificationCenter);
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -159,8 +159,9 @@ - (void)setUp {`
`159`	`159`	`// Update +sharedProxy to always return our test instance`
`160`	`160`	`OCMStub([_mockProxyClass sharedProxy]).andReturn(self.proxy);`
`161`	`161`	`if (@available(macOS 10.14, iOS 10.0, *)) {`
`162`		`- _mockUserNotificationCenter =`
`163`		`- OCMPartialMock([UNUserNotificationCenter currentNotificationCenter]);`
	`162`	`+ _mockUserNotificationCenter = OCMClassMock([UNUserNotificationCenter class]);`
	`163`	`+ OCMStub([_mockUserNotificationCenter currentNotificationCenter])`
	`164`	`+ .andReturn(_mockUserNotificationCenter);`
`164`	`165`	`}`
`165`	`166`	`}`
`166`	`167`