Select topics to be able to scroll

Change-Id: Ib7a2a3e87ab3b5d8da6c0fc1b77855f46ef5236a

benchmarks/src/main/java/com/google/samples/apps/nowinandroid/baselineprofile/BaselineProfileGenerator.kt

@@ -21,6 +21,7 @@ import androidx.benchmark.macro.junit4.BaselineProfileRule
 import androidx.test.uiautomator.By
 import com.google.samples.apps.nowinandroid.PACKAGE_NAME
 import com.google.samples.apps.nowinandroid.foryou.forYouScrollFeedDownUp
+import com.google.samples.apps.nowinandroid.foryou.forYouSelectTopics
 import com.google.samples.apps.nowinandroid.foryou.forYouWaitForContent
 import com.google.samples.apps.nowinandroid.interests.interestsScrollPeopleDownUp
 import com.google.samples.apps.nowinandroid.interests.interestsScrollTopicsDownUp
@@ -46,6 +47,7 @@ class BaselineProfileGenerator {
 
             // Scroll the feed critical user journey
             forYouWaitForContent()
+            forYouSelectTopics(true)
             forYouScrollFeedDownUp()
 
             // Navigate to saved screen

benchmarks/src/main/java/com/google/samples/apps/nowinandroid/foryou/ForYouActions.kt

@@ -23,15 +23,52 @@ import androidx.test.uiautomator.untilHasChildren
 import com.google.samples.apps.nowinandroid.flingElementDownUp
 
 fun MacrobenchmarkScope.forYouWaitForContent() {
-    // Wait until content is loaded by checking if authors are loaded
+    // Wait until content is loaded by checking if topics are loaded
     device.wait(Until.gone(By.res("forYou:loadingWheel")), 5_000)
     // Sometimes, the loading wheel is gone, but the content is not loaded yet
-    // So we'll wait here for authors to be sure
+    // So we'll wait here for topics to be sure
     val obj = device.findObject(By.res("forYou:topicSelection"))
     // Timeout here is quite big, because sometimes data loading takes a long time!
     obj.wait(untilHasChildren(), 60_000)
 }
 
+/**
+ * Selects some topics, which will show the feed content for them.
+ * [recheckTopicsIfChecked] Topics may be already checked from the previous iteration.
+ */
+fun MacrobenchmarkScope.forYouSelectTopics(recheckTopicsIfChecked: Boolean = false) {
+    val topics = device.findObject(By.res("forYou:topicSelection"))
+
+    // Set gesture margin from sides not to trigger system gesture navigation
+    val horizontalMargin = 10 * topics.visibleBounds.width() / 100
+    topics.setGestureMargins(horizontalMargin, 0, horizontalMargin, 0)
+
+    // Select some topics to show some feed content
+    repeat(3) { index ->
+        val topic = topics.children[index % topics.childCount]
+
+        when {
+            // Topic wasn't checked, so just do that
+            !topic.isChecked -> {
+                topic.click()
+                device.waitForIdle()
+            }
+
+            // Topic was checked already and we want to recheck it, so just do it twice
+            recheckTopicsIfChecked -> {
+                repeat(2) {
+                    topic.click()
+                    device.waitForIdle()
+                }
+            }
+
+            else -> {
+                // Topic is checked, but we don't recheck it
+            }
+        }
+    }
+}
+
 fun MacrobenchmarkScope.forYouScrollFeedDownUp() {
     val feedList = device.findObject(By.res("forYou:feed"))
     device.flingElementDownUp(feedList)