Examples with ByteString org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString used on opensource projects

Example 26 with ByteString

use of org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString in project jetcd by coreos.

the class MaintenanceUnitTest method testWrite.

@Test(timeOut = 1000)
void testWrite() throws IOException {
    Snapshot snapshot = maintenanceCli.snapshot();
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    ByteString blob = ByteString.copyFromUtf8("blob");
    responseObserverRef.get().onNext(SnapshotResponse.newBuilder().setBlob(blob).setRemainingBytes(0).build());
    responseObserverRef.get().onCompleted();
    snapshot.write(out);
    assertThat(out.toByteArray()).isEqualTo(blob.toByteArray());
}

Example 27 with ByteString

use of org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString in project jetcd by coreos.

the class KVTest method putKeysWithPrefix.

private void putKeysWithPrefix(String prefix, int numPrefixes) throws ExecutionException, InterruptedException {
    for (int i = 0; i < numPrefixes; i++) {
        ByteString key = ByteString.copyFromUtf8(prefix + i);
        ByteString value = ByteString.copyFromUtf8(String.valueOf(i));
        kvClient.put(key, value).get();
    }
}

Example 28 with ByteString

use of org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString in project jetcd by coreos.

the class KVTest method testDelete.

@Test(dependsOnMethods = "testPut")
public void testDelete() throws Exception {
    ByteString keyToDelete = ByteString.copyFrom("sample_key", "UTF-8");
    try {
        // count keys about to delete
        ListenableFuture<RangeResponse> getFeature = kvClient.get(keyToDelete);
        RangeResponse resp = getFeature.get();
        // delete the keys
        ListenableFuture<DeleteRangeResponse> deleteFuture = kvClient.delete(keyToDelete);
        DeleteRangeResponse delResp = deleteFuture.get();
        test.assertEquals(resp.getKvsList().size(), delResp.getDeleted());
    } catch (Exception e) {
    // empty
    }
}

Example 29 with ByteString

use of org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString in project jetcd by coreos.

the class KVTest method testTxn.

@Test
public void testTxn() throws Exception {
    ByteString sampleKey = ByteString.copyFrom("txn_key", "UTF-8");
    ByteString sampleValue = ByteString.copyFrom("xyz", "UTF-8");
    ByteString cmpValue = ByteString.copyFrom("abc", "UTF-8");
    ByteString putValue = ByteString.copyFrom("XYZ", "UTF-8");
    ByteString putValueNew = ByteString.copyFrom("ABC", "UTF-8");
    ListenableFuture<PutResponse> feature = kvClient.put(sampleKey, sampleValue);
    try {
        // put the original txn key value pair
        PutResponse putResp = feature.get();
        // construct txn operation
        Cmp cmp = new Cmp(sampleKey, Cmp.Op.GREATER, CmpTarget.value(cmpValue));
        Txn txn = Txn.newBuilder().If(cmp).Then(Op.put(sampleKey, putValue, PutOption.DEFAULT)).Else(Op.put(sampleKey, putValueNew, PutOption.DEFAULT)).build();
        ListenableFuture<TxnResponse> txnResp = kvClient.commit(txn);
        txnResp.get();
        // get the value
        RangeResponse getResp = kvClient.get(sampleKey).get();
        test.assertEquals(getResp.getKvsList().size(), 1);
        test.assertEquals(getResp.getKvs(0).getValue().toStringUtf8(), "XYZ");
    } catch (Exception e) {
    // empty
    }
}

Example 30 with ByteString

use of org.apache.beam.vendor.grpc.v1p43p2.com.google.protobuf.ByteString in project google-cloud-java by GoogleCloudPlatform.

the class SubscriptionAdminClientTest method getIamPolicyTest.

@Test
@SuppressWarnings("all")
public void getIamPolicyTest() {
    int version = 351608024;
    ByteString etag = ByteString.copyFromUtf8("21");
    Policy expectedResponse = Policy.newBuilder().setVersion(version).setEtag(etag).build();
    mockIAMPolicy.addResponse(expectedResponse);
    String formattedResource = SubscriptionName.create("[PROJECT]", "[SUBSCRIPTION]").toString();
    Policy actualResponse = client.getIamPolicy(formattedResource);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<GeneratedMessageV3> actualRequests = mockIAMPolicy.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    GetIamPolicyRequest actualRequest = (GetIamPolicyRequest) actualRequests.get(0);
    Assert.assertEquals(formattedResource, actualRequest.getResource());
}