feat: update tinkerpop query

Signed-off-by: Otavio Santana <[email protected]>

Author: otaviojava

jnosql-tinkerpop/src/main/java/org/eclipse/jnosql/databases/tinkerpop/communication/DefaultTinkerpopGraphDatabaseManager.java

@@ -199,11 +199,11 @@ public void remove(CommunicationEntity source, String label, CommunicationEntity
         Objects.requireNonNull(label, "label is required");
 
         Vertex sourceVertex = findVertexById(source.find(ID_PROPERTY)
-                .orElseThrow(() -> new CommunicationException("Source entity must have an ID")))
+                .orElseThrow(() -> new CommunicationException("Source entity must have an ID")).get())
                 .orElseThrow(() -> new EmptyResultException("Source entity not found"));
 
         Vertex targetVertex = findVertexById(target.find(ID_PROPERTY)
-                .orElseThrow(() -> new CommunicationException("Target entity must have an ID")))
+                .orElseThrow(() -> new CommunicationException("Target entity must have an ID")).get())
                 .orElseThrow(() -> new EmptyResultException("Target entity not found"));
 
         Iterator<Edge> edges = sourceVertex.edges(Direction.OUT, label);

jnosql-tinkerpop/src/test/java/org/eclipse/jnosql/databases/tinkerpop/communication/DefaultTinkerpopGraphDatabaseManagerTest.java

@@ -450,12 +450,11 @@ void shouldRemoveEdge() {
         var person1 = entityManager.insert(getEntity());
         var person2 = entityManager.insert(getEntity());
 
-        entityManager.edge(person1, "FRIEND", person2, Map.of());
+        CommunicationEdge communicationEdge = entityManager.edge(person1, "FRIEND", person2, Map.of());
 
         entityManager.remove(person1, "FRIEND", person2);
 
-        var edgesQuery = select().from(COLLECTION_NAME).where("_id").eq(person1.find("_id").orElseThrow().get()).build();
-        var edges = entityManager.select(edgesQuery).toList();
+        var edges = entityManager.findEdgeById(communicationEdge.id());
 
         assertThat(edges).isEmpty();
     }