Wait and refresh router metadata

linkedin · Oct 25, 2023 · 61f43d7 · 61f43d7
1 parent 4339871
commit 61f43d7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/PartialUpdateTest.java b/...-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/PartialUpdateTest.java
@@ -322,6 +322,7 @@ public void testIncrementalPushPartialUpdateClassicFormat() throws IOException {
         runVPJ(vpjProperties, 1, childControllerClient);
       }
       VeniceClusterWrapper veniceClusterWrapper = childDatacenters.get(0).getClusters().get(CLUSTER_NAME);
+      veniceClusterWrapper.waitVersion(storeName, 1);
       try (AvroGenericStoreClient<Object, Object> storeReader = ClientFactory.getAndStartGenericAvroClient(
           ClientConfig.defaultGenericClientConfig(storeName).setVeniceURL(veniceClusterWrapper.getRandomRouterURL()))) {
         TestUtils.waitForNonDeterministicAssertion(10, TimeUnit.SECONDS, true, () -> {
@@ -385,6 +386,8 @@ public void testIncrementalPushPartialUpdateNewFormat() throws IOException {
         runVPJ(vpjProperties, 1, childControllerClient);
       }
       VeniceClusterWrapper veniceClusterWrapper = childDatacenters.get(0).getClusters().get(CLUSTER_NAME);
+      veniceClusterWrapper.waitVersion(storeName, 1);
+
       try (AvroGenericStoreClient<Object, Object> storeReader = ClientFactory.getAndStartGenericAvroClient(
           ClientConfig.defaultGenericClientConfig(storeName).setVeniceURL(veniceClusterWrapper.getRandomRouterURL()))) {
         TestUtils.waitForNonDeterministicAssertion(10, TimeUnit.SECONDS, true, () -> {
@@ -448,6 +451,7 @@ public void testPartialUpdateOnBatchPushedKeys(CompressionStrategy compressionSt
       try (ControllerClient childControllerClient = new ControllerClient(CLUSTER_NAME, childControllerUrl)) {
         runVPJ(vpjProperties, 1, childControllerClient);
       }
+      veniceClusterWrapper.waitVersion(storeName, 1);
       // Produce partial updates on batch pushed keys
       SystemProducer veniceProducer = getSamzaProducer(veniceClusterWrapper, storeName, Version.PushType.STREAM);
       for (int i = 1; i < 100; i++) {
@@ -1025,7 +1029,7 @@ public void testWriteComputeWithHybridLeaderFollowerLargeRecord(
         try (ControllerClient childControllerClient = new ControllerClient(CLUSTER_NAME, childControllerUrl)) {
           runVPJ(vpjProperties, 1, childControllerClient);
         }
-
+        veniceClusterWrapper.waitVersion(storeName, 1);
         try (AvroGenericStoreClient<Object, Object> storeReader = ClientFactory.getAndStartGenericAvroClient(
             ClientConfig.defaultGenericClientConfig(storeName)
                 .setVeniceURL(veniceClusterWrapper.getRandomRouterURL()))) {