diff options
| -rw-r--r-- | tests/lib/src/com/android/testutils/NetworkStatsUtils.kt | 39 | ||||
| -rw-r--r-- | tests/unit/src/android/net/testutils/NetworkStatsUtilsTest.kt | 58 |
2 files changed, 97 insertions, 0 deletions
diff --git a/tests/lib/src/com/android/testutils/NetworkStatsUtils.kt b/tests/lib/src/com/android/testutils/NetworkStatsUtils.kt new file mode 100644 index 0000000..c82011c --- /dev/null +++ b/tests/lib/src/com/android/testutils/NetworkStatsUtils.kt @@ -0,0 +1,39 @@ +/* + * Copyright (C) 2020 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.testutils + +import android.net.NetworkStats + +fun orderInsensitiveEquals( + leftStats: NetworkStats, + rightStats: NetworkStats +): Boolean { + if (leftStats == rightStats) return true + if (leftStats.getElapsedRealtime() != rightStats.getElapsedRealtime() || + leftStats.size() != rightStats.size()) return false + val left = NetworkStats.Entry() + val right = NetworkStats.Entry() + // Order insensitive compare. + for (i in 0 until leftStats.size()) { + leftStats.getValues(i, left) + val j: Int = rightStats.findIndexHinted(left.iface, left.uid, left.set, left.tag, + left.metered, left.roaming, left.defaultNetwork, i) + rightStats.getValues(j, right) + if (left != right) return false + } + return true +} diff --git a/tests/unit/src/android/net/testutils/NetworkStatsUtilsTest.kt b/tests/unit/src/android/net/testutils/NetworkStatsUtilsTest.kt new file mode 100644 index 0000000..08788e0 --- /dev/null +++ b/tests/unit/src/android/net/testutils/NetworkStatsUtilsTest.kt @@ -0,0 +1,58 @@ +/* + * Copyright (C) 2020 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.testutils + +import android.net.NetworkStats +import android.net.NetworkStats.SET_DEFAULT +import android.net.NetworkStats.TAG_NONE +import com.android.testutils.orderInsensitiveEquals +import org.junit.Test +import org.junit.runner.RunWith +import org.junit.runners.JUnit4 +import kotlin.test.assertFalse +import kotlin.test.assertTrue + +private const val TEST_IFACE = "test0" +private const val TEST_IFACE2 = "test2" +private const val TEST_START = 1194220800000L + +@RunWith(JUnit4::class) +class NetworkStatsUtilsTest { + @Test + fun testOrderInsensitiveEquals() { + val testEntry = arrayOf( + NetworkStats.Entry(TEST_IFACE, 100, SET_DEFAULT, TAG_NONE, 128L, 8L, 0L, 2L, 20L), + NetworkStats.Entry(TEST_IFACE2, 100, SET_DEFAULT, TAG_NONE, 512L, 32L, 0L, 0L, 0L)) + + // Verify equals of empty stats regardless of initial capacity. + val red = NetworkStats(TEST_START, 0) + val blue = NetworkStats(TEST_START, 1) + assertTrue(orderInsensitiveEquals(red, blue)) + assertTrue(orderInsensitiveEquals(blue, red)) + + // Verify not equal. + red.combineValues(testEntry[1]) + blue.combineValues(testEntry[0]).combineValues(testEntry[1]) + assertFalse(orderInsensitiveEquals(red, blue)) + assertFalse(orderInsensitiveEquals(blue, red)) + + // Verify equals even if the order of entries are not the same. + red.combineValues(testEntry[0]) + assertTrue(orderInsensitiveEquals(red, blue)) + assertTrue(orderInsensitiveEquals(blue, red)) + } +}
\ No newline at end of file |