diff --git a/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/EntrySetExampleUnitTest.java b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/EntrySetExampleUnitTest.java new file mode 100644 index 0000000000..fea88f729b --- /dev/null +++ b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/EntrySetExampleUnitTest.java @@ -0,0 +1,30 @@ +package com.baeldung.map.keysetValuesEntrySet; + +import org.junit.Test; + +import java.util.AbstractMap.SimpleEntry; +import java.util.HashMap; +import java.util.Map; +import java.util.Set; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +public class EntrySetExampleUnitTest { + + @Test + public void givenHashMap_whenEntrySetApplied_thenShouldReturnSetOfEntries() { + + Map map = new HashMap<>(); + map.put("one", 1); + map.put("two", 2); + + Set> actualValues = map.entrySet(); + + assertEquals(2, actualValues.size()); + assertTrue(actualValues.contains(new SimpleEntry<>("one", 1))); + assertTrue(actualValues.contains(new SimpleEntry<>("two", 2))); + + } + +} diff --git a/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/KeySetExampleUnitTest.java b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/KeySetExampleUnitTest.java new file mode 100644 index 0000000000..6cb0620e35 --- /dev/null +++ b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/KeySetExampleUnitTest.java @@ -0,0 +1,27 @@ +package com.baeldung.map.keysetValuesEntrySet; + +import org.junit.Test; + +import java.util.HashMap; +import java.util.Map; +import java.util.Set; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +public class KeySetExampleUnitTest { + + @Test + public void givenHashMap_whenKeySetApplied_thenShouldReturnSetOfKeys() { + Map map = new HashMap<>(); + map.put("one", 1); + map.put("two", 2); + + Set actualValues = map.keySet(); + + assertEquals(2, actualValues.size()); + assertTrue(actualValues.contains("one")); + assertTrue(actualValues.contains("two")); + } + +} diff --git a/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/ValuesExampleUnitTest.java b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/ValuesExampleUnitTest.java new file mode 100644 index 0000000000..ba9369a122 --- /dev/null +++ b/java-collections-maps-3/src/test/java/com/baeldung/map/keysetValuesEntrySet/ValuesExampleUnitTest.java @@ -0,0 +1,27 @@ +package com.baeldung.map.keysetValuesEntrySet; + +import org.junit.Test; + +import java.util.Collection; +import java.util.HashMap; +import java.util.Map; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +public class ValuesExampleUnitTest { + + @Test + public void givenHashMap_whenValuesApplied_thenShouldReturnCollectionOfValues() { + Map map = new HashMap<>(); + map.put("one", 1); + map.put("two", 2); + + Collection actualValues = map.values(); + + assertEquals(2, actualValues.size()); + assertTrue(actualValues.contains(1)); + assertTrue(actualValues.contains(2)); + } + +}