[Home] Add lifecycle events and some tests

chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java

Index: chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java
diff --git a/chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java b/chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java
index 6a81592403e5b85097e68ee01ecfc4907f768ebe..c3043f769df7050ca1ec80879a073de5a2132b74 100644
--- a/chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java
+++ b/chrome/test/android/javatests/src/org/chromium/chrome/test/BottomSheetTestCaseBase.java
@@ -20,6 +20,9 @@ import org.chromium.chrome.test.util.browser.RecyclerViewTestUtils;
 @CommandLineFlags.Add({"enable-features=ChromeHome"})
 @Restriction(RESTRICTION_TYPE_PHONE) // ChromeHome is only enabled on phones
 public abstract class BottomSheetTestCaseBase extends ChromeTabbedActivityTestBase {
+    /** A handle to the bottom sheet. */
+    protected BottomSheet mBottomSheet;
+
     private boolean mOldChromeHomeFlagValue;
     @Override
     protected void setUp() throws Exception {
@@ -41,6 +44,8 @@ public abstract class BottomSheetTestCaseBase extends ChromeTabbedActivityTestBa
         });
         RecyclerViewTestUtils.waitForStableRecyclerView(
                 getBottomSheetContent().getScrollingContentView());
+
+        mBottomSheet = getActivity().getBottomSheet();
     }
 
     @Override
@@ -54,6 +59,33 @@ public abstract class BottomSheetTestCaseBase extends ChromeTabbedActivityTestBa
         ChromePreferenceManager.getInstance().setChromeHomeEnabled(mOldChromeHomeFlagValue);
     }
 
+    /**
+     * Set the bottom sheet's state on the UI thread.
+     * @param state The state to set the sheet to.
+     * @param animate If the sheet should animate to the provided state.
+     */
+    protected void setSheetState(final int state, final boolean animate) {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                mBottomSheet.setSheetState(state, animate);
+            }
+        });
+    }
+
+    /**
+     * Set the bottom sheet's offset from the bottom of the screen on the UI thread.
+     * @param offset The offset from the bottom that the sheet should be.
+     */
+    protected void setSheetOffsetFromBottom(final float offset) {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                mBottomSheet.setSheetOffsetFromBottomForTesting(offset);
+            }
+        });
+    }
+
     protected BottomSheetContent getBottomSheetContent() {
         return getActivity().getBottomSheet().getCurrentSheetContent();
     }