test: create mock to avoid NPE at graph

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse-jnosql · Nov 18, 2023 · 0452723 · 0452723
1 parent 0049cd1
commit 0452723
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/...nosql-mapping-graph/src/test/java/org/eclipse/jnosql/mapping/graph/cdi/GraphProducer.java b/...nosql-mapping-graph/src/test/java/org/eclipse/jnosql/mapping/graph/cdi/GraphProducer.java
@@ -90,7 +90,7 @@ public Graph getGraphMock() {
         when(vertex.keys()).thenReturn(singleton("name"));
         when(vertex.value("name")).thenReturn("nameMock");
         when(graphMock.addVertex(Mockito.anyString())).thenReturn(vertex);
-
+        when(graphMock.vertices(Mockito.any())).thenReturn(Collections.emptyIterator());
         return graphMock;
     }
 

diff --git a/...ping-graph/src/test/java/org/eclipse/jnosql/mapping/graph/producer/AbstractGraphMock.java b/...ping-graph/src/test/java/org/eclipse/jnosql/mapping/graph/producer/AbstractGraphMock.java
@@ -22,6 +22,7 @@
 import org.apache.tinkerpop.gremlin.structure.Vertex;
 import org.eclipse.jnosql.communication.Settings;
 
+import java.util.Collections;
 import java.util.Iterator;
 import java.util.Optional;
 
@@ -51,7 +52,7 @@ public GraphComputer compute() throws IllegalArgumentException {
 
     @Override
     public Iterator<Vertex> vertices(Object... vertexIds) {
-        return null;
+        return Collections.emptyIterator();
     }
 
     @Override

diff --git a/...st/java/org/eclipse/jnosql/mapping/graph/spi/GraphConfigurationProducerExtensionTest.java b/...st/java/org/eclipse/jnosql/mapping/graph/spi/GraphConfigurationProducerExtensionTest.java
@@ -71,7 +71,7 @@ void shouldSave() {
 
     @Test
     void shouldInjectRepository() {
-        repositoryMock.save(Book.builder().withName("book").build());
+        repositoryMock.save(Book.builder().withName("book").withAge(12).build());
         assertNotNull(repository.findById("10"));
     }