tools/android/webgpu/src/androidTest/java/androidx/webgpu/ExtensionsTest.kt

/*
 * Copyright 2026 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 androidx.webgpu

import android.graphics.Color
import android.os.Build
import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.filters.SdkSuppress
import androidx.webgpu.GPUColor
import androidx.webgpu.helper.toGPUColor
import org.junit.Assert.assertEquals
import org.junit.Test
import org.junit.runner.RunWith

@RunWith(AndroidJUnit4::class)
class ExtensionsTest {

    private val delta = 1e-6

    @Test
    fun testIntToGpuColor() {
        // Create an integer color for opaque red
        val redInt = Color.argb(255, 255, 0, 0)
        val expectedColor = GPUColor(1.0, 0.0, 0.0, 1.0)
        val actualColor = redInt.toGPUColor()

        assertEquals(expectedColor.r, actualColor.r, delta)
        assertEquals(expectedColor.g, actualColor.g, delta)
        assertEquals(expectedColor.b, actualColor.b, delta)
        assertEquals(expectedColor.a, actualColor.a, delta)
    }

    @Test
    fun testIntToGpuColorWithAlpha() {
        // Create an integer color for semi-transparent green.
        val alphaInt = 128
        val greenInt = Color.argb(alphaInt, 0, 255, 0)
        val expectedColor = GPUColor(0.0, 1.0, 0.0, alphaInt / 255.0)
        val actualColor = greenInt.toGPUColor()

        assertEquals(expectedColor.r, actualColor.r, delta)
        assertEquals(expectedColor.g, actualColor.g, delta)
        assertEquals(expectedColor.b, actualColor.b, delta)
        assertEquals(expectedColor.a, actualColor.a, delta)
    }

    @Test
    @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O)
    fun testLongToGpuColor() {
        // Create a long color for opaque blue
        val blueLong = Color.pack(0.0f, 0.0f, 1.0f)
        val color = Color.valueOf(blueLong)

        val expectedR = color.red().toDouble()
        val expectedG = color.green().toDouble()
        val expectedB = color.blue().toDouble()
        val expectedA = color.alpha().toDouble()

        val actualColor = blueLong.toGPUColor()

        assertEquals(expectedR, actualColor.r, delta)
        assertEquals(expectedG, actualColor.g, delta)
        assertEquals(expectedB, actualColor.b, delta)
        assertEquals(expectedA, actualColor.a, delta)
    }
}