fix: Re-add test dependencies and revert to mockito-kotlin syntax

Author: google-labs-jules[bot]

firebase-ai/firebase-ai.gradle.kts

@@ -123,6 +123,8 @@ dependencies {
   testImplementation(libs.robolectric)
   testImplementation(libs.truth)
   testImplementation(libs.mockito.core)
+  testImplementation(libs.mockito.kotlin)
+  testImplementation(libs.ktor.client.mock)
 
   androidTestImplementation(libs.androidx.espresso.core)
   androidTestImplementation(libs.androidx.test.junit)

firebase-ai/src/main/kotlin/com/google/firebase/ai/type/LiveSession.kt

@@ -17,15 +17,15 @@
 package com.google.firebase.ai.type
 
 import android.Manifest.permission.RECORD_AUDIO
-import android.content.pm.PackageManager
 import android.media.AudioFormat
 import android.media.AudioTrack
 import android.util.Log
 import androidx.annotation.RequiresPermission
-import com.google.firebase.FirebaseApp
 import com.google.firebase.ai.common.JSON
 import com.google.firebase.ai.common.util.CancelledCoroutineScope
 import com.google.firebase.ai.common.util.accumulateUntil
+import com.google.firebase.FirebaseApp
+import android.content.pm.PackageManager
 import com.google.firebase.ai.common.util.childJob
 import com.google.firebase.annotations.concurrent.Blocking
 import io.ktor.client.plugins.websocket.ClientWebSocketSession
@@ -97,9 +97,8 @@ internal constructor(
     functionCallHandler: ((FunctionCallPart) -> FunctionResponsePart)? = null
   ) {
     val context = firebaseApp.applicationContext
-    if (
-      context.checkSelfPermission(RECORD_AUDIO) !=
-        android.content.pm.PackageManager.PERMISSION_GRANTED
+    if (context.checkSelfPermission(RECORD_AUDIO) !=
+      android.content.pm.PackageManager.PERMISSION_GRANTED
     ) {
       throw PermissionMissingException("Missing RECORD_AUDIO permission.")
     }

firebase-ai/src/test/java/com/google/firebase/ai/type/LiveSessionTest.kt

@@ -5,27 +5,29 @@ import android.content.pm.PackageManager
 import com.google.firebase.FirebaseApp
 import com.google.firebase.ai.util.doBlocking
 import io.kotest.assertions.throwables.shouldThrow
-import io.ktor.client.plugins.websocket.testing.EmptyWebSockets
+import io.ktor.client.plugins.websocket.testing.*
 import kotlinx.coroutines.Dispatchers
 import org.junit.Test
-import org.mockito.Mockito
+import org.mockito.kotlin.*
 
 class LiveSessionTest {
 
-  @Test
-  fun `startAudioConversation without permission throws exception`() = doBlocking {
-    val mockContext = Mockito.mock(Context::class.java)
-    Mockito.`when`(mockContext.checkSelfPermission(android.Manifest.permission.RECORD_AUDIO))
-      .thenReturn(PackageManager.PERMISSION_DENIED)
-    val mockFirebaseApp = Mockito.mock(FirebaseApp::class.java)
-    Mockito.`when`(mockFirebaseApp.applicationContext).thenReturn(mockContext)
-    val session =
-      LiveSession(
-        session = EmptyWebSockets.client.session,
-        blockingDispatcher = Dispatchers.IO,
-        firebaseApp = mockFirebaseApp
-      )
+    @Test
+    fun `startAudioConversation without permission throws exception`() = doBlocking {
+        val mockContext = mock<Context> {
+            on { checkSelfPermission(android.Manifest.permission.RECORD_AUDIO) } doReturn PackageManager.PERMISSION_DENIED
+        }
+        val mockFirebaseApp = mock<FirebaseApp> {
+            on { applicationContext } doReturn mockContext
+        }
+        val session = LiveSession(
+            session = EmptyWebSockets.client.session,
+            blockingDispatcher = Dispatchers.IO,
+            firebaseApp = mockFirebaseApp
+        )
 
-    shouldThrow<PermissionMissingException> { session.startAudioConversation() }
-  }
+        shouldThrow<PermissionMissingException> {
+            session.startAudioConversation()
+        }
+    }
 }