1 files changed, 123 insertions, 0 deletions
diff --git a/tests/src/org/lineageos/tests/util/ColorUtilTest.java b/tests/src/org/lineageos/tests/util/ColorUtilTest.java
new file mode 100644
index 00000000..9ce91f96
--- /dev/null
+++ b/tests/src/org/lineageos/tests/util/ColorUtilTest.java
@@ -0,0 +1,123 @@
+/**
+ * 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.util;
+
+import android.graphics.Bitmap;
+import android.graphics.Canvas;
+import android.graphics.Color;
+import android.graphics.Paint;
+import android.graphics.drawable.BitmapDrawable;
+import android.graphics.drawable.ColorDrawable;
+import android.test.AndroidTestCase;
+import lineageos.util.ColorUtils;
+import org.mockito.Mockito;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+
+public class ColorUtilTest extends AndroidTestCase {
+    private ColorUtils mColorUtils;
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        mColorUtils = new ColorUtils();
+    }
+
+    public void testDropAlpha() {
+        // With pre-existing alpha
+        int color = 0xBBCCDD;
+        int alpha = 0xAA000000;
+        int result = mColorUtils.dropAlpha(color | alpha);
+        // Assert alpha was removed
+        assertEquals(Color.alpha(result), 0);
+        // Ensure color is preserved
+        assertEquals(result & color, color);
+
+        // Without pre-existing alpha
+        color = Color.argb(0, 100, 200 ,300);
+        result = mColorUtils.dropAlpha(color);
+        // Assert alpha was removed
+        assertEquals(Color.alpha(result), 0);
+        // Ensure color is preserved
+        assertEquals(result, color);
+    }
+
+    public void testGenerateAlertColorFromDrawable() {
+        // Test null drawable
+        int color = mColorUtils.generateAlertColorFromDrawable(null);
+        assertEquals(color, Color.BLACK);
+
+        Bitmap bitmap = Bitmap.createBitmap(10, 10, Bitmap.Config.ARGB_8888);
+        BitmapDrawable bitmapDrawable = new BitmapDrawable(bitmap);
+        bitmapDrawable.setBounds(0, 0, 10, 10);
+        Canvas canvas = new Canvas(bitmap);
+        canvas.drawColor(Color.RED);
+
+        // Test fully red bitmap
+        color = mColorUtils.generateAlertColorFromDrawable(bitmapDrawable);
+        assertEquals(color, Color.RED);
+        color = mColorUtils.generateAlertColorFromDrawable(
+                getColorDrawableFromBitmapDrawable(bitmapDrawable));
+        assertEquals(color, Color.RED);
+
+        // Test blue/red bitmap with blue dominating
+        Paint p = new Paint();
+        p.setColor(Color.BLUE);
+        p.setStyle(Paint.Style.FILL_AND_STROKE);
+        canvas.drawRect(0, 0, 8, 8, p);
+        color = mColorUtils.generateAlertColorFromDrawable(bitmapDrawable);
+        assertEquals(color, Color.BLUE);
+        color = mColorUtils.generateAlertColorFromDrawable(
+                getColorDrawableFromBitmapDrawable(bitmapDrawable));
+        assertEquals(color, Color.BLUE);
+
+        // Test large white + small blue scenario
+        canvas.drawColor(Color.WHITE);
+        canvas.drawRect(0, 0, 2, 2, p);
+        color = mColorUtils.generateAlertColorFromDrawable(bitmapDrawable);
+        assertEquals(color, Color.BLUE);
+        color = mColorUtils.generateAlertColorFromDrawable(
+                getColorDrawableFromBitmapDrawable(bitmapDrawable));
+        assertEquals(color, Color.BLUE);
+
+        // Test large white + small black scenario
+        canvas.drawColor(Color.WHITE);
+        p.setColor(Color.BLACK);
+        canvas.drawRect(0, 0, 2, 2, p);
+        color = mColorUtils.generateAlertColorFromDrawable(bitmapDrawable);
+        assertEquals(color, Color.WHITE);
+        color = mColorUtils.generateAlertColorFromDrawable(
+                getColorDrawableFromBitmapDrawable(bitmapDrawable));
+        assertEquals(color, Color.WHITE);
+
+        assertEquals(bitmap.isRecycled(), false);
+        bitmap.recycle();
+    }
+
+    private ColorDrawable getColorDrawableFromBitmapDrawable(final BitmapDrawable bitmapDrawable) {
+        ColorDrawable colorDrawable = Mockito.mock(ColorDrawable.class);
+        Mockito.doAnswer(new Answer() {
+            @Override
+            public Object answer(InvocationOnMock invocation) throws Throwable {
+                Canvas canvas = (Canvas) invocation.getArguments()[0];
+                bitmapDrawable.draw(canvas);
+                return null;
+            }
+        }).when(colorDrawable).draw(Mockito.any(Canvas.class));
+        return colorDrawable;
+    }
+}