[dvc] Config flag enabled for subscribing on disk partition automatic…

…ally. Integration Test [In Writing]
linkedin · Jan 12, 2025 · 7ba15f0 · 7ba15f0
1 parent 9bfe418
commit 7ba15f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientTest.java b/...-test-common/src/integrationTest/java/com/linkedin/venice/endToEnd/DaVinciClientTest.java
@@ -1343,13 +1343,13 @@ public void testBootstrapSubscription(DaVinciConfig daVinciConfig) throws Except
       }
 
       client1.subscribe(partitions);
-      client1.unsubscribeAll();
-      assertEquals(client1.getPartitionCount(), 2);
+      assertEquals(client1.getPartitionCount(), 3);
 
       DaVinciBackend daVinciBackend = AvroGenericDaVinciClient.getBackend();
       if (daVinciBackend != null) {
         StoreBackend storeBackend = daVinciBackend.getStoreOrThrow(storeName1);
         ComplementSet<Integer> subscription = storeBackend.getSubscription();
+        subscription.removeAll(ComplementSet.wrap(partitions));
         assertTrue(subscription.isEmpty());
       }
     }