commit a038396830964567cdcc7cc4538e67aa0dd81a84
author Kweku Adams <kwekua@google.com> Mon Jul 19 11:24:29 2021 -0700
committer Kweku Adams <kwekua@google.com> Mon Jul 19 11:24:29 2021 -0700
tree 479cf4210f607770bc218f1de86a3597d4db965f
parent 1f023f59f33a90e7a673bd467901e1c9d1f50239

Add test for forEach.

Bug: 194098491
Test: atest CtsUtilTestCases:ArraySetTest
Change-Id: I8328f2a4e954756f9df1e40947d915600aaf2a71

tests/tests/util/src/android/util/cts/ArraySetTest.java

diff --git a/tests/tests/util/src/android/util/cts/ArraySetTest.java b/tests/tests/util/src/android/util/cts/ArraySetTest.java
index 70d307a..42309e3 100644
--- a/tests/tests/util/src/android/util/cts/ArraySetTest.java
+++ b/tests/tests/util/src/android/util/cts/ArraySetTest.java
@@ -412,6 +412,20 @@
     }
 
     @Test
+    public void testForEach() {
+        ArraySet<Integer> set = new ArraySet<>();
+
+        for (int i = 0; i < 50; ++i) {
+            set.add(i * 10);
+        }
+
+        // Make sure forEach goes through all of the elements.
+        HashSet<Integer> seen = new HashSet<>();
+        set.forEach(seen::add);
+        compareSets(seen, set);
+    }
+
+    @Test
     public void testIndexOf() {
         ArraySet<Integer> set = new ArraySet<>();