tests/net/common/java/android/net/metrics/ValidationProbeEventTest.kt

/*
 * Copyright (C) 2019 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 android.net.metrics

import android.os.Parcelable
import androidx.test.filters.SmallTest
import androidx.test.runner.AndroidJUnit4
import com.android.internal.util.ParcelableTestUtil
import com.android.internal.util.TestUtils
import java.lang.reflect.Modifier
import org.junit.Assert.assertEquals
import org.junit.Assert.assertTrue
import org.junit.Test
import org.junit.runner.RunWith

private const val FIRST_VALIDATION: Int = 1 shl 8
private const val REVALIDATION: Int = 2 shl 8

@RunWith(AndroidJUnit4::class)
@SmallTest
class ValidationProbeEventTest {
    private fun <T: Parcelable> testParcel(obj: T, fieldCount: Int) {
        ParcelableTestUtil.assertFieldCountEquals(fieldCount, obj::class.java)
        TestUtils.assertParcelingIsLossless(obj)
    }

    private infix fun Int.hasType(type: Int) = (type and this) == type

    @Test
    fun testBuilderAndParcel() {
        var validationProbeEvent = ValidationProbeEvent.Builder()
                .setProbeType(ValidationProbeEvent.PROBE_DNS, false).build()

        assertTrue(validationProbeEvent.probeType hasType REVALIDATION)

        validationProbeEvent = ValidationProbeEvent.Builder()
                .setDurationMs(Long.MAX_VALUE)
                .setProbeType(ValidationProbeEvent.PROBE_DNS, true)
                .setReturnCode(ValidationProbeEvent.DNS_SUCCESS)
                .build()

        assertEquals(Long.MAX_VALUE, validationProbeEvent.durationMs)
        assertTrue(validationProbeEvent.probeType hasType ValidationProbeEvent.PROBE_DNS)
        assertTrue(validationProbeEvent.probeType hasType FIRST_VALIDATION)
        assertEquals(ValidationProbeEvent.DNS_SUCCESS, validationProbeEvent.returnCode)

        testParcel(validationProbeEvent, 3)
    }

    @Test
    fun testGetProbeName() {
        val probeFields = ValidationProbeEvent::class.java.declaredFields.filter {
            it.type == Int::class.javaPrimitiveType
              && Modifier.isPublic(it.modifiers) && Modifier.isStatic(it.modifiers)
              && it.name.contains("PROBE")
        }

        probeFields.forEach {
            val intValue = it.getInt(null)
            val stringValue = ValidationProbeEvent.getProbeName(intValue)
            assertEquals(it.name, stringValue)
        }

    }
}