Trade testing hacks

Change-Id: I4b4c8aa0c26738bdaa74dac846194c59744ca73a

Author: samtstern

auth/src/main/java/com/firebase/ui/auth/AuthUI.java

@@ -26,6 +26,7 @@
 import android.support.annotation.RestrictTo;
 import android.support.annotation.StringDef;
 import android.support.annotation.StyleRes;
+import android.support.annotation.VisibleForTesting;
 import android.text.TextUtils;
 import android.util.Log;
 
@@ -90,6 +91,9 @@
  */
 public final class AuthUI {
 
+    @VisibleForTesting
+    protected static FirebaseAuth sDefaultAuth;
+
     @RestrictTo(RestrictTo.Scope.LIBRARY_GROUP)
     public static final String TAG = "AuthUI";
 
@@ -142,7 +146,13 @@ public final class AuthUI {
 
     private AuthUI(FirebaseApp app) {
         mApp = app;
-        mAuth = FirebaseAuth.getInstance(mApp);
+
+        // TODO: This is a bad testing hack
+        if (sDefaultAuth != null) {
+            mAuth = sDefaultAuth;
+        } else {
+            mAuth = FirebaseAuth.getInstance(mApp);
+        }
 
         try {
             mAuth.setFirebaseUIVersion(BuildConfig.VERSION_NAME);

auth/src/test/java/com/firebase/ui/auth/AuthUITest.java

@@ -22,6 +22,7 @@
 import com.firebase.ui.auth.util.ExtraConstants;
 import com.google.firebase.FirebaseApp;
 import com.google.firebase.auth.EmailAuthProvider;
+import com.google.firebase.auth.FirebaseAuth;
 
 import org.junit.Before;
 import org.junit.Test;
@@ -31,12 +32,14 @@
 import java.util.Arrays;
 
 import static junit.framework.Assert.assertEquals;
+import static org.mockito.Mockito.mock;
 
 @RunWith(RobolectricTestRunner.class)
 public class AuthUITest {
     @Before
     public void setUp() {
         TestHelper.initialize();
+        AuthUI.sDefaultAuth = mock(FirebaseAuth.class);
     }
 
     @Test

auth/src/test/java/com/firebase/ui/auth/testhelpers/TestHelper.java

@@ -30,12 +30,10 @@
 import com.firebase.ui.auth.util.ExtraConstants;
 import com.firebase.ui.auth.util.data.ProviderUtils;
 import com.google.android.gms.auth.api.credentials.Credential;
-import com.google.android.gms.tasks.Tasks;
 import com.google.firebase.FirebaseApp;
 import com.google.firebase.FirebaseOptions;
 import com.google.firebase.auth.EmailAuthProvider;
 import com.google.firebase.auth.FacebookAuthProvider;
-import com.google.firebase.auth.FirebaseAuth;
 import com.google.firebase.auth.FirebaseUser;
 import com.google.firebase.auth.GoogleAuthProvider;
 import com.google.firebase.auth.PhoneAuthProvider;
@@ -46,16 +44,11 @@
 import org.robolectric.Shadows;
 import org.robolectric.shadows.ShadowActivity;
 
-import java.lang.reflect.Field;
-import java.lang.reflect.ParameterizedType;
-import java.lang.reflect.Type;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
-import java.util.Map;
 
 import static junit.framework.Assert.assertEquals;
-import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.spy;
 import static org.mockito.Mockito.when;
@@ -67,8 +60,7 @@ public class TestHelper {
     public static void initialize() {
         spyContextAndResources();
         AuthUI.setApplicationContext(RuntimeEnvironment.application);
-        FirebaseApp app = initializeApp(RuntimeEnvironment.application);
-        injectMockFirebaseAuth(app);
+        initializeApp(RuntimeEnvironment.application);
         initializeProviders();
     }
 
@@ -94,48 +86,6 @@ private static FirebaseApp initializeApp(Context context) {
         }
     }
 
-    /**
-     * This method finds the map of FirebaseAuth instances and injects of a mock instance associated
-     * with the given FirebaseApp for testing purposes.
-     */
-    private static void injectMockFirebaseAuth(FirebaseApp app) {
-        for (Field field : FirebaseAuth.class.getDeclaredFields()) {
-            field.setAccessible(true);
-
-            Object o;
-            try {
-                o = field.get(null);
-            } catch (IllegalAccessException e) {
-                throw new IllegalStateException(e);
-            } catch (NullPointerException e) {
-                continue;
-            }
-
-            Type genericType = field.getGenericType();
-            if (o instanceof Map && genericType instanceof ParameterizedType) {
-                Type[] parameterTypes = ((ParameterizedType) genericType).getActualTypeArguments();
-                if (parameterTypes.length != 2 || parameterTypes[0] != String.class
-                        || parameterTypes[1] != FirebaseAuth.class) {
-                    continue;
-                }
-
-                //noinspection unchecked
-                Map<String, FirebaseAuth> instances = (Map<String, FirebaseAuth>) o;
-
-                FirebaseAuth.getInstance(app);
-                for (String id : instances.keySet()) {
-                    instances.put(id, mock(FirebaseAuth.class));
-                }
-
-                break;
-            }
-        }
-
-        FirebaseAuth mockAuth = FirebaseAuth.getInstance(app);
-        when(mockAuth.setFirebaseUIVersion(anyString()))
-                .thenReturn(Tasks.<Void>forResult(null));
-    }
-
     private static void initializeProviders() {
         Context context = RuntimeEnvironment.application;
         when(context.getString(R.string.default_web_client_id)).thenReturn("abc");