fix remove vertex

lor6 · lor6 · commit 8ac77800a0a6 · 2019-05-26T19:38:40.000+03:00
diff --git a/core-java-modules/core-java/src/main/java/com/baeldung/graph/Graph.java b/core-java-modules/core-java/src/main/java/com/baeldung/graph/Graph.java
@@ -19,7 +19,7 @@ void addVertex(String label) {
 
     void removeVertex(String label) {
         Vertex v = new Vertex(label);
-        adjVertices.values().stream().map(e -> e.remove(v)).collect(Collectors.toList());
+        adjVertices.values().stream().forEach(e -> e.remove(v));
         adjVertices.remove(new Vertex(label));
     }
 
diff --git a/core-java-modules/core-java/src/test/java/com/baeldung/graph/GraphUnitTest.java b/core-java-modules/core-java/src/test/java/com/baeldung/graph/GraphUnitTest.java
@@ -1,20 +1,31 @@
 package com.baeldung.graph;
 
-import org.junit.Assert;
+import static org.junit.Assert.assertEquals;
 import org.junit.Test;
 
-public class GraphTraversalUnitTest {
+public class GraphUnitTest {
     @Test
     public void givenAGraph_whenTraversingDepthFirst_thenExpectedResult() {
         Graph graph = createGraph();
-        Assert.assertEquals("[Bob, Rob, Maria, Alice, Mark]", 
+        assertEquals("[Bob, Rob, Maria, Alice, Mark]", 
             GraphTraversal.depthFirstTraversal(graph, "Bob").toString());
     }
     
     @Test
     public void givenAGraph_whenTraversingBreadthFirst_thenExpectedResult() {
         Graph graph = createGraph();
-        Assert.assertEquals("[Bob, Alice, Rob, Mark, Maria]", 
+       assertEquals("[Bob, Alice, Rob, Mark, Maria]", 
+            GraphTraversal.breadthFirstTraversal(graph, "Bob").toString());
+    }
+    
+    @Test
+    public void givenAGraph_whenRemoveVertex_thenVertedNotFound() {
+        Graph graph = createGraph();
+        assertEquals("[Bob, Alice, Rob, Mark, Maria]", 
+            GraphTraversal.breadthFirstTraversal(graph, "Bob").toString());
+        
+        graph.removeVertex("Maria");
+        assertEquals("[Bob, Alice, Rob, Mark]", 
             GraphTraversal.breadthFirstTraversal(graph, "Bob").toString());
     }
     

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@ void addVertex(String label) {`
`19`	`19`
`20`	`20`	`void removeVertex(String label) {`
`21`	`21`	`Vertex v = new Vertex(label);`
`22`		`- adjVertices.values().stream().map(e -> e.remove(v)).collect(Collectors.toList());`
	`22`	`+ adjVertices.values().stream().forEach(e -> e.remove(v));`
`23`	`23`	`adjVertices.remove(new Vertex(label));`
`24`	`24`	`}`
`25`	`25`