1 files changed, 111 insertions, 0 deletions
diff --git a/tests/src/org/lineageos/tests/power/unit/PerfomanceManagerTest.java b/tests/src/org/lineageos/tests/power/unit/PerfomanceManagerTest.java
new file mode 100644
index 00000000..4259ef62
--- /dev/null
+++ b/tests/src/org/lineageos/tests/power/unit/PerfomanceManagerTest.java
@@ -0,0 +1,111 @@
+/**
+ * Copyright (c) 2016, The CyanogenMod 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 org.lineageos.tests.power.unit;
+
+import android.test.AndroidTestCase;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import lineageos.app.LineageContextConstants;
+import lineageos.power.IPerformanceManager;
+import lineageos.power.PerformanceManager;
+import lineageos.power.PerformanceProfile;
+
+/**
+ * Code coverage for public facing {@link PerformanceManager} interfaces.
+ * The test below will save and restore the current performance profile to
+ * not impact successive tests.
+ */
+public class PerfomanceManagerTest extends AndroidTestCase {
+    private static final String TAG = PerfomanceManagerTest.class.getSimpleName();
+    private static final int IMPOSSIBLE_POWER_PROFILE = -1;
+    private PerformanceManager mLineagePerformanceManager;
+    private PerformanceProfile mSavedPerfProfile;
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        // Only run this if we support performance abstraction
+        org.junit.Assume.assumeTrue(mContext.getPackageManager().hasSystemFeature(
+                LineageContextConstants.Features.PERFORMANCE));
+        mLineagePerformanceManager = PerformanceManager.getInstance(mContext);
+        // Save the perf profile for later restore.
+        mSavedPerfProfile = mLineagePerformanceManager.getPowerProfile(
+                mLineagePerformanceManager.getPowerProfile());
+    }
+
+    @SmallTest
+    public void testManagerExists() {
+        assertNotNull(mLineagePerformanceManager);
+    }
+
+    @SmallTest
+    public void testManagerServiceIsAvailable() {
+        IPerformanceManager ilineageStatusBarManager = mLineagePerformanceManager.getService();
+        assertNotNull(ilineageStatusBarManager);
+    }
+
+    @SmallTest
+    public void testPowerProfileCantBeSetIfNoneSupported() {
+        // Assert that if we attempt to set a power profile if none supported
+        // then we receive a failed response from the service.
+        if (mLineagePerformanceManager.getNumberOfProfiles() == 0) {
+            for (int powerProfile = 0; powerProfile <
+                    PerformanceManager.POSSIBLE_POWER_PROFILES.length; powerProfile++) {
+                assertFalse(mLineagePerformanceManager.setPowerProfile(powerProfile));
+            }
+        }
+    }
+
+    @SmallTest
+    public void testGetPowerProfile() {
+        assertNotSame(IMPOSSIBLE_POWER_PROFILE, mSavedPerfProfile);
+    }
+
+    @SmallTest
+    public void testSetAndGetPowerProfile() {
+        // Identify what power profiles are supported. The api currently returns
+        // the total number of profiles supported in an ordered manner, thus we can
+        // assume what they are and if we can set everything correctly.
+        // TODO: Don't skip powersave. Skipped due to powersave being ignored while device plugged
+        for (int powerProfile = 1; powerProfile <
+                PerformanceManager.POSSIBLE_POWER_PROFILES.length; powerProfile++) {
+            if (powerProfile < mLineagePerformanceManager.getNumberOfProfiles()) {
+                //It is supported, set it and test if it was set
+                if (mLineagePerformanceManager.getPowerProfile() != powerProfile) {
+                    mLineagePerformanceManager.setPowerProfile(powerProfile);
+                    // Verify that it was set correctly.
+                    assertEquals(powerProfile, mLineagePerformanceManager.getPowerProfile());
+                }
+            } else {
+                assertFalse(mLineagePerformanceManager.setPowerProfile(powerProfile));
+            }
+        }
+    }
+
+    @SmallTest
+    public void testGetPerfProfileHasAppProfiles() {
+        // No application has power save by default
+        assertEquals(false, mLineagePerformanceManager.getProfileHasAppProfiles(
+                PerformanceManager.PROFILE_POWER_SAVE));
+    }
+
+    @Override
+    protected void tearDown() throws Exception {
+        super.tearDown();
+        // Reset
+        mLineagePerformanceManager.setPowerProfile(mSavedPerfProfile.getId());
+    }
+}