test: add samples for IN, NOT_IN, and NOT_EQUALS.

googleapis · May 25, 2022 · eef1851 · eef1851
1 parent e8ba44f
commit eef1851
Showing 1 changed file with 77 additions and 0 deletions.
diff --git a/samples/snippets/src/test/java/com/google/datastore/snippets/ConceptsTest.java b/samples/snippets/src/test/java/com/google/datastore/snippets/ConceptsTest.java
@@ -1044,4 +1044,81 @@ public void testPropertyFilteringRunQuery() {
         ImmutableMap.of("Task", ImmutableSet.of("priority", "tag"));
     assertEquals(expected, propertiesByKind);
   }
+
+  @Test
+  public void testInQuery() {
+    setUpQueryTests();
+    // [START datastore_in_query]
+    Query<Entity> query =
+        Query.newEntityQueryBuilder()
+            .setKind("Task")
+            .setFilter(
+                PropertyFilter.in("tag", ListValue.of("learn", "study"))
+            )
+            .build();
+    // [END datastore_in_query]
+    assertValidQuery(query);
+  }
+
+  @Test
+  public void testNotEqualsQuery() {
+    setUpQueryTests();
+    // [START datastore_not_equals_query]
+    Query<Entity> query =
+        Query.newEntityQueryBuilder()
+            .setKind("Task")
+            .setFilter(
+                PropertyFilter.neq("category", "Work")
+            )
+            .build();
+    // [END datastore_not_equals_query]
+    assertValidQuery(query);
+  }
+
+  @Test
+  public void testNotInQuery() {
+    setUpQueryTests();
+    // [START datastore_not_in_query]
+    Query<Entity> query =
+        Query.newEntityQueryBuilder()
+            .setKind("Task")
+            .setFilter(
+                PropertyFilter.not_in("category", ListValue.of("Work", "Chores", "School"))
+            )
+            .build();
+    // [END datastore_not_in_query]
+    assertValidQuery(query);
+  }
+
+  @Test
+  public void testEqQuerySorted() {
+    setUpQueryTests();
+    // [START datastore_eq_query_sorted]
+    Query<Entity> query =
+        Query.newEntityQueryBuilder()
+            .setKind("Task")
+            .setFilter(
+                PropertyFilter.eq("tag", "learn")
+            )
+            .setOrderBy(OrderBy.asc("tag"))
+            .build();
+    // [END datastore_eq_query_sorted]
+    assertValidQuery(query);
+  }
+
+  @Test
+  public void testInQuerySorted() {
+    setUpQueryTests();
+    // [START datastore_in_query_sorted]
+    Query<Entity> query =
+        Query.newEntityQueryBuilder()
+            .setKind("Task")
+            .setFilter(
+                PropertyFilter.in("tag", ListValue.of("learn", "study"))
+            )
+            .setOrderBy(OrderBy.asc("tag"))
+            .build();
+    // [END datastore_in_query_sorted]
+    assertValidQuery(query);
+  }
 }