assert first word of document during tests

Author: TomTasche

app/build.gradle

@@ -147,6 +147,7 @@ dependencies {
     androidTestImplementation 'androidx.test:runner:1.6.2'
     androidTestImplementation 'androidx.test.ext:junit:1.2.1'
     androidTestImplementation 'androidx.test.espresso:espresso-intents:3.6.1'
+    androidTestImplementation 'androidx.test.uiautomator:uiautomator:2.3.0'
     // espresso-idling-resource is used in main sourceSet as well. cannot be just androidTestImplementation
     implementation 'androidx.test.espresso:espresso-idling-resource:3.6.1'
     implementation 'androidx.annotation:annotation:1.9.1'

app/src/androidTest/java/at/tomtasche/reader/test/MainActivityTests.java

@@ -48,6 +48,10 @@
 public class MainActivityTests {
     private IdlingResource m_idlingResource;
     private static final Map<String, File> s_testFiles = new ArrayMap<>();
+    private static final String EXPECTED_FIRST_WORD_ODT = "This";
+    private static final String EXPECTED_FIRST_WORD_PDF = "Dummy";
+    private static final String EXPECTED_FIRST_WORD_DOCX = "Table";
+    private static final String EXPECTED_FIRST_WORD_PASSWORD_ODT = "Hallo";
 
     // Yes, this is ActivityTestRule instead of ActivityScenario, because ActivityScenario does not actually work.
     // Issue ID may or may not be added later.
@@ -138,6 +142,7 @@ public void testODT() throws InterruptedException {
         PageView pageView = documentFragment.getPageView();
         Assert.assertNotNull(pageView);
         Assert.assertTrue("ODT should load", waitForPageLoaded(pageView, 10000));
+        assertFirstWord(pageView, EXPECTED_FIRST_WORD_ODT, "ODT");
 
         String fileType = documentFragment.getLastFileType();
         Assert.assertNotNull(fileType);
@@ -154,6 +159,7 @@ public void testPDF() throws InterruptedException {
         PageView pageView = documentFragment.getPageView();
         Assert.assertNotNull(pageView);
         Assert.assertTrue("PDF should load", waitForPageLoaded(pageView, 10000));
+        assertFirstWord(pageView, EXPECTED_FIRST_WORD_PDF, "PDF");
 
         String fileType = documentFragment.getLastFileType();
         Assert.assertNotNull(fileType);
@@ -186,6 +192,7 @@ public void testPasswordProtectedODT() throws InterruptedException {
         Assert.assertNotNull(pageView);
         Assert.assertTrue("Password-protected ODT should load with correct password",
                 waitForPageLoaded(pageView, 10000));
+        assertFirstWord(pageView, EXPECTED_FIRST_WORD_PASSWORD_ODT, "Password-protected ODT");
     }
 
     @Test
@@ -194,10 +201,12 @@ public void testODTEditMode() throws InterruptedException {
         Assert.assertNotNull(testFile);
         MainActivity activity = mainActivityActivityTestRule.getActivity();
         DocumentFragment documentFragment = loadDocument(activity, testFile);
-        enterEditMode(activity, documentFragment);
 
         PageView pageView = documentFragment.getPageView();
         Assert.assertNotNull(pageView);
+        assertFirstWord(pageView, EXPECTED_FIRST_WORD_ODT, "ODT");
+
+        enterEditMode(activity, documentFragment);
         Assert.assertTrue(
                 "ODT should become editable after entering edit mode",
                 waitForEditableState(pageView, true, 10000)
@@ -210,11 +219,12 @@ public void testDOCXEditMode() throws InterruptedException {
         Assert.assertNotNull(testFile);
         MainActivity activity = mainActivityActivityTestRule.getActivity();
         DocumentFragment documentFragment = loadDocument(activity, testFile);
-        enterEditMode(activity, documentFragment);
 
         PageView pageView = documentFragment.getPageView();
         Assert.assertNotNull(pageView);
+        assertFirstWord(pageView, EXPECTED_FIRST_WORD_DOCX, "DOCX");
 
+        enterEditMode(activity, documentFragment);
         Assert.assertTrue(
                 "DOCX should become editable after entering edit mode",
                 waitForEditableState(pageView, true, 10000)
@@ -298,6 +308,37 @@ private boolean waitForPageLoaded(PageView pageView, long timeoutMs) throws Inte
         return false;
     }
 
+    private void assertFirstWord(PageView pageView, String expected, String label) throws InterruptedException {
+        String firstWord = waitForFirstWord(pageView, 10000);
+        Assert.assertEquals(label + " first word mismatch", expected, firstWord);
+    }
+
+    private String waitForFirstWord(PageView pageView, long timeoutMs) throws InterruptedException {
+        long startMs = SystemClock.elapsedRealtime();
+        while (SystemClock.elapsedRealtime() - startMs < timeoutMs) {
+            String firstWord = getFirstWord(pageView);
+            if (!firstWord.isEmpty()) {
+                return firstWord;
+            }
+            SystemClock.sleep(250);
+        }
+        return "";
+    }
+
+    private String getFirstWord(PageView pageView) throws InterruptedException {
+        String result = evaluateJavascript(pageView,
+                "(function(){"
+                        + "var text = document.body ? (document.body.innerText || '') : '';"
+                        + "text = text.replace(/\\s+/g,' ').trim();"
+                        + "if (!text) return '';"
+                        + "return text.split(' ')[0];"
+                        + "})()");
+        if (result == null) {
+            return "";
+        }
+        return result.replace("\"", "").trim();
+    }
+
     private String getPageViewUrl(PageView pageView) throws InterruptedException {
         AtomicReference<String> url = new AtomicReference<>();
         CountDownLatch latch = new CountDownLatch(1);