[Fix] debug testResumption (#8082)

Author: milaGGL

ai-logic/firebase-ai/firebase-ai.gradle.kts

@@ -26,7 +26,7 @@ plugins {
 }
 
 firebaseLibrary {
-  testLab.enabled = false
+  testLab.enabled = true
   publishJavadoc = true
   releaseNotes {
     name.set("{{firebase_ai_logic}}")

ai-logic/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/LiveSessionTests.kt

@@ -279,6 +279,7 @@ class LiveSessionTests {
     val session = liveModel.connect(SessionResumptionConfig())
     session.send("My favorite color is blue. Remember that.", true)
     var lastResumptionUpdate: LiveSessionResumptionUpdate? = null
+    var gotTurnComplete = false
     withTimeout(30.seconds) {
       session
         .receive()
@@ -287,10 +288,10 @@ class LiveSessionTests {
             lastResumptionUpdate = it
           }
           if (it is LiveServerContent && it.turnComplete) {
-            false
-          } else {
-            true
+            gotTurnComplete = true
           }
+          // Stop when we've seen a turn complete and we have a new handle
+          !(gotTurnComplete && lastResumptionUpdate?.newHandle != null)
         }
         .collect {}
     }