Stop test

Author: kirklandsign

llm/android/LlamaDemo/app/src/androidTest/java/com/example/executorchllamademo/UIWorkflowTest.java

@@ -356,6 +356,89 @@ public void testStopGeneration() throws Exception {
         }
     }
 
+    /**
+     * Tests that trying to send a new message during generation is not possible:
+     * 1. Load model
+     * 2. Send a message to start generation
+     * 3. Wait for generation to start
+     * 4. Verify input field is cleared
+     * 5. Type new text in input field
+     * 6. Verify send button shows stop icon (not send) - button is enabled for stopping
+     * 7. Stop generation and verify button returns to send mode
+     */
+    @Test
+    public void testSendDuringGeneration() throws Exception {
+        try (ActivityScenario<MainActivity> scenario = ActivityScenario.launch(MainActivity.class)) {
+            // Wait for activity to fully load
+            Thread.sleep(1000);
+
+            // Dismiss the "Please Select a Model" dialog
+            onView(withText(android.R.string.ok)).inRoot(isDialog()).perform(click());
+
+            // --- Load model ---
+            onView(withId(R.id.settings)).perform(click());
+            Thread.sleep(500);
+
+            // Select model
+            onView(withId(R.id.modelImageButton)).perform(click());
+            Thread.sleep(300);
+            onData(hasToString(endsWith(modelFile))).inRoot(isDialog()).perform(click());
+            Thread.sleep(300);
+
+            // Select tokenizer
+            onView(withId(R.id.tokenizerImageButton)).perform(click());
+            Thread.sleep(300);
+            onData(hasToString(endsWith(tokenizerFile))).inRoot(isDialog()).perform(click());
+            Thread.sleep(300);
+
+            // Load model
+            onView(withId(R.id.loadModelButton)).perform(click());
+            onView(withText(android.R.string.yes)).inRoot(isDialog()).perform(click());
+
+            // Wait for model to load
+            boolean modelLoaded = waitForModelLoaded(scenario, 60000);
+            assertTrue("Model should be loaded successfully", modelLoaded);
+
+            // --- Send a message to start generation (use a longer prompt for slower generation) ---
+            onView(withId(R.id.editTextMessage)).perform(typeText("Write a very long detailed story about a brave knight who goes on an adventure"), ViewActions.closeSoftKeyboard());
+            onView(withId(R.id.sendButton)).perform(click());
+
+            // --- Wait for generation to start ---
+            boolean generationStarted = waitForResponseStarted(scenario, 30000);
+            assertTrue("Generation should start", generationStarted);
+
+            // --- Verify input field is cleared after sending ---
+            onView(withId(R.id.editTextMessage)).check(matches(withText("")));
+
+            // --- Check if still generating (button enabled means stop mode) ---
+            AtomicBoolean isStillGenerating = new AtomicBoolean(false);
+            scenario.onActivity(activity -> {
+                ImageButton sendButton = activity.findViewById(R.id.sendButton);
+                if (sendButton != null) {
+                    isStillGenerating.set(sendButton.isEnabled());
+                }
+            });
+
+            // If generation already completed, skip the during-generation checks
+            if (!isStillGenerating.get()) {
+                android.util.Log.i("UIWorkflowTest", "Generation completed quickly, skipping during-generation checks");
+                return;
+            }
+
+            // --- Type new text during generation ---
+            onView(withId(R.id.editTextMessage)).perform(typeText("Another message"), ViewActions.closeSoftKeyboard());
+
+            // --- Stop generation ---
+            onView(withId(R.id.sendButton)).perform(click());
+            Thread.sleep(1000);
+
+            // --- Verify generation stopped and we can now send ---
+            // After stopping, the input still has text, so send button should be enabled
+            onView(withId(R.id.editTextMessage)).check(matches(withText("Another message")));
+            onView(withId(R.id.sendButton)).check(matches(isEnabled()));
+        }
+    }
+
     /**
      * Waits for generation to start by checking for model response text.
      *