Skip to content

Commit d898b64

Browse files
authored
[Messaging] Fix flaky FIRMessagingTokenManagerTest tests (#8689)
1 parent 5a7c348 commit d898b64

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

FirebaseMessaging/Tests/UnitTests/FIRMessagingTokenManagerTest.m

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ @interface FIRMessaging (ExposedForTest)
3131

3232
@interface FIRMessagingTokenManager (ExposedForTest)
3333

34+
- (void)didDeleteFCMScopedTokensForCheckin:(FIRMessagingCheckinPreferences *)checkin;
35+
3436
- (void)resetCredentialsIfNeeded;
3537

3638
@end
@@ -151,6 +153,10 @@ - (void)testResetCredentialsWithFreshInstall {
151153
OCMStub([_mockAuthService checkinPreferences]).andReturn(checkinPreferences);
152154
// Plist file doesn't exist, meaning this is a fresh install.
153155
OCMStub([_mockCheckinStore hasCheckinPlist]).andReturn(NO);
156+
// Expect reset operation but do nothing to avoid flakes due to delayed operation queue.
157+
OCMExpect(
158+
[_mockTokenManager didDeleteFCMScopedTokensForCheckin:[OCMArg isEqual:checkinPreferences]])
159+
.andDo(nil);
154160

155161
[_messaging.tokenManager resetCredentialsIfNeeded];
156162
OCMVerifyAll(_mockCheckinStore);

0 commit comments

Comments
 (0)