Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

samples: Add sample snippets for IN, NOT_EQUALS, and NOT_IN #749

Merged
merged 5 commits into from
Jun 14, 2022
Merged

samples: Add sample snippets for IN, NOT_EQUALS, and NOT_IN #749

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
eef1851
test: add samples for IN, NOT_IN, and NOT_EQUALS.
jlara310 May 25, 2022
9ec7ffb
Merge branch 'googleapis:main' into in-not-equals-samples
jlara310 Jun 13, 2022
34e8de2
Set up another Datastore client for tests that require a real backend.
jlara310 Jun 14, 2022
ec832b2
Add a method for setting up a test entity. Fix linting errors.
jlara310 Jun 14, 2022
9c2ddd9
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jun 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
103 changes: 102 additions & 1 deletion samples/snippets/src/test/java/com/google/datastore/snippets/ConceptsTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
import com.google.cloud.datastore.Cursor;
import com.google.cloud.datastore.Datastore;
import com.google.cloud.datastore.DatastoreException;
import com.google.cloud.datastore.DatastoreOptions;
import com.google.cloud.datastore.Entity;
import com.google.cloud.datastore.EntityQuery;
import com.google.cloud.datastore.FullEntity;
Expand Down Expand Up @@ -83,6 +84,7 @@ public class ConceptsTest {
private static final FullEntity<IncompleteKey> TEST_FULL_ENTITY = FullEntity.newBuilder().build();

private Datastore datastore;
private Datastore datastoreRealBackend;
private KeyFactory keyFactory;
private Key taskKey;
private Entity testEntity;
Expand Down Expand Up @@ -123,6 +125,8 @@ public void setUp() {
endDate = Timestamp.of(calendar.getTime());
calendar.set(1999, DECEMBER, 31);
includedDate = Timestamp.of(calendar.getTime());
// Create a client for tests that require a real backend
datastoreRealBackend = DatastoreOptions.getDefaultInstance().getService();
}

/**
Expand Down Expand Up @@ -384,7 +388,7 @@ private void setUpQueryTests() {
.set(
"description",
StringValue.newBuilder("Learn Cloud Datastore").setExcludeFromIndexes(true).build())
.set("tag", "fun", "l", "programming")
.set("tag", "fun", "l", "programming", "learn")
.build());
}

Expand Down Expand Up @@ -1044,4 +1048,101 @@ public void testPropertyFilteringRunQuery() {
ImmutableMap.of("Task", ImmutableSet.of("priority", "tag"));
assertEquals(expected, propertiesByKind);
}

@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);
}

/** Start tests using a real backend. */
private <V> V assertValidQueryRealBackend(Query<V> query) {
QueryResults<V> results = datastoreRealBackend.run(query);
V result = results.next();
// assertFalse(results.hasNext());
return result;
}

private void setUpQueryTestsRealBackend() {
Key taskKey =
datastoreRealBackend
.newKeyFactory()
.setKind("Task")
.addAncestors(PathElement.of("TaskList", "default"))
.newKey("someTask");
datastoreRealBackend.put(
Entity.newBuilder(taskKey)
.set("category", "Personal")
.set("done", false)
.set("completed", false)
.set("priority", 4)
.set("created", includedDate)
.set("percent_complete", 10.0)
.set(
"description",
StringValue.newBuilder("Learn Cloud Datastore").setExcludeFromIndexes(true).build())
.set("tag", "fun", "l", "programming", "learn")
.build());
}

@Test
public void testInQuery() {
setUpQueryTestsRealBackend();
// [START datastore_in_query]
Query<Entity> query =
Query.newEntityQueryBuilder()
.setKind("Task")
.setFilter(PropertyFilter.in("tag", ListValue.of("learn", "study")))
.build();
// [END datastore_in_query]
assertValidQueryRealBackend(query);
}

@Test
public void testNotEqualsQuery() {
setUpQueryTestsRealBackend();
// [START datastore_not_equals_query]
Query<Entity> query =
Query.newEntityQueryBuilder()
.setKind("Task")
.setFilter(PropertyFilter.neq("category", "Work"))
.build();
// [END datastore_not_equals_query]
assertValidQueryRealBackend(query);
}

@Test
public void testNotInQuery() {
setUpQueryTestsRealBackend();
// [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]
assertValidQueryRealBackend(query);
}

@Test
public void testInQuerySorted() {
setUpQueryTestsRealBackend();
// [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]
assertValidQueryRealBackend(query);
}
}