Search in sources :

Example 6 with CheckAndMutateRowRequest

use of com.google.bigtable.v2.CheckAndMutateRowRequest in project java-bigtable-hbase by googleapis.

the class TestCheckAndMutateUtil method testRowMutations.

@Test
public /**
 * Tests that a CheckAndMutate with a {@link RowMutations} works correctly for the filter and
 * mutation aspects
 */
void testRowMutations() throws IOException {
    CheckAndMutateUtil.RequestBuilder underTest = createRequestBuilder();
    RowMutations rowMutations = new RowMutations(rowKey);
    rowMutations.add(PUT);
    CheckAndMutateRowRequest result = underTest.qualifier(qual).ifMatches(CompareOp.EQUAL, checkValue).withMutations(rowMutations).build().toProto(REQUEST_CONTEXT);
    Assert.assertEquals(1, result.getTrueMutationsCount());
    checkPutMutation(result.getTrueMutations(0));
    checkPredicate(result);
}
Also used : CheckAndMutateRowRequest(com.google.bigtable.v2.CheckAndMutateRowRequest) RowMutations(org.apache.hadoop.hbase.client.RowMutations) Test(org.junit.Test)

Example 7 with CheckAndMutateRowRequest

use of com.google.bigtable.v2.CheckAndMutateRowRequest in project java-bigtable-hbase by googleapis.

the class TestCheckAndMutateUtil method testIfNotExists.

@Test
public /**
 * Test to make sure that {@link CheckAndMutateUtil.RequestBuilder#ifNotExists()} works correctly
 */
void testIfNotExists() throws DoNotRetryIOException {
    CheckAndMutateUtil.RequestBuilder underTest = createRequestBuilder();
    CheckAndMutateRowRequest result = underTest.qualifier(qual).ifNotExists().withPut(PUT).build().toProto(REQUEST_CONTEXT);
    Assert.assertEquals(1, result.getFalseMutationsCount());
    checkPutMutation(result.getFalseMutations(0));
    RowFilter expected = FILTERS.chain().filter(FAMILY_AND_QUAL_FILTER).filter(FILTERS.limit().cellsPerColumn(1)).toProto();
    Assert.assertEquals(expected, result.getPredicateFilter());
}
Also used : RowFilter(com.google.bigtable.v2.RowFilter) CheckAndMutateRowRequest(com.google.bigtable.v2.CheckAndMutateRowRequest) Test(org.junit.Test)

Example 8 with CheckAndMutateRowRequest

use of com.google.bigtable.v2.CheckAndMutateRowRequest in project java-bigtable-hbase by googleapis.

the class TestCheckAndMutateUtil method testPut.

// ************************** TEST METHODS ***************************
@Test
public /**
 * Tests that a CheckAndMutate with a {@link Put} works correctly for the filter and mutation
 * aspects
 */
void testPut() throws DoNotRetryIOException {
    CheckAndMutateUtil.RequestBuilder underTest = createRequestBuilder();
    CheckAndMutateRowRequest result = underTest.qualifier(qual).ifMatches(CompareOp.EQUAL, checkValue).withPut(PUT).build().toProto(REQUEST_CONTEXT);
    Assert.assertEquals(1, result.getTrueMutationsCount());
    checkPutMutation(result.getTrueMutations(0));
    checkPredicate(result);
}
Also used : CheckAndMutateRowRequest(com.google.bigtable.v2.CheckAndMutateRowRequest) Test(org.junit.Test)

Example 9 with CheckAndMutateRowRequest

use of com.google.bigtable.v2.CheckAndMutateRowRequest in project java-bigtable by googleapis.

the class ConditionalRowMutationTest method conditionTest.

@Test
public void conditionTest() {
    ConditionalRowMutation mutation = ConditionalRowMutation.create(TABLE_ID, TEST_KEY).condition(Filters.FILTERS.key().regex("a.*")).then(Mutation.create().deleteRow());
    CheckAndMutateRowRequest actualProto = mutation.toProto(REQUEST_CONTEXT);
    assertThat(actualProto.getPredicateFilter()).isEqualTo(RowFilter.newBuilder().setRowKeyRegexFilter(ByteString.copyFromUtf8("a.*")).build());
}
Also used : CheckAndMutateRowRequest(com.google.bigtable.v2.CheckAndMutateRowRequest) Test(org.junit.Test)

Example 10 with CheckAndMutateRowRequest

use of com.google.bigtable.v2.CheckAndMutateRowRequest in project java-bigtable by googleapis.

the class ConditionalRowMutationTest method thenTest.

@Test
public void thenTest() {
    ConditionalRowMutation mutation = ConditionalRowMutation.create(TABLE_ID, TEST_KEY).then(Mutation.create().deleteCells("family1", "qualifier1")).then(Mutation.create().deleteCells("family2", "qualifier2"));
    CheckAndMutateRowRequest actualProto = mutation.toProto(REQUEST_CONTEXT);
    assertThat(actualProto.getTrueMutationsList()).containsExactly(com.google.bigtable.v2.Mutation.newBuilder().setDeleteFromColumn(DeleteFromColumn.newBuilder().setFamilyName("family1").setColumnQualifier(ByteString.copyFromUtf8("qualifier1"))).build(), com.google.bigtable.v2.Mutation.newBuilder().setDeleteFromColumn(DeleteFromColumn.newBuilder().setFamilyName("family2").setColumnQualifier(ByteString.copyFromUtf8("qualifier2"))).build()).inOrder();
}
Also used : CheckAndMutateRowRequest(com.google.bigtable.v2.CheckAndMutateRowRequest) Test(org.junit.Test)

Aggregations

CheckAndMutateRowRequest (com.google.bigtable.v2.CheckAndMutateRowRequest)21 Test (org.junit.Test)18 Mutation (com.google.bigtable.v2.Mutation)10 RowFilter (com.google.bigtable.v2.RowFilter)8 CheckAndMutateRowResponse (com.google.bigtable.v2.CheckAndMutateRowResponse)7 ArrayList (java.util.ArrayList)7 ByteString (com.google.protobuf.ByteString)6 AbstractMessage (com.google.protobuf.AbstractMessage)4 TableName (com.google.bigtable.v2.TableName)2 BaseBigtableDataClient (com.google.cloud.bigtable.data.v2.BaseBigtableDataClient)2 RowMutations (org.apache.hadoop.hbase.client.RowMutations)2 HeaderTracerUnaryCallable (com.google.cloud.bigtable.data.v2.stub.metrics.HeaderTracerUnaryCallable)1 StatsHeadersUnaryCallable (com.google.cloud.bigtable.data.v2.stub.metrics.StatsHeadersUnaryCallable)1 ImmutableMap (com.google.common.collect.ImmutableMap)1 Map (java.util.Map)1 Delete (org.apache.hadoop.hbase.client.Delete)1 Put (org.apache.hadoop.hbase.client.Put)1 CompareOp (org.apache.hadoop.hbase.filter.CompareFilter.CompareOp)1