feat: Added new error codes for IdP management and multitenancy (#458)

* feat: Added new error codes for IdP management and multitenancy

* fix: Updated integration tests

* fix: Renamed helper method

Author: hiranya911

src/main/java/com/google/firebase/auth/AuthErrorCode.java

@@ -26,6 +26,11 @@ public enum AuthErrorCode {
    */
   CERTIFICATE_FETCH_FAILED,
 
+  /**
+   * No IdP configuration found for the given identifier.
+   */
+  CONFIGURATION_NOT_FOUND,
+
   /**
    * A user already exists with the provided email.
    */
@@ -71,8 +76,16 @@ public enum AuthErrorCode {
    */
   REVOKED_SESSION_COOKIE,
 
+  /**
+   * Tenant ID in the JWT does not match.
+   */
   TENANT_ID_MISMATCH,
 
+  /**
+   * No tenant found for the given identifier.
+   */
+  TENANT_NOT_FOUND,
+
   /**
    * A user already exists with the provided UID.
    */

src/main/java/com/google/firebase/auth/internal/AuthErrorHandler.java

@@ -36,6 +36,12 @@ final class AuthErrorHandler extends AbstractHttpErrorHandler<FirebaseAuthExcept
 
   private static final Map<String, AuthError> ERROR_CODES =
       ImmutableMap.<String, AuthError>builder()
+          .put(
+              "CONFIGURATION_NOT_FOUND",
+              new AuthError(
+                  ErrorCode.NOT_FOUND,
+                  "No IdP configuration found corresponding to the provided identifier",
+                  AuthErrorCode.CONFIGURATION_NOT_FOUND))
           .put(
               "DUPLICATE_EMAIL",
               new AuthError(
@@ -67,6 +73,12 @@ final class AuthErrorHandler extends AbstractHttpErrorHandler<FirebaseAuthExcept
                   ErrorCode.ALREADY_EXISTS,
                   "The user with the provided phone number already exists",
                   AuthErrorCode.PHONE_NUMBER_ALREADY_EXISTS))
+          .put(
+              "TENANT_NOT_FOUND",
+              new AuthError(
+                  ErrorCode.NOT_FOUND,
+                  "No tenant found for the given identifier",
+                  AuthErrorCode.TENANT_NOT_FOUND))
           .put(
               "UNAUTHORIZED_DOMAIN",
               new AuthError(

src/main/java/com/google/firebase/auth/internal/AuthHttpClient.java

@@ -40,10 +40,12 @@ public final class AuthHttpClient {
   private static final String CLIENT_VERSION = "Java/Admin/" + SdkUtils.getVersion();
 
   private final ErrorHandlingHttpClient<FirebaseAuthException> httpClient;
+  private final JsonFactory jsonFactory;
 
   public AuthHttpClient(JsonFactory jsonFactory, HttpRequestFactory requestFactory) {
     AuthErrorHandler authErrorHandler = new AuthErrorHandler(jsonFactory);
     this.httpClient = new ErrorHandlingHttpClient<>(requestFactory, jsonFactory, authErrorHandler);
+    this.jsonFactory = jsonFactory;
   }
 
   public static Set<String> generateMask(Map<String, Object> properties) {

src/main/java/com/google/firebase/auth/multitenancy/FirebaseTenantClient.java

@@ -46,15 +46,19 @@ final class FirebaseTenantClient {
   private final AuthHttpClient httpClient;
 
   FirebaseTenantClient(FirebaseApp app) {
-    checkNotNull(app, "FirebaseApp must not be null");
-    String projectId = ImplFirebaseTrampolines.getProjectId(app);
+    this(
+        ImplFirebaseTrampolines.getProjectId(checkNotNull(app)),
+        app.getOptions().getJsonFactory(),
+        ApiClientUtils.newAuthorizedRequestFactory(app));
+  }
+
+  FirebaseTenantClient(
+      String projectId, JsonFactory jsonFactory, HttpRequestFactory requestFactory) {
     checkArgument(!Strings.isNullOrEmpty(projectId),
         "Project ID is required to access the auth service. Use a service account credential or "
             + "set the project ID explicitly via FirebaseOptions. Alternatively you can also "
             + "set the project ID via the GOOGLE_CLOUD_PROJECT environment variable.");
     this.tenantMgtBaseUrl = String.format(ID_TOOLKIT_URL, "v2", projectId);
-    JsonFactory jsonFactory = app.getOptions().getJsonFactory();
-    HttpRequestFactory requestFactory = ApiClientUtils.newAuthorizedRequestFactory(app);
     this.httpClient = new AuthHttpClient(jsonFactory, requestFactory);
   }
 

src/main/java/com/google/firebase/auth/multitenancy/TenantManager.java

@@ -54,8 +54,13 @@ public final class TenantManager {
    * @hide
    */
   public TenantManager(FirebaseApp firebaseApp) {
-    this.firebaseApp = firebaseApp;
-    this.tenantClient = new FirebaseTenantClient(firebaseApp);
+    this(firebaseApp, new FirebaseTenantClient(firebaseApp));
+  }
+
+  @VisibleForTesting
+  TenantManager(FirebaseApp firebaseApp, FirebaseTenantClient tenantClient) {
+    this.firebaseApp = checkNotNull(firebaseApp);
+    this.tenantClient = checkNotNull(tenantClient);
     this.tenantAwareAuths = new HashMap<>();
   }