improves in in the mapper query

jakartaee · May 23, 2020 · 893f409 · 893f409
1 parent 22810c7
commit 893f409
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/...-column/src/main/java/jakarta/nosql/tck/mapping/column/ColumnMapperSelectBuilderTest.java b/...-column/src/main/java/jakarta/nosql/tck/mapping/column/ColumnMapperSelectBuilderTest.java
@@ -33,6 +33,7 @@
 
 import javax.inject.Inject;
 import java.math.BigDecimal;
+import java.util.Arrays;
 
 import static jakarta.nosql.column.ColumnQuery.select;
 import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -137,6 +138,16 @@ public void shouldSelectWhereNameBetween() {
         assertEquals(queryExpected, query);
     }
 
+    @Test
+    public void shouldSelectWhereNameIn() {
+        ColumnQuery query = mapperBuilder.selectFrom(Person.class).where("id")
+                .in(Arrays.asList("Ada", "Poliana")).build();
+        ColumnQuery queryExpected = select().from("Person").where("_id")
+                .in(Arrays.asList("Ada", "Poliana")).build();
+        assertEquals(queryExpected, query);
+    }
+
+
     @Test
     public void shouldSelectWhereNameNot() {
         ColumnQuery query = mapperBuilder.selectFrom(Person.class).where("name").not().like("Ada").build();

diff --git a/...ent/src/main/java/jakarta/nosql/tck/mapping/document/DocumentMapperSelectBuilderTest.java b/...ent/src/main/java/jakarta/nosql/tck/mapping/document/DocumentMapperSelectBuilderTest.java
@@ -33,6 +33,7 @@
 
 import javax.inject.Inject;
 import java.math.BigDecimal;
+import java.util.Arrays;
 import java.util.Optional;
 import java.util.stream.Stream;
 
@@ -138,6 +139,15 @@ public void shouldSelectWhereNameBetween() {
         assertEquals(queryExpected, query);
     }
 
+    @Test
+    public void shouldSelectWhereNameIn() {
+        DocumentQuery query = mapperBuilder.selectFrom(Person.class).where("id")
+                .in(Arrays.asList("Ada", "Poliana")).build();
+        DocumentQuery queryExpected = select().from("Person").where("_id")
+                .in(Arrays.asList("Ada", "Poliana")).build();
+        assertEquals(queryExpected, query);
+    }
+
     @Test
     public void shouldSelectWhereNameNot() {
         DocumentQuery query = mapperBuilder.selectFrom(Person.class).where("name").not().like("Ada").build();