diff --git a/staticlibs/framework/com/android/net/module/util/CollectionUtils.java b/staticlibs/framework/com/android/net/module/util/CollectionUtils.java index 74f738dafd..cb1e3e7435 100644 --- a/staticlibs/framework/com/android/net/module/util/CollectionUtils.java +++ b/staticlibs/framework/com/android/net/module/util/CollectionUtils.java @@ -65,6 +65,28 @@ public final class CollectionUtils { return array; } + /** + * @return True if all elements satisfy the predicate, false otherwise. + * Note that means this always returns true for empty collections. + */ + public static boolean all(@NonNull Collection elem, @NonNull Predicate predicate) { + for (final T e : elem) { + if (!predicate.test(e)) return false; + } + return true; + + } + /** + * @return True if any element satisfies the predicate, false otherwise. + * Note that means this always returns false for empty collections. + */ + public static boolean any(@NonNull Collection elem, @NonNull Predicate predicate) { + for (final T e : elem) { + if (predicate.test(e)) return true; + } + return false; + } + /** * @return True if there exists at least one element in the sparse array for which * condition {@code predicate} diff --git a/staticlibs/tests/unit/src/com/android/net/module/util/CollectionUtilsTest.kt b/staticlibs/tests/unit/src/com/android/net/module/util/CollectionUtilsTest.kt new file mode 100644 index 0000000000..00077420de --- /dev/null +++ b/staticlibs/tests/unit/src/com/android/net/module/util/CollectionUtilsTest.kt @@ -0,0 +1,54 @@ +/* + * Copyright (C) 2021 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.net.module.util + +import androidx.test.filters.SmallTest +import androidx.test.runner.AndroidJUnit4 +import org.junit.Test +import org.junit.runner.RunWith +import kotlin.test.assertFalse +import kotlin.test.assertTrue + +@RunWith(AndroidJUnit4::class) +@SmallTest +class CollectionUtilsTest { + @Test + fun testAny() { + assertTrue(CollectionUtils.any(listOf("A", "B", "C", "D", "E")) { it == "E" }) + assertFalse(CollectionUtils.any(listOf("A", "B", "C", "D", "E")) { it == "F" }) + assertTrue(CollectionUtils.any(listOf("AA", "BBB")) { it.length >= 3 }) + assertFalse(CollectionUtils.any(listOf("A", "BB", "CCC")) { it.length >= 4 }) + assertFalse(CollectionUtils.any(listOf("A", "BB", "CCC")) { it.length < 0 }) + assertFalse(CollectionUtils.any(listOf()) { true }) + assertFalse(CollectionUtils.any(listOf()) { false }) + assertTrue(CollectionUtils.any(listOf("A")) { true }) + assertFalse(CollectionUtils.any(listOf("A")) { false }) + } + + @Test + fun testAll() { + assertFalse(CollectionUtils.all(listOf("A", "B", "C", "D", "E")) { it != "E" }) + assertTrue(CollectionUtils.all(listOf("A", "B", "C", "D", "E")) { it != "F" }) + assertFalse(CollectionUtils.all(listOf("A", "BB", "CCC")) { it.length > 2 }) + assertTrue(CollectionUtils.all(listOf("A", "BB", "CCC")) { it.length >= 1 }) + assertTrue(CollectionUtils.all(listOf("A", "BB", "CCC")) { it.length < 4 }) + assertTrue(CollectionUtils.all(listOf()) { true }) + assertTrue(CollectionUtils.all(listOf()) { false }) + assertTrue(CollectionUtils.all(listOf(1)) { true }) + assertFalse(CollectionUtils.all(listOf(1)) { false }) + } +}