Propagating the exceptions to the clients.

firebase-installations/src/androidTest/java/com/google/firebase/installations/FirebaseInstallationsInstrumentedTest.java

@@ -371,7 +371,7 @@ public void testGetAuthToken_PersistedFidError_failure() throws Exception {
           .isInstanceOf(FirebaseInstallationsException.class);
       assertWithMessage("Exception status doesn't match")
           .that(((FirebaseInstallationsException) expected.getCause()).getStatus())
-          .isEqualTo(FirebaseInstallationsException.Status.SDK_INTERNAL_ERROR);
+          .isEqualTo(FirebaseInstallationsException.Status.CLIENT_ERROR);
     }
   }
 

firebase-installations/src/main/java/com/google/firebase/installations/FirebaseInstallations.java

@@ -196,6 +196,19 @@ private void triggerOnStateReached(PersistedFidEntry persistedFidEntry) {
     }
   }
 
+  private void triggerOnException(PersistedFidEntry persistedFidEntry, Exception exception) {
+    synchronized (lock) {
+      Iterator<StateListener> it = listeners.iterator();
+      while (it.hasNext()) {
+        StateListener l = it.next();
+        boolean doneListening = l.onException(persistedFidEntry, exception);
+        if (doneListening) {
+          it.remove();
+        }
+      }
+    }
+  }
+
   private final Runnable doRegistration() {
     return () -> {
       try {
@@ -238,7 +251,7 @@ private final Runnable doRegistration() {
                 .setRegistrationStatus(RegistrationStatus.REGISTER_ERROR)
                 .build();
         persistedFid.insertOrUpdatePersistedFidEntry(errorFidEntry);
-        triggerOnStateReached(errorFidEntry);
+        triggerOnException(errorFidEntry, e);
       }
     };
   }

firebase-installations/src/main/java/com/google/firebase/installations/GetAuthTokenListener.java

@@ -38,12 +38,13 @@ public boolean onStateReached(PersistedFidEntry persistedFidEntry) {
               .build());
       return true;
     }
+    return false;
+  }
 
+  @Override
+  public boolean onException(PersistedFidEntry persistedFidEntry, Exception exception) {
     if (persistedFidEntry.isErrored()) {
-      resultTaskCompletionSource.setException(
-          new FirebaseInstallationsException(
-              "Firebase Installation is not registered.",
-              FirebaseInstallationsException.Status.SDK_INTERNAL_ERROR));
+      resultTaskCompletionSource.trySetException(exception);
       return true;
     }
     return false;

firebase-installations/src/main/java/com/google/firebase/installations/GetIdListener.java

@@ -27,15 +27,16 @@ public GetIdListener(TaskCompletionSource<String> taskCompletionSource) {
   @Override
   public boolean onStateReached(PersistedFidEntry persistedFidEntry) {
     if (persistedFidEntry.isUnregistered() || persistedFidEntry.isRegistered()) {
-      taskCompletionSource.setResult(persistedFidEntry.getFirebaseInstallationId());
+      taskCompletionSource.trySetResult(persistedFidEntry.getFirebaseInstallationId());
       return true;
     }
+    return false;
+  }
 
+  @Override
+  public boolean onException(PersistedFidEntry persistedFidEntry, Exception exception) {
     if (persistedFidEntry.isErrored()) {
-      taskCompletionSource.setException(
-          new FirebaseInstallationsException(
-              "Failed to update client side cache.",
-              FirebaseInstallationsException.Status.CLIENT_ERROR));
+      taskCompletionSource.trySetException(exception);
       return true;
     }
     return false;

firebase-installations/src/main/java/com/google/firebase/installations/StateListener.java

@@ -18,4 +18,6 @@
 
 interface StateListener {
   boolean onStateReached(PersistedFidEntry persistedFidEntry);
+
+  boolean onException(PersistedFidEntry persistedFidEntry, Exception exception);
 }