|
25 | 25 | import static org.mockito.Mockito.spy; |
26 | 26 | import static org.mockito.Mockito.times; |
27 | 27 | import static org.mockito.Mockito.verify; |
28 | | -import static org.mockito.Mockito.verifyNoMoreInteractions; |
| 28 | +import static org.mockito.Mockito.verifyNoInteractions; |
29 | 29 | import static org.mockito.Mockito.when; |
30 | 30 | import static org.robolectric.Shadows.shadowOf; |
31 | 31 |
|
@@ -337,7 +337,8 @@ public void getToken_withFiid() throws Exception { |
337 | 337 |
|
338 | 338 | ShadowLooper.idleMainLooper(); |
339 | 339 | assertThat(Tasks.await(getTokenTask, 5, SECONDS)).isEqualTo("fake_token"); |
340 | | - verify(mockGmsRpc).getToken(false); |
| 340 | + verify(mockFiid).getTokenTask(); |
| 341 | + verifyNoInteractions(mockGmsRpc); |
341 | 342 | } |
342 | 343 |
|
343 | 344 | @Test |
@@ -387,15 +388,17 @@ public void deleteToken_withFiid() throws Exception { |
387 | 388 | Runnable::run); |
388 | 389 |
|
389 | 390 | writeTokenToStore("fake_token"); |
390 | | - when(mockGmsRpc.deleteToken(false)).thenReturn(Tasks.forResult(null)); |
391 | 391 | Task<Void> deleteTokenTask = messaging.deleteToken(); |
392 | 392 |
|
393 | 393 | ShadowLooper.idleMainLooper(); |
394 | 394 | Tasks.await(deleteTokenTask, 5, SECONDS); |
395 | 395 | ShadowLooper.idleMainLooper(); |
396 | 396 | Tasks.await(deleteTokenTask, 5, SECONDS); |
397 | | - verify(mockGmsRpc).deleteToken(false); |
398 | | - verifyNoMoreInteractions(mockGmsRpc); |
| 397 | + verify(mockFiid) |
| 398 | + .deleteToken( |
| 399 | + Metadata.getDefaultSenderId(FirebaseApp.getInstance()), |
| 400 | + FirebaseMessaging.INSTANCE_ID_SCOPE); |
| 401 | + verifyNoInteractions(mockGmsRpc); |
399 | 402 | } |
400 | 403 |
|
401 | 404 | @Test |
|
0 commit comments