1 files changed, 135 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/tv/BaseMainActivityTestCase.java b/tests/unit/src/com/android/tv/BaseMainActivityTestCase.java
new file mode 100644
index 00000000..99d5ea3e
--- /dev/null
+++ b/tests/unit/src/com/android/tv/BaseMainActivityTestCase.java
@@ -0,0 +1,135 @@
+/*
+ * Copyright (C) 2015 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.android.tv;
+
+import android.content.Context;
+import android.os.SystemClock;
+import android.test.ActivityInstrumentationTestCase2;
+
+import com.android.tv.data.Channel;
+import com.android.tv.data.ChannelDataManager;
+import com.android.tv.testing.ChannelInfo;
+import com.android.tv.testing.testinput.ChannelStateData;
+import com.android.tv.testing.testinput.TestInputControlConnection;
+import com.android.tv.testing.testinput.TestInputControlUtils;
+
+import java.util.List;
+
+/**
+ * Base TestCase for tests that need a {@link MainActivity}.
+ */
+public abstract class BaseMainActivityTestCase
+        extends ActivityInstrumentationTestCase2<MainActivity> {
+    private static final String TAG = "BaseMainActivityTest";
+    private static final int CHANNEL_LOADING_CHECK_INTERVAL_MS = 10;
+
+    protected final TestInputControlConnection mConnection = new TestInputControlConnection();
+
+    protected MainActivity mActivity;
+
+    public BaseMainActivityTestCase(Class<MainActivity> activityClass) {
+        super(activityClass);
+    }
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        // TODO: ensure the SampleInputs are setup.
+        setActivityInitialTouchMode(false);
+        mActivity = getActivity();
+        getInstrumentation().getContext()
+                .bindService(TestInputControlUtils.createIntent(), mConnection,
+                        Context.BIND_AUTO_CREATE);
+    }
+
+    @Override
+    protected void tearDown() throws Exception {
+        if (mConnection.isBound()) {
+            getInstrumentation().getContext().unbindService(mConnection);
+        }
+        super.tearDown();
+    }
+
+    /**
+     * Tune to {@code channel}.
+     *
+     * @param channel the channel to tune to.
+     */
+    protected void tuneToChannel(final Channel channel) {
+        // Run on UI thread so views can be modified
+        try {
+            runTestOnUiThread(new Runnable() {
+                @Override
+                public void run() {
+                    mActivity.tuneToChannel(channel);
+                }
+            });
+        } catch (Throwable throwable) {
+            throw new RuntimeException(throwable);
+        }
+    }
+
+    /**
+     * Sleep until  @{@link ChannelDataManager#isDbLoadFinished()} is true.
+     */
+    protected void waitUntilChannelLoadingFinish() {
+        ChannelDataManager channelDataManager = mActivity.getChannelDataManager();
+        while (!channelDataManager.isDbLoadFinished()) {
+            getInstrumentation().waitForIdleSync();
+            SystemClock.sleep(CHANNEL_LOADING_CHECK_INTERVAL_MS);
+        }
+    }
+
+    /**
+     * Tune to the channel with {@code name}.
+     *
+     * @param name the name of the channel to find.
+     */
+    protected void tuneToChannel(String name) {
+        Channel c = findChannelWithName(name);
+        tuneToChannel(c);
+    }
+
+    /**
+     * Tune to channel.
+     */
+    protected void tuneToChannel(ChannelInfo channel) {
+        tuneToChannel(channel.name);
+    }
+
+    /**
+     * Update the channel state to {@code data} then tune to that channel.
+     *
+     * @param data    the state to update the channel with.
+     * @param channel the channel to tune to
+     */
+    protected void updateThenTune(ChannelStateData data, ChannelInfo channel) {
+        mConnection.updateChannelState(channel, data);
+        tuneToChannel(channel);
+    }
+
+    private Channel findChannelWithName(String displayName) {
+        waitUntilChannelLoadingFinish();
+        List<Channel> channelList = mActivity.getChannelDataManager().getChannelList();
+        for (Channel c : channelList) {
+            if (c.getDisplayName().equals(displayName)) {
+                return c;
+            }
+        }
+        throw new AssertionError("'" + displayName + "' channel not found");
+    }
+
+}