From 8841a720a914ec063e4bd4d37654bbef3e3e5484 Mon Sep 17 00:00:00 2001 From: Sam Gardner Date: Wed, 2 Aug 2023 12:05:17 +0100 Subject: [PATCH] BAEL-6837 Add example code for getting the first item from a Set --- .../GetFirstItemFromSetUnitTest.java | 35 +++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 core-java-modules/core-java-collections-set-2/src/test/java/com/baeldung/firstitemfromset/GetFirstItemFromSetUnitTest.java 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); + } + } + +}