Fix test

linkedin · Oct 30, 2023 · 92244e1 · 92244e1
1 parent 73ed929
commit 92244e1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/internal/venice-common/src/test/java/com/linkedin/venice/kafka/TopicManagerTest.java b/internal/venice-common/src/test/java/com/linkedin/venice/kafka/TopicManagerTest.java
@@ -518,8 +518,7 @@ public void testTimeoutOnGettingMaxOffset() throws IOException {
     PubSubTopicPartition pubSubTopicPartition = new PubSubTopicPartitionImpl(topic, 0);
     // Mock an admin client to pass topic existence check
     PubSubAdminAdapter mockPubSubAdminAdapter = mock(PubSubAdminAdapter.class);
-    doReturn(true).when(mockPubSubAdminAdapter)
-        .containsTopicWithPartitionCheckExpectationAndRetry(eq(pubSubTopicPartition), anyInt(), eq(true));
+    doReturn(true).when(mockPubSubAdminAdapter).containsTopicWithExpectationAndRetry(eq(topic), anyInt(), eq(true));
     PubSubConsumerAdapter mockPubSubConsumer = mock(PubSubConsumerAdapter.class);
     doThrow(new PubSubOpTimeoutException("Timed out while fetching end offsets")).when(mockPubSubConsumer)
         .endOffsets(any(), any());