changes as per review (eugenp#6341)

anuraggoyal1 · jzheaux · commit 96fbef6a14a9 · 2019-02-18T08:44:38.000-07:00
* [BAEL-2471] Guide to Apache Commons MultiValuedMap

* Update MultiValuedMapUnitTest.java

* added empty lines

* updated as per review comments

* changes as per review
diff --git a/java-collections-maps/src/test/java/com/baeldung/java/map/MultiValuedMapUnitTest.java b/java-collections-maps/src/test/java/com/baeldung/java/map/MultiValuedMapUnitTest.java
@@ -9,8 +9,10 @@
 import java.util.Collection;
 import java.util.Map;
 import java.util.Map.Entry;
+import java.util.Set;
 
 import org.apache.commons.collections4.MultiMapUtils;
+import org.apache.commons.collections4.MultiSet;
 import org.apache.commons.collections4.MultiValuedMap;
 import org.apache.commons.collections4.multimap.ArrayListValuedHashMap;
 import org.apache.commons.collections4.multimap.HashSetValuedHashMap;
@@ -65,25 +67,28 @@ public void givenMultiValuesMap_whenUsingEntriesMethod_thenReturningMappings() {
     @Test
     public void givenMultiValuesMap_whenUsingKeysMethod_thenReturningAllKeys() {
         MultiValuedMap<String, String> map = new ArrayListValuedHashMap<>();
-
         map.put("fruits", "apple");
         map.put("fruits", "orange");
         map.put("vehicles", "car");
         map.put("vehicles", "bike");
         
-        assertThat(((Collection<String>) map.keys())).contains("fruits", "vehicles");
+        MultiSet<String> keys = map.keys();
+        
+        assertThat((keys)).contains("fruits", "vehicles");
+
     }
 
     @Test
     public void givenMultiValuesMap_whenUsingKeySetMethod_thenReturningAllKeys() {
         MultiValuedMap<String, String> map = new ArrayListValuedHashMap<>();
-
         map.put("fruits", "apple");
         map.put("fruits", "orange");
         map.put("vehicles", "car");
         map.put("vehicles", "bike");
         
-        assertThat((Collection<String>) map.keySet()).contains("fruits", "vehicles");
+        Set<String> keys = map.keySet();
+        
+        assertThat(keys).contains("fruits", "vehicles");
 
     }
 
