Add E2E test on Android LeapChat to have two turns of chat

Author: linmx0130

Android/LeapChat/app/src/androidTest/java/ai/liquid/leapchat/MainActivityTest.kt

@@ -0,0 +1,62 @@
+package ai.liquid.leapchat
+
+import androidx.compose.ui.test.ExperimentalTestApi
+import androidx.compose.ui.test.assertIsDisplayed
+import androidx.compose.ui.test.assertTextContains
+import androidx.compose.ui.test.hasTestTag
+import androidx.compose.ui.test.hasText
+import androidx.compose.ui.test.junit4.createAndroidComposeRule
+import androidx.compose.ui.test.onAllNodesWithTag
+import androidx.compose.ui.test.performClick
+import androidx.compose.ui.test.performTextInput
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import org.junit.Rule
+import org.junit.Test
+import org.junit.runner.RunWith
+
+@RunWith(AndroidJUnit4::class)
+class MainActivityE2EAssetTests {
+
+    @get:Rule
+    val composeTestRule = createAndroidComposeRule<MainActivity>()
+
+
+    @OptIn(ExperimentalTestApi::class)
+    @Test
+    fun testEndToEndChat() {
+        val modelLoadingIndicatorMatcher = hasTestTag("ModelLoadingIndicator")
+        val inputBoxMatcher = hasTestTag("InputBox")
+        val sendButtonMatcher = hasText("Send")
+
+        // Wait for the model to be downloaded and loaded
+        composeTestRule.onNode(modelLoadingIndicatorMatcher).assertIsDisplayed()
+        composeTestRule.waitUntilDoesNotExist(
+            modelLoadingIndicatorMatcher,
+            timeoutMillis = MODEL_LOADING_TIMEOUT
+        )
+        composeTestRule.waitUntilAtLeastOneExists(sendButtonMatcher, timeoutMillis = 5000L)
+
+        // Send an input to the model
+        composeTestRule.onNode(inputBoxMatcher)
+            .performTextInput("How many 'r' are there in the word 'strawberry'?")
+        composeTestRule.onNode(sendButtonMatcher).performClick()
+        composeTestRule.waitUntilAtLeastOneExists(
+            hasTestTag("AssistantMessageView"),
+            timeoutMillis = 5000L
+        )
+        composeTestRule.onNode(hasTestTag("AssistantMessageViewText"))
+            .assertTextContains("strawberry", substring = true)
+
+        // Continue the chat with a second prompt
+        composeTestRule.onNode(inputBoxMatcher).performTextInput("What about letter 'a'?")
+        composeTestRule.onNode(sendButtonMatcher).performClick()
+        composeTestRule.waitUntil(timeoutMillis = 5000L) {
+            composeTestRule.onAllNodesWithTag("AssistantMessageView")
+                .fetchSemanticsNodes().size == 2
+        }
+    }
+
+    companion object {
+        const val MODEL_LOADING_TIMEOUT = 5L * 60L * 1000L
+    }
+}

Android/LeapChat/app/src/main/java/ai/liquid/leapchat/MainActivity.kt

@@ -52,6 +52,7 @@ import androidx.compose.ui.Modifier
 import androidx.compose.ui.focus.FocusRequester
 import androidx.compose.ui.focus.focusRequester
 import androidx.compose.ui.platform.LocalContext
+import androidx.compose.ui.platform.testTag
 import androidx.compose.ui.unit.dp
 import androidx.lifecycle.MutableLiveData
 import androidx.lifecycle.lifecycleScope
@@ -152,7 +153,7 @@ class MainActivity : ComponentActivity() {
                                 onValueChange = { userInputFieldText = it },
                                 modifier = Modifier
                                     .padding(4.dp)
-                                    .fillMaxWidth(1.0f),
+                                    .fillMaxWidth(1.0f).testTag("InputBox"),
                                 enabled = !isInGeneration.value
                             )
                             Row(
@@ -469,8 +470,8 @@ class MainActivity : ComponentActivity() {
     }
 
     companion object {
-        const val MODEL_SLUG = "lfm2-1.2b"
-        const val QUANTIZATION_SLUG = "lfm2-1.2b-20250710-8da4w"
+        const val MODEL_SLUG = "lfm2-350m"
+        const val QUANTIZATION_SLUG = "lfm2-350m-20250710-8da4w"
     }
 }
 
@@ -497,7 +498,9 @@ fun ModelLoadingIndicator(
     Box(
         Modifier
             .padding(4.dp)
-            .fillMaxSize(1.0f), contentAlignment = Alignment.Center
+            .fillMaxSize(1.0f)
+            .testTag("ModelLoadingIndicator"),
+        contentAlignment = Alignment.Center
     ) {
         Text(modelLoadingStatusText, style = MaterialTheme.typography.titleSmall)
     }

Android/LeapChat/app/src/main/java/ai/liquid/leapchat/views/AssistantMessage.kt

@@ -17,6 +17,7 @@ import androidx.compose.material3.Text
 import androidx.compose.runtime.Composable
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
+import androidx.compose.ui.platform.testTag
 import androidx.compose.ui.res.painterResource
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
@@ -27,7 +28,7 @@ fun AssistantMessage(
     reasoningText: String?,
 ) {
     val reasoningText = reasoningText?.trim()
-    Row(modifier = Modifier.padding(all = 8.dp).fillMaxWidth(1.0f), horizontalArrangement = Arrangement.Absolute.Left) {
+    Row(modifier = Modifier.padding(all = 8.dp).fillMaxWidth(1.0f).testTag("AssistantMessageView"), horizontalArrangement = Arrangement.Absolute.Left) {
         Image(
             painter = painterResource(R.drawable.smart_toy_outline),
             contentDescription = "Assistant icon",
@@ -43,7 +44,7 @@ fun AssistantMessage(
                 Text(text = reasoningText, style = MaterialTheme.typography.bodySmall, color = MaterialTheme.colorScheme.secondary)
             }
             Spacer(modifier = Modifier.height(4.dp))
-            Text(text = text, style = MaterialTheme.typography.bodyMedium)
+            Text(text = text, style = MaterialTheme.typography.bodyMedium, modifier = Modifier.testTag("AssistantMessageViewText"))
         }
     }
 }