update from proto names

apache · Feb 13, 2023 · 07d37ea · 07d37ea
1 parent fe9ab2a
commit 07d37ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/control/ProcessBundleHandler.java b/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/control/ProcessBundleHandler.java
@@ -165,7 +165,6 @@ public class ProcessBundleHandler {
   private final Cache<Object, Object> processWideCache;
   @VisibleForTesting final BundleProcessorCache bundleProcessorCache;
   private final Set<String> runnerCapabilities;
-
   private final @Nullable DataSampler dataSampler;
 
   public ProcessBundleHandler(

diff --git a/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/debug/DataSamplerTest.java b/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/debug/DataSamplerTest.java
@@ -57,7 +57,7 @@ byte[] encodeString(String s) throws IOException {
   BeamFnApi.InstructionResponse getAllSamples(DataSampler dataSampler) {
     BeamFnApi.InstructionRequest request =
         BeamFnApi.InstructionRequest.newBuilder()
-            .setSample(BeamFnApi.SampleDataRequest.newBuilder().build())
+            .setSampleData(BeamFnApi.SampleDataRequest.newBuilder().build())
             .build();
     return dataSampler.handleDataSampleRequest(request).build();
   }
@@ -66,7 +66,7 @@ BeamFnApi.InstructionResponse getSamplesForPCollection(
       DataSampler dataSampler, String pcollection) {
     BeamFnApi.InstructionRequest request =
         BeamFnApi.InstructionRequest.newBuilder()
-            .setSample(
+            .setSampleData(
                 BeamFnApi.SampleDataRequest.newBuilder().addPcollectionIds(pcollection).build())
             .build();
     return dataSampler.handleDataSampleRequest(request).build();
@@ -76,7 +76,7 @@ BeamFnApi.InstructionResponse getSamplesForPCollections(
       DataSampler dataSampler, Iterable<String> pcollections) {
     BeamFnApi.InstructionRequest request =
         BeamFnApi.InstructionRequest.newBuilder()
-            .setSample(
+            .setSampleData(
                 BeamFnApi.SampleDataRequest.newBuilder().addAllPcollectionIds(pcollections).build())
             .build();
     return dataSampler.handleDataSampleRequest(request).build();
@@ -85,7 +85,7 @@ BeamFnApi.InstructionResponse getSamplesForPCollections(
   void assertHasSamples(
       BeamFnApi.InstructionResponse response, String pcollection, Iterable<byte[]> elements) {
     Map<String, BeamFnApi.SampleDataResponse.ElementList> elementSamplesMap =
-        response.getSample().getElementSamplesMap();
+        response.getSampleData().getElementSamplesMap();
 
     assertFalse(elementSamplesMap.isEmpty());
 
@@ -189,7 +189,7 @@ public void testFiltersMultiplePCollectionIds() throws Exception {
     generateStringSamples(sampler);
 
     BeamFnApi.InstructionResponse samples = getSamplesForPCollections(sampler, pcollectionIds);
-    assertThat(samples.getSample().getElementSamplesMap().size(), equalTo(2));
+    assertThat(samples.getSampleData().getElementSamplesMap().size(), equalTo(2));
     assertHasSamples(samples, "a", ImmutableList.of(encodeString("a1"), encodeString("a2")));
     assertHasSamples(samples, "c", ImmutableList.of(encodeString("c1"), encodeString("c2")));
   }