Skip to content
This repository was archived by the owner on May 8, 2026. It is now read-only.

chore(test): Add test to verify that query ranges work as expected #445

Merged
igorbernstein2 merged 2 commits into from
Oct 12, 2020
Merged

chore(test): Add test to verify that query ranges work as expected #445

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
093dc69
chore(test): Add test to verify that query ranges work as expected
igorbernstein2 Oct 6, 2020
30db5c2
format
igorbernstein2 Oct 6, 2020
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
75 changes: 75 additions & 0 deletions google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/it/ReadIT.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,14 @@
import com.google.api.core.SettableApiFuture;
import com.google.api.gax.rpc.ResponseObserver;
import com.google.api.gax.rpc.StreamController;
import com.google.cloud.bigtable.data.v2.BigtableDataClient;
import com.google.cloud.bigtable.data.v2.models.BulkMutation;
import com.google.cloud.bigtable.data.v2.models.Query;
import com.google.cloud.bigtable.data.v2.models.Range.ByteStringRange;
import com.google.cloud.bigtable.data.v2.models.Row;
import com.google.cloud.bigtable.data.v2.models.RowCell;
import com.google.cloud.bigtable.data.v2.models.RowMutation;
import com.google.cloud.bigtable.data.v2.models.RowMutationEntry;
import com.google.cloud.bigtable.test_helpers.env.TestEnvRule;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
Expand Down Expand Up @@ -149,6 +153,77 @@ public void read() throws Throwable {
assertThat(actualRowFuture.get()).isEqualTo(expectedRows.get(0));
}

@Test
public void rangeQueries() {
BigtableDataClient client = testEnvRule.env().getDataClient();
String tableId = testEnvRule.env().getTableId();
String familyId = testEnvRule.env().getFamilyId();
String uniqueKey = prefix + "-range-queries";
String keyA = uniqueKey + "-" + "a";
String keyZ = uniqueKey + "-" + "z";

long timestampMicros = System.currentTimeMillis() * 1_000;

client.bulkMutateRows(
BulkMutation.create(tableId)
.add(RowMutationEntry.create(keyA).setCell(familyId, "", timestampMicros, "A"))
.add(RowMutationEntry.create(keyZ).setCell(familyId, "", timestampMicros, "Z")));

Row expectedRowA =
Row.create(
ByteString.copyFromUtf8(keyA),
ImmutableList.of(
RowCell.create(
testEnvRule.env().getFamilyId(),
ByteString.copyFromUtf8(""),
timestampMicros,
ImmutableList.<String>of(),
ByteString.copyFromUtf8("A"))));

Row expectedRowZ =
Row.create(
ByteString.copyFromUtf8(keyZ),
ImmutableList.of(
RowCell.create(
testEnvRule.env().getFamilyId(),
ByteString.copyFromUtf8(""),
timestampMicros,
ImmutableList.<String>of(),
ByteString.copyFromUtf8("Z"))));

// Closed/Open
assertThat(
ImmutableList.copyOf(
client.readRows(
Query.create(tableId)
.range(ByteStringRange.unbounded().startClosed(keyA).endOpen(keyZ)))))
.containsExactly(expectedRowA);

// Closed/Closed
assertThat(
ImmutableList.copyOf(
client.readRows(
Query.create(tableId)
.range(ByteStringRange.unbounded().startClosed(keyA).endClosed(keyZ)))))
.containsExactly(expectedRowA, expectedRowZ);

// Open/Closed
assertThat(
ImmutableList.copyOf(
client.readRows(
Query.create(tableId)
.range(ByteStringRange.unbounded().startOpen(keyA).endClosed(keyZ)))))
.containsExactly(expectedRowZ);

// Open/Open
assertThat(
ImmutableList.copyOf(
client.readRows(
Query.create(tableId)
.range(ByteStringRange.unbounded().startOpen(keyA).endOpen(keyZ)))))
.isEmpty();
}

@Test
public void readSingleNonexistentAsyncCallback() throws Exception {
ApiFuture<Row> future =
Expand Down