From 05f67e3806657968eb53bcb2270e401b3d9d963e Mon Sep 17 00:00:00 2001
From: Lee Kellogg <lkellogg@google.com>
Date: Tue, 26 Apr 2022 15:23:44 -0400
Subject: [PATCH] Fix the appdistro test app check for update button

---
 .../firebase/appdistribution/testapp/MainActivity.kt | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt b/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
index d9f428f91fb..8165362dacd 100644
--- a/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
+++ b/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
@@ -122,13 +122,15 @@ class MainActivity : AppCompatActivity() {
             executorService.execute {
                 firebaseAppDistribution
                     .checkForNewRelease()
-                    .addOnSuccessListener { release = it }
+                    .addOnSuccessListener {
+                        release = it
+                        setupUI(
+                            isSignedIn = firebaseAppDistribution.isTesterSignedIn,
+                            isUpdateAvailable = release != null,
+                            release = release)
+                    }
                     .addOnFailureListener { failureListener(it) }
             }
-            setupUI(
-                isSignedIn = firebaseAppDistribution.isTesterSignedIn,
-                isUpdateAvailable = release != null,
-                release = release)
         }
 
         signOutButton.setOnClickListener {