1 files changed, 106 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/tv/MainActivityTest.java b/tests/unit/src/com/android/tv/MainActivityTest.java
new file mode 100644
index 00000000..ae789895
--- /dev/null
+++ b/tests/unit/src/com/android/tv/MainActivityTest.java
@@ -0,0 +1,106 @@
+/*
+ * 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.test.suitebuilder.annotation.MediumTest;
+import android.view.View;
+import android.widget.TextView;
+
+import com.android.tv.data.Channel;
+import com.android.tv.testing.testinput.TvTestInputConstants;
+import com.android.tv.ui.ChannelBannerView;
+
+import java.util.List;
+
+/**
+ * Tests for {@link MainActivity}.
+ */
+@MediumTest
+public class MainActivityTest extends BaseMainActivityTestCase {
+
+    public MainActivityTest() {
+        super(MainActivity.class);
+    }
+
+    public void testInitialConditions() {
+        waitUntilChannelLoadingFinish();
+        List<Channel> channelList = mActivity.getChannelDataManager().getChannelList();
+        assertTrue("Expected at least one channel", channelList.size() > 0);
+        assertFalse("PIP disabled", mActivity.isPipEnabled());
+    }
+
+    public void testTuneToChannel() throws Throwable {
+        tuneToChannel(TvTestInputConstants.CH_2);
+        assertChannelBannerShown(true);
+        assertChannelName(TvTestInputConstants.CH_2.name);
+    }
+
+    public void testShowProgramGuide() throws Throwable {
+        tuneToChannel(TvTestInputConstants.CH_2);
+        showProgramGuide();
+        assertChannelBannerShown(false);
+        assertProgramGuide(true);
+    }
+
+    private void showProgramGuide() throws Throwable {
+        // Run on UI thread so views can be modified
+        runTestOnUiThread(new Runnable() {
+            @Override
+            public void run() {
+                mActivity.getOverlayManager().showProgramGuide();
+            }
+        });
+    }
+
+    private void assertChannelName(String displayName) {
+        TextView channelNameView = (TextView) mActivity.findViewById(R.id.channel_name);
+        assertEquals("Channel Name", displayName, channelNameView.getText());
+    }
+
+    private View assertProgramGuide(boolean isShown) {
+        return assertViewIsShown("Program Guide", R.id.program_guide, isShown);
+    }
+
+    private ChannelBannerView assertChannelBannerShown(boolean isShown) {
+        View v = assertExpectedBannerSceneClassShown(ChannelBannerView.class, isShown);
+        return (ChannelBannerView) v;
+    }
+
+    private View assertExpectedBannerSceneClassShown(Class<ChannelBannerView> expectedClass,
+            boolean expectedShown) throws AssertionError {
+        View v = assertViewIsShown(expectedClass.getSimpleName(), R.id.scene_transition_common,
+                expectedShown);
+        if (v != null) {
+            assertEquals(expectedClass, v.getClass());
+        }
+        return v;
+    }
+
+    private View assertViewIsShown(String viewName, int viewId, boolean expected)
+            throws AssertionError {
+        View view = mActivity.findViewById(viewId);
+        if (view == null) {
+            if (expected) {
+                throw new AssertionError("View " + viewName + " not found");
+            } else {
+                return null;
+            }
+        }
+        assertEquals(viewName + " shown", expected, view.isShown());
+        return view;
+    }
+
+}