fix flaky test in AdminApiOffloadTest (#11028)

(cherry picked from commit 1a80429)
apache · Dec 11, 2021 · bdb27c2 · bdb27c2
1 parent 0c48d23
commit bdb27c2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiOffloadTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiOffloadTest.java
@@ -277,6 +277,8 @@ public void testOffload(boolean isPartitioned) throws Exception {
         Map<NamespaceName, LedgerOffloader> map = new HashMap<>();
         map.put(TopicName.get(topicName).getNamespaceObject(), namespaceOffloader);
         doReturn(map).when(pulsar).getLedgerOffloaderMap();
+        doReturn(namespaceOffloader).when(pulsar)
+                .getManagedLedgerOffloader(TopicName.get(topicName).getNamespaceObject(), null);
 
         admin.topics().removeOffloadPolicies(topicName);
         Awaitility.await().atMost(10, TimeUnit.SECONDS).untilAsserted(()