diff --git a/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientDiskFullTest.java b/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientDiskFullTest.java index c7a7a32176..6487a9747b 100644 --- a/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientDiskFullTest.java +++ b/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientDiskFullTest.java @@ -219,7 +219,7 @@ public void testDaVinciDiskFullFailure(boolean useDaVinciSpecificExecutionStatus // Do a VPJ push runVPJ(vpjProperties, 1, controllerClient); - + venice.refreshAllRouterMetaData(); // Verify some records (note, record 1 have been pushed) TestUtils.waitForNonDeterministicAssertion(10, TimeUnit.SECONDS, true, () -> { try { diff --git a/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientMemoryLimitTest.java b/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientMemoryLimitTest.java index b8ab2c13ef..b2ebb270c2 100644 --- a/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientMemoryLimitTest.java +++ b/internal/venice-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientMemoryLimitTest.java @@ -167,6 +167,7 @@ public void testDaVinciMemoryLimitShouldFailLargeDataPush( // Do an VPJ push runVPJ(vpjProperties, 1, controllerClient); + venice.refreshAllRouterMetaData(); // Verify some records (note, records 1-100 have been pushed) TestUtils.waitForNonDeterministicAssertion(10, TimeUnit.SECONDS, true, () -> {