3636import org .junit .runner .RunWith ;
3737import org .robolectric .Robolectric ;
3838import org .robolectric .RobolectricTestRunner ;
39+ import org .robolectric .android .controller .ActivityController ;
3940import org .robolectric .Shadows ;
4041import org .robolectric .annotation .LooperMode ;
4142import org .robolectric .shadows .ShadowActivity ;
@@ -77,7 +78,8 @@ public void testOnCreate_emailLinkNormalFlow_expectCheckEmailFlowStarted() {
7778 public void testOnCreate_emailLinkLinkingFlow_expectSendEmailLinkFlowStarted () {
7879 // This is normally done by EmailLinkSendEmailHandler, saving the IdpResponse is done
7980 // in EmailActivity but it will not be saved if we haven't previously set the email
80- EmailLinkPersistenceManager .getInstance ().saveEmail (ApplicationProvider .getApplicationContext (),
81+ EmailLinkPersistenceManager .getInstance ().saveEmail (
82+ ApplicationProvider .getApplicationContext (),
8183 EMAIL , TestConstants .SESSION_ID , TestConstants .UID );
8284
8385 EmailActivity emailActivity = createActivity (AuthUI .EMAIL_LINK_PROVIDER , true , false );
@@ -99,9 +101,7 @@ public void testOnCreate_emailLinkLinkingFlow_expectSendEmailLinkFlowStarted() {
99101 // @Test TODO(lsirac): uncomment after figuring out why this no longer works
100102 public void testOnTroubleSigningIn_expectTroubleSigningInFragment () {
101103 EmailActivity emailActivity = createActivity (AuthUI .EMAIL_LINK_PROVIDER );
102-
103104 emailActivity .onTroubleSigningIn (EMAIL );
104-
105105 TroubleSigningInFragment fragment = (TroubleSigningInFragment ) emailActivity
106106 .getSupportFragmentManager ().findFragmentByTag (TroubleSigningInFragment .TAG );
107107 assertThat (fragment ).isNotNull ();
@@ -110,32 +110,24 @@ public void testOnTroubleSigningIn_expectTroubleSigningInFragment() {
110110 @ Test
111111 public void testOnClickResendEmail_expectSendEmailLinkFlowStarted () {
112112 EmailActivity emailActivity = createActivity (AuthUI .EMAIL_LINK_PROVIDER );
113-
114113 emailActivity .onClickResendEmail (EMAIL );
115-
116114 shadowOf (Looper .getMainLooper ()).idle ();
117-
118115 EmailLinkFragment fragment = (EmailLinkFragment ) emailActivity
119116 .getSupportFragmentManager ().findFragmentByTag (EmailLinkFragment .TAG );
120117 assertThat (fragment ).isNotNull ();
121118 }
122119
123-
124120 @ Test
125121 public void testSignUpButton_validatesFields () {
126122 EmailActivity emailActivity = createActivity (EmailAuthProvider .PROVIDER_ID );
127-
128123 // Trigger RegisterEmailFragment (bypass check email)
129124 emailActivity .onNewUser (
130125 new User .Builder (EmailAuthProvider .PROVIDER_ID , TestConstants .EMAIL ).build ());
131-
132126 shadowOf (Looper .getMainLooper ()).idle ();
133127 Button button = emailActivity .findViewById (R .id .button_create );
134128 button .performClick ();
135-
136129 TextInputLayout nameLayout = emailActivity .findViewById (R .id .name_layout );
137130 TextInputLayout passwordLayout = emailActivity .findViewById (R .id .password_layout );
138-
139131 assertEquals (
140132 emailActivity .getString (R .string .fui_missing_first_and_last_name ),
141133 nameLayout .getError ().toString ());
@@ -153,42 +145,33 @@ public void testSignUpButton_validatesFields() {
153145 @ Test
154146 public void testSetDefaultEmail_validField () {
155147 EmailActivity emailActivity = createActivity (EmailAuthProvider .PROVIDER_ID , false , true );
156-
157148 CheckEmailFragment fragment = (CheckEmailFragment ) emailActivity
158149 .getSupportFragmentManager ().findFragmentByTag (CheckEmailFragment .TAG );
159150 assertThat (fragment ).isNotNull ();
160-
161- TextInputEditText email = emailActivity .findViewById (R .id .email );
162- assertEquals (TestConstants .EMAIL , email .getText ().toString ());
163-
151+ TextInputEditText email = emailActivity .findViewById (R .id .email );
152+ assertEquals (TestConstants .EMAIL , email .getText ().toString ());
164153 }
165154
166155 @ Test
167156 public void testSetDefaultEmail_expectWelcomeBackPasswordPrompt () {
168157 EmailActivity emailActivity = createActivity (EmailAuthProvider .PROVIDER_ID , false , true );
169-
170158 emailActivity .onExistingEmailUser (new User .Builder (EmailAuthProvider .PROVIDER_ID , TestConstants .EMAIL ).build ());
171-
172159 ShadowActivity .IntentForResult nextIntent =
173- shadowOf (emailActivity ).getNextStartedActivityForResult ();
160+ Shadows . shadowOf (emailActivity ).getNextStartedActivityForResult ();
174161 assertEquals (WelcomeBackPasswordPrompt .class .getName (),
175162 nextIntent .intent .getComponent ().getClassName ());
176-
177163 }
178164
179165 @ Test
180166 public void testSetDefaultEmail_expectRegisterEmailFragment () {
181167 EmailActivity emailActivity = createActivity (EmailAuthProvider .PROVIDER_ID , false , true );
182-
183168 emailActivity .onNewUser (new User .Builder (EmailAuthProvider .PROVIDER_ID , TestConstants .EMAIL ).build ());
184-
185169 shadowOf (Looper .getMainLooper ()).idle ();
186170 RegisterEmailFragment registerEmailFragment = (RegisterEmailFragment ) emailActivity
187171 .getSupportFragmentManager ().findFragmentByTag (RegisterEmailFragment .TAG );
188172 assertThat (registerEmailFragment ).isNotNull ();
189173 }
190174
191-
192175 private EmailActivity createActivity (String providerId ) {
193176 return createActivity (providerId , false , false );
194177 }
@@ -199,7 +182,7 @@ private EmailActivity createActivity(String providerId, boolean emailLinkLinking
199182 TestHelper .getFlowParameters (Collections .singletonList (providerId )));
200183
201184 if (hasDefaultEmail ) {
202- startIntent = EmailActivity .createIntent (
185+ startIntent = EmailActivity .createIntent (
203186 ApplicationProvider .getApplicationContext (),
204187 TestHelper .getFlowParameters (Collections .singletonList (providerId ), false , null , true ));
205188 }
@@ -209,11 +192,11 @@ private EmailActivity createActivity(String providerId, boolean emailLinkLinking
209192 startIntent .putExtra (ExtraConstants .IDP_RESPONSE , buildGoogleIdpResponse ());
210193 }
211194
212- return Robolectric . buildActivity ( EmailActivity . class , startIntent )
213- . create ()
214- . start ()
215- . visible ()
216- .get ();
195+ ActivityController < EmailActivity > controller =
196+ Robolectric . buildActivity ( EmailActivity . class , startIntent );
197+ EmailActivity activity = controller . get ();
198+ activity . setTheme ( R . style . Theme_AppCompat );
199+ return controller . create (). start (). visible () .get ();
217200 }
218201
219202 private IdpResponse buildGoogleIdpResponse () {
0 commit comments