Add query in test

eclipse · Feb 12, 2018 · da2ab0b · da2ab0b
1 parent 899271c
commit da2ab0b
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/...c/test/java/org/jnosql/diana/orientdb/document/OrientDBDocumentCollectionManagerTest.java b/...c/test/java/org/jnosql/diana/orientdb/document/OrientDBDocumentCollectionManagerTest.java
@@ -33,6 +33,8 @@
 import java.util.Random;
 import java.util.function.Consumer;
 import java.util.logging.Logger;
+import java.util.stream.Collectors;
+import java.util.stream.StreamSupport;
 
 import static java.util.Arrays.asList;
 import static java.util.Collections.singletonMap;
@@ -292,6 +294,24 @@ public void shouldQueryLesserEqualsThan() {
         assertTrue(entityManager.select(query3).size() == 1);
     }
 
+    @Test
+    public void shouldQueryIn() {
+        List<DocumentEntity> entities = StreamSupport.stream(entityManager.insert(getEntities()).spliterator(), false)
+                .collect(Collectors.toList());
+
+        DocumentQuery query = select().from(COLLECTION_NAME)
+                .where("city").in(asList("Salvador", "Assis"))
+                .build();
+        assertTrue(entityManager.select(query).size() == 2);
+
+        DocumentDeleteQuery deleteQuery = delete().from(COLLECTION_NAME)
+                .where("city").in(asList("Salvador", "Assis", "Sao Paulo"))
+                .build();
+        entityManager.delete(deleteQuery);
+
+        assertTrue(entityManager.select(query).isEmpty());
+    }
+
     @Test
     public void shouldLive() throws InterruptedException {
         List<DocumentEntity> entities = new ArrayList<>();
@@ -357,4 +377,16 @@ private DocumentEntity getEntity() {
         documents.forEach(entity::add);
         return entity;
     }
+
+    private List<DocumentEntity> getEntities() {
+        DocumentEntity otavio = DocumentEntity.of(COLLECTION_NAME);
+        otavio.add(Document.of("name", "Otavio"));
+        otavio.add(Document.of("city", "Sao Paulo"));
+
+        DocumentEntity lucas = DocumentEntity.of(COLLECTION_NAME);
+        lucas.add(Document.of("name", "Lucas"));
+        lucas.add(Document.of("city", "Assis"));
+
+        return asList(getEntity(), otavio, lucas);
+    }
 }