22
22
import static org .mockito .Mockito .when ;
23
23
24
24
import android .app .Activity ;
25
+ import com .google .android .gms .tasks .SuccessContinuation ;
25
26
import com .google .android .gms .tasks .Task ;
26
27
import com .google .android .gms .tasks .Tasks ;
27
28
import com .google .firebase .appdistribution .FirebaseAppDistributionException .Status ;
28
29
import com .google .firebase .appdistribution .FirebaseAppDistributionLifecycleNotifier .ActivityConsumer ;
29
- import com .google .firebase .appdistribution .FirebaseAppDistributionLifecycleNotifier .ActivityContinuation ;
30
30
import org .junit .Before ;
31
31
import org .junit .Test ;
32
32
import org .junit .runner .RunWith ;
@@ -95,7 +95,7 @@ public void applyToForegroundActivityTask_withCurrentActivity_succeedsAndCallsCo
95
95
// applyToForegroundActivityTask is called
96
96
lifecycleNotifier .onActivityResumed (activity );
97
97
98
- ActivityContinuation < String > continuation = spy (ActivityContinuation .class );
98
+ SuccessContinuation < Activity , String > continuation = spy (SuccessContinuation .class );
99
99
when (continuation .then (activity )).thenReturn (Tasks .forResult ("result" ));
100
100
Task <String > task = lifecycleNotifier .applyToForegroundActivityTask (continuation );
101
101
@@ -107,7 +107,7 @@ public void applyToForegroundActivityTask_withCurrentActivity_succeedsAndCallsCo
107
107
@ Test
108
108
public void applyToForegroundActivityTask_noCurrentActivity_succeedsAndCallsConsumer ()
109
109
throws Exception {
110
- ActivityContinuation < String > continuation = spy (ActivityContinuation .class );
110
+ SuccessContinuation < Activity , String > continuation = spy (SuccessContinuation .class );
111
111
when (continuation .then (activity )).thenReturn (Tasks .forResult ("result" ));
112
112
Task <String > task = lifecycleNotifier .applyToForegroundActivityTask (continuation );
113
113
@@ -121,7 +121,7 @@ public void applyToForegroundActivityTask_noCurrentActivity_succeedsAndCallsCons
121
121
122
122
@ Test
123
123
public void applyToForegroundActivityTask_continuationThrows_fails () throws Exception {
124
- ActivityContinuation < String > continuation = spy (ActivityContinuation .class );
124
+ SuccessContinuation < Activity , String > continuation = spy (SuccessContinuation .class );
125
125
FirebaseAppDistributionException continuationException =
126
126
new FirebaseAppDistributionException (
127
127
"exception in continuation task" , Status .AUTHENTICATION_CANCELED );
@@ -137,7 +137,7 @@ public void applyToForegroundActivityTask_continuationThrows_fails() throws Exce
137
137
@ Test
138
138
public void applyToForegroundActivityTask_continuationThrowsUnknownException_wrapsException ()
139
139
throws Exception {
140
- ActivityContinuation < String > continuation = spy (ActivityContinuation .class );
140
+ SuccessContinuation < Activity , String > continuation = spy (SuccessContinuation .class );
141
141
RuntimeException continuationException = new RuntimeException ("exception in continuation" );
142
142
when (continuation .then (activity )).thenThrow (continuationException );
143
143
Task <String > task = lifecycleNotifier .applyToForegroundActivityTask (continuation );
@@ -151,7 +151,7 @@ public void applyToForegroundActivityTask_continuationThrowsUnknownException_wra
151
151
@ Test
152
152
public void applyToForegroundActivityTask_continuationTaskFails_failsWithSameException ()
153
153
throws Exception {
154
- ActivityContinuation < String > continuation = spy (ActivityContinuation .class );
154
+ SuccessContinuation < Activity , String > continuation = spy (SuccessContinuation .class );
155
155
RuntimeException continuationException = new RuntimeException ("exception in continuation" );
156
156
when (continuation .then (activity )).thenReturn (Tasks .forException (continuationException ));
157
157
Task <String > task = lifecycleNotifier .applyToForegroundActivityTask (continuation );
0 commit comments