diff --git a/core-java-modules/core-java-collections-set-2/src/test/java/com/baeldung/firstitemfromset/GetFirstItemFromSetUnitTest.java b/core-java-modules/core-java-collections-set-2/src/test/java/com/baeldung/firstitemfromset/GetFirstItemFromSetUnitTest.java new file mode 100644 index 0000000000..ec81c17f93 --- /dev/null +++ b/core-java-modules/core-java-collections-set-2/src/test/java/com/baeldung/firstitemfromset/GetFirstItemFromSetUnitTest.java @@ -0,0 +1,35 @@ +package com.baeldung.firstitemfromset; + +import static org.junit.Assert.assertEquals; + +import java.util.Arrays; +import java.util.Iterator; +import java.util.LinkedHashSet; +import java.util.Optional; +import java.util.Set; + +import org.junit.Test; + +public class GetFirstItemFromSetUnitTest { + + @Test + public void givenASet_whenUsingIterator_thenRetrieveAnItem(){ + Set set = new LinkedHashSet<>(Arrays.asList(1, 2, 3, 4, 5)); + Iterator iterator = set.iterator(); + if(iterator.hasNext()){ + int retrieved = (int) iterator.next(); + assertEquals(retrieved, 1); + } + } + + @Test + public void givenASet_whenUsingStreams_thenRetrieveAnItem(){ + Set set = new LinkedHashSet<>(Arrays.asList(1, 2, 3, 4, 5)); + Optional optional = set.stream().findFirst(); + if(optional.isPresent()){ + int retrieved = optional.get(); + assertEquals(retrieved, 1); + } + } + +}