This is an automated email from the ASF dual-hosted git repository.
mmarshall pushed a commit to branch branch-2.8
in repository https://gitbox.apache.org/repos/asf/pulsar.git
The following commit(s) were added to refs/heads/branch-2.8 by this push:
new 31d16dc [Tests] Fix flaky
org.mockito.exceptions.misusing.WrongTypeOfReturnValue issue in broker tests
(#13621)
31d16dc is described below
commit 31d16dc3f8d4fa87802fbe0229503d680b585d65
Author: Lari Hotari <[email protected]>
AuthorDate: Wed Jan 5 18:05:32 2022 +0200
[Tests] Fix flaky org.mockito.exceptions.misusing.WrongTypeOfReturnValue
issue in broker tests (#13621)
- replace spying of a instance with a spy created by Mockito with given
constructor arguments
(cherry picked from commit 17a971a132eff342e065d009a9cc553004e09b06)
---
.../mledger/impl/OffloadPrefixReadTest.java | 4 +-
.../bookkeeper/test/MockedBookKeeperTestCase.java | 1 -
.../org/apache/pulsar/broker/BrokerTestUtil.java | 15 ++++++
.../broker/admin/AdminApiGetLastMessageIdTest.java | 2 +-
.../org/apache/pulsar/broker/admin/AdminTest.java | 16 +++----
.../apache/pulsar/broker/admin/NamespacesTest.java | 6 +--
.../pulsar/broker/admin/PersistentTopicsTest.java | 5 +-
.../pulsar/broker/admin/ResourceGroupsTest.java | 2 +-
.../broker/auth/MockedPulsarServiceBaseTest.java | 5 +-
.../loadbalance/LeaderElectionServiceTest.java | 5 +-
.../loadbalance/SimpleLoadManagerImplTest.java | 5 +-
.../broker/lookup/http/HttpTopicLookupv2Test.java | 10 ++--
.../OwnerShipForCurrentServerTestBase.java | 7 +--
.../PersistentDispatcherFailoverConsumerTest.java | 17 +++----
.../service/PersistentTopicConcurrentTest.java | 24 ++++------
.../pulsar/broker/service/PersistentTopicTest.java | 55 ++++++++++------------
.../pulsar/broker/service/ServerCnxTest.java | 21 +++++----
.../service/TransactionMarkerDeleteTest.java | 5 +-
.../service/persistent/MessageDuplicationTest.java | 7 +--
.../persistent/PersistentSubscriptionTest.java | 7 +--
.../broker/transaction/TransactionTestBase.java | 5 +-
.../coordinator/TransactionMetaStoreTestBase.java | 3 +-
.../apache/pulsar/broker/web/WebServiceTest.java | 4 +-
.../client/api/ClientDeduplicationFailureTest.java | 2 +-
.../client/api/DispatcherBlockConsumerTest.java | 4 +-
.../client/impl/BrokerClientIntegrationTest.java | 14 +++---
.../pulsar/client/impl/ConnectionPoolTest.java | 9 ++--
.../worker/PulsarFunctionE2ESecurityTest.java | 2 +-
.../worker/PulsarFunctionLocalRunTest.java | 2 +-
.../worker/PulsarFunctionPublishTest.java | 2 +-
.../worker/PulsarWorkerAssignmentTest.java | 2 +-
.../apache/pulsar/io/AbstractPulsarE2ETest.java | 2 +-
.../apache/pulsar/io/PulsarFunctionAdminTest.java | 2 +-
.../apache/pulsar/io/PulsarFunctionTlsTest.java | 2 +-
.../websocket/proxy/ProxyAuthenticationTest.java | 12 ++---
.../websocket/proxy/ProxyAuthorizationTest.java | 7 +--
.../websocket/proxy/ProxyConfigurationTest.java | 6 +--
.../websocket/proxy/ProxyPublishConsumeTest.java | 12 ++---
.../proxy/ProxyPublishConsumeTlsTest.java | 6 +--
.../proxy/ProxyPublishConsumeWithoutZKTest.java | 6 +--
.../proxy/v1/V1_ProxyAuthenticationTest.java | 8 +---
.../pulsar/client/api/MessageRouterTest.java | 4 +-
.../worker/FunctionRuntimeManagerTest.java | 4 +-
.../pulsar/sql/presto/TestPulsarConnector.java | 4 +-
.../pulsar/sql/presto/TestPulsarRecordCursor.java | 2 +-
.../sql/presto/decoder/AbstractDecoderTester.java | 2 +-
46 files changed, 169 insertions(+), 178 deletions(-)
diff --git
a/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/OffloadPrefixReadTest.java
b/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/OffloadPrefixReadTest.java
index 426a00d..e761b04 100644
---
a/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/OffloadPrefixReadTest.java
+++
b/managed-ledger/src/test/java/org/apache/bookkeeper/mledger/impl/OffloadPrefixReadTest.java
@@ -66,7 +66,7 @@ import org.testng.annotations.Test;
public class OffloadPrefixReadTest extends MockedBookKeeperTestCase {
@Test
public void testOffloadRead() throws Exception {
- MockLedgerOffloader offloader = spy(new MockLedgerOffloader());
+ MockLedgerOffloader offloader = spy(MockLedgerOffloader.class);
ManagedLedgerConfig config = new ManagedLedgerConfig();
config.setMaxEntriesPerLedger(10);
config.setMinimumRolloverTime(0, TimeUnit.SECONDS);
@@ -122,7 +122,7 @@ public class OffloadPrefixReadTest extends
MockedBookKeeperTestCase {
@Test
public void testBookkeeperFirstOffloadRead() throws Exception {
- MockLedgerOffloader offloader = spy(new MockLedgerOffloader());
+ MockLedgerOffloader offloader = spy(MockLedgerOffloader.class);
MockClock clock = new MockClock();
offloader.getOffloadPolicies()
.setManagedLedgerOffloadedReadPriority(OffloadedReadPriority.BOOKKEEPER_FIRST);
diff --git
a/managed-ledger/src/test/java/org/apache/bookkeeper/test/MockedBookKeeperTestCase.java
b/managed-ledger/src/test/java/org/apache/bookkeeper/test/MockedBookKeeperTestCase.java
index a936165..5d2470c 100644
---
a/managed-ledger/src/test/java/org/apache/bookkeeper/test/MockedBookKeeperTestCase.java
+++
b/managed-ledger/src/test/java/org/apache/bookkeeper/test/MockedBookKeeperTestCase.java
@@ -22,7 +22,6 @@ import java.lang.reflect.Method;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
-
import org.apache.bookkeeper.client.PulsarMockBookKeeper;
import org.apache.bookkeeper.common.util.OrderedScheduler;
import org.apache.bookkeeper.mledger.impl.ManagedLedgerFactoryImpl;
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/BrokerTestUtil.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/BrokerTestUtil.java
index ff03e42..224060c 100644
--- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/BrokerTestUtil.java
+++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/BrokerTestUtil.java
@@ -19,6 +19,7 @@
package org.apache.pulsar.broker;
import java.util.UUID;
+import org.mockito.Mockito;
/**
* Holds util methods used in test.
@@ -29,4 +30,18 @@ public class BrokerTestUtil {
return prefix + "-" + UUID.randomUUID();
}
+ /**
+ * Creates a Mockito spy directly without an intermediate instance to spy.
+ * This is to address flaky test issue where a spy created with a given
instance fails with
+ * {@link org.mockito.exceptions.misusing.WrongTypeOfReturnValue}
exception.
+ *
+ * @param classToSpy the class to spy
+ * @param args the constructor arguments to use when creating the spy
instance
+ * @return a spy of the provided class created with given constructor
arguments
+ */
+ public static <T> T spyWithClassAndConstructorArgs(Class<T> classToSpy,
Object... args) {
+ return Mockito.mock(classToSpy, Mockito.withSettings()
+ .useConstructor(args)
+ .defaultAnswer(Mockito.CALLS_REAL_METHODS));
+ }
}
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiGetLastMessageIdTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiGetLastMessageIdTest.java
index 533a2a5..cb2a8f1 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiGetLastMessageIdTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminApiGetLastMessageIdTest.java
@@ -83,7 +83,7 @@ public class AdminApiGetLastMessageIdTest extends
MockedPulsarServiceBaseTest {
new TenantInfoImpl(Sets.newHashSet("appid1"),
Sets.newHashSet("test")));
admin.namespaces().createNamespace("prop/ns-abc");
admin.namespaces().setNamespaceReplicationClusters("prop/ns-abc",
Sets.newHashSet("test"));
- persistentTopics = spy(new PersistentTopics());
+ persistentTopics = spy(PersistentTopics.class);
persistentTopics.setServletContext(new MockServletContext());
persistentTopics.setPulsar(pulsar);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminTest.java
index f807016..c19aa86 100644
--- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminTest.java
+++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/AdminTest.java
@@ -131,17 +131,17 @@ public class AdminTest extends
MockedPulsarServiceBaseTest {
configurationCache = pulsar.getConfigurationCache();
- clusters = spy(new Clusters());
+ clusters = spy(Clusters.class);
clusters.setPulsar(pulsar);
doReturn("test").when(clusters).clientAppId();
doNothing().when(clusters).validateSuperUserAccess();
- properties = spy(new Properties());
+ properties = spy(Properties.class);
properties.setPulsar(pulsar);
doReturn("test").when(properties).clientAppId();
doNothing().when(properties).validateSuperUserAccess();
- namespaces = spy(new Namespaces());
+ namespaces = spy(Namespaces.class);
namespaces.setServletContext(new MockServletContext());
namespaces.setPulsar(pulsar);
doReturn(mockZooKeeper).when(namespaces).localZk();
@@ -151,7 +151,7 @@ public class AdminTest extends MockedPulsarServiceBaseTest {
doNothing().when(namespaces).validateAdminAccessForTenant("other-tenant");
doNothing().when(namespaces).validateAdminAccessForTenant("new-property");
- brokers = spy(new Brokers());
+ brokers = spy(Brokers.class);
brokers.setPulsar(pulsar);
doReturn("test").when(brokers).clientAppId();
doNothing().when(brokers).validateSuperUserAccess();
@@ -159,7 +159,7 @@ public class AdminTest extends MockedPulsarServiceBaseTest {
uriField = PulsarWebResource.class.getDeclaredField("uri");
uriField.setAccessible(true);
- persistentTopics = spy(new PersistentTopics());
+ persistentTopics = spy(PersistentTopics.class);
persistentTopics.setServletContext(new MockServletContext());
persistentTopics.setPulsar(pulsar);
doReturn(mockZooKeeper).when(persistentTopics).localZk();
@@ -170,11 +170,11 @@ public class AdminTest extends
MockedPulsarServiceBaseTest {
doNothing().when(persistentTopics).validateAdminAccessForTenant("other-tenant");
doNothing().when(persistentTopics).validateAdminAccessForTenant("prop-xyz");
- resourceQuotas = spy(new ResourceQuotas());
+ resourceQuotas = spy(ResourceQuotas.class);
resourceQuotas.setServletContext(new MockServletContext());
resourceQuotas.setPulsar(pulsar);
- brokerStats = spy(new BrokerStats());
+ brokerStats = spy(BrokerStats.class);
brokerStats.setServletContext(new MockServletContext());
brokerStats.setPulsar(pulsar);
@@ -183,7 +183,7 @@ public class AdminTest extends MockedPulsarServiceBaseTest {
doReturn("test").when(persistentTopics).clientAppId();
doReturn(mock(AuthenticationDataHttps.class)).when(persistentTopics).clientAuthData();
- schemasResource = spy(new SchemasResource());
+ schemasResource = spy(SchemasResource.class);
schemasResource.setServletContext(new MockServletContext());
schemasResource.setPulsar(pulsar);
}
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/NamespacesTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/NamespacesTest.java
index 1f840b1..ed04b50 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/NamespacesTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/NamespacesTest.java
@@ -158,7 +158,7 @@ public class NamespacesTest extends
MockedPulsarServiceBaseTest {
conf.setClusterName(testLocalCluster);
super.internalSetup();
- namespaces = spy(new Namespaces());
+ namespaces = spy(Namespaces.class);
namespaces.setServletContext(new MockServletContext());
namespaces.setPulsar(pulsar);
doReturn(mockZooKeeper).when(namespaces).localZk();
@@ -1102,7 +1102,7 @@ public class NamespacesTest extends
MockedPulsarServiceBaseTest {
ownership.setAccessible(true);
ownership.set(pulsar.getNamespaceService(), MockOwnershipCache);
TopicName topicName =
TopicName.get(testNs.getPersistentTopicName("my-topic"));
- PersistentTopics topics = spy(new PersistentTopics());
+ PersistentTopics topics = spy(PersistentTopics.class);
topics.setServletContext(new MockServletContext());
topics.setPulsar(pulsar);
doReturn(false).when(topics).isRequestHttps();
@@ -1267,7 +1267,7 @@ public class NamespacesTest extends
MockedPulsarServiceBaseTest {
admin.tenants().deleteTenant("my-tenants");
}
- class MockLedgerOffloader implements LedgerOffloader {
+ public static class MockLedgerOffloader implements LedgerOffloader {
ConcurrentHashMap<Long, UUID> offloads = new ConcurrentHashMap<Long,
UUID>();
ConcurrentHashMap<Long, UUID> deletes = new ConcurrentHashMap<Long,
UUID>();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/PersistentTopicsTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/PersistentTopicsTest.java
index 5bc55a3..8cb5b76 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/PersistentTopicsTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/PersistentTopicsTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.admin;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyString;
@@ -110,7 +111,7 @@ public class PersistentTopicsTest extends
MockedPulsarServiceBaseTest {
@BeforeMethod
protected void setup() throws Exception {
super.internalSetup();
- persistentTopics = spy(new PersistentTopics());
+ persistentTopics = spy(PersistentTopics.class);
persistentTopics.setServletContext(new MockServletContext());
persistentTopics.setPulsar(pulsar);
doReturn(mockZooKeeper).when(persistentTopics).localZk();
@@ -121,7 +122,7 @@ public class PersistentTopicsTest extends
MockedPulsarServiceBaseTest {
doNothing().when(persistentTopics).validateAdminAccessForTenant(this.testTenant);
doReturn(mock(AuthenticationDataHttps.class)).when(persistentTopics).clientAuthData();
- nonPersistentTopic = spy(new NonPersistentTopics());
+ nonPersistentTopic = spy(NonPersistentTopics.class);
nonPersistentTopic.setServletContext(new MockServletContext());
nonPersistentTopic.setPulsar(pulsar);
doReturn(mockZooKeeper).when(nonPersistentTopic).localZk();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/ResourceGroupsTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/ResourceGroupsTest.java
index 40d82ca..bf02e00 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/ResourceGroupsTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/admin/ResourceGroupsTest.java
@@ -52,7 +52,7 @@ public class ResourceGroupsTest extends
MockedPulsarServiceBaseTest {
@Override
protected void setup() throws Exception {
super.internalSetup();
- resourcegroups = spy(new ResourceGroups());
+ resourcegroups = spy(ResourceGroups.class);
resourcegroups.setServletContext(new MockServletContext());
resourcegroups.setPulsar(pulsar);
doReturn(mockZooKeeper).when(resourcegroups).localZk();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
index 0726a5d..a982831 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/auth/MockedPulsarServiceBaseTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.auth;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
@@ -316,7 +317,7 @@ public abstract class MockedPulsarServiceBaseTest extends
TestRetrySupport {
doReturn(createLocalMetadataStore()).when(pulsar).createLocalMetadataStore();
doReturn(createConfigurationMetadataStore()).when(pulsar).createConfigurationMetadataStore();
- Supplier<NamespaceService> namespaceServiceSupplier = () -> spy(new
NamespaceService(pulsar));
+ Supplier<NamespaceService> namespaceServiceSupplier = () ->
spyWithClassAndConstructorArgs(NamespaceService.class, pulsar);
doReturn(namespaceServiceSupplier).when(pulsar).getNamespaceServiceProvider();
doReturn(sameThreadOrderedSafeExecutor).when(pulsar).getOrderedExecutor();
@@ -380,7 +381,7 @@ public abstract class MockedPulsarServiceBaseTest extends
TestRetrySupport {
}
public static NonClosableMockBookKeeper
createMockBookKeeper(OrderedExecutor executor) throws Exception {
- return spy(new NonClosableMockBookKeeper(executor));
+ return spyWithClassAndConstructorArgs(NonClosableMockBookKeeper.class,
executor);
}
// Prevent the MockBookKeeper instance from being closed when the broker
is restarted within a test
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/LeaderElectionServiceTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/LeaderElectionServiceTest.java
index faf5125..5559a61 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/LeaderElectionServiceTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/LeaderElectionServiceTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.loadbalance;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import com.google.common.collect.Sets;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
@@ -75,7 +76,7 @@ public class LeaderElectionServiceTest {
config.setAdvertisedAddress("localhost");
config.setZookeeperServers("127.0.0.1" + ":" +
bkEnsemble.getZookeeperPort());
@Cleanup
- PulsarService pulsar = Mockito.spy(new MockPulsarService(config));
+ PulsarService pulsar =
spyWithClassAndConstructorArgs(MockPulsarService.class, config);
pulsar.start();
// mock pulsar.getLeaderElectionService() in a thread safe way
@@ -133,7 +134,7 @@ public class LeaderElectionServiceTest {
}
}
- private static class MockPulsarService extends PulsarService {
+ public static class MockPulsarService extends PulsarService {
public MockPulsarService(ServiceConfiguration config) {
super(config);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/SimpleLoadManagerImplTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/SimpleLoadManagerImplTest.java
index 10ddc23..72af7b8 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/SimpleLoadManagerImplTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/loadbalance/SimpleLoadManagerImplTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.loadbalance;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
@@ -121,7 +122,7 @@ public class SimpleLoadManagerImplTest {
bkEnsemble.start();
// Start broker 1
- ServiceConfiguration config1 = spy(new ServiceConfiguration());
+ ServiceConfiguration config1 = spy(ServiceConfiguration.class);
config1.setClusterName("use");
config1.setWebServicePort(Optional.of(0));
config1.setZookeeperServers("127.0.0.1" + ":" +
bkEnsemble.getZookeeperPort());
@@ -379,7 +380,7 @@ public class SimpleLoadManagerImplTest {
@Test(enabled = true)
public void testDoLoadShedding() throws Exception {
- SimpleLoadManagerImpl loadManager = spy(new
SimpleLoadManagerImpl(pulsar1));
+ SimpleLoadManagerImpl loadManager =
spyWithClassAndConstructorArgs(SimpleLoadManagerImpl.class, pulsar1);
PulsarResourceDescription rd = new PulsarResourceDescription();
rd.put("memory", new ResourceUsage(1024, 4096));
rd.put("cpu", new ResourceUsage(10, 100));
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/HttpTopicLookupv2Test.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/HttpTopicLookupv2Test.java
index 0f2977e..bc33132 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/HttpTopicLookupv2Test.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/HttpTopicLookupv2Test.java
@@ -88,7 +88,7 @@ public class HttpTopicLookupv2Test {
clustersListCache = mock(ZooKeeperChildrenCache.class);
clustersCache = mock(ZooKeeperDataCache.class);
policiesCache = mock(ZooKeeperDataCache.class);
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
clusters = new TreeSet<>();
clusters.add("use");
@@ -119,7 +119,7 @@ public class HttpTopicLookupv2Test {
@Test
public void crossColoLookup() throws Exception {
- TopicLookup destLookup = spy(new TopicLookup());
+ TopicLookup destLookup = spy(TopicLookup.class);
doReturn(false).when(destLookup).isRequestHttps();
destLookup.setPulsar(pulsar);
doReturn("null").when(destLookup).clientAppId();
@@ -141,7 +141,7 @@ public class HttpTopicLookupv2Test {
WebApplicationException wae = (WebApplicationException) arg.getValue();
assertEquals(wae.getResponse().getStatus(),
Status.TEMPORARY_REDIRECT.getStatusCode());
}
-
+
@Test
public void testNotEnoughLookupPermits() throws Exception {
@@ -149,7 +149,7 @@ public class HttpTopicLookupv2Test {
BrokerService brokerService = pulsar.getBrokerService();
doReturn(new
Semaphore(0)).when(brokerService).getLookupRequestSemaphore();
- TopicLookup destLookup = spy(new TopicLookup());
+ TopicLookup destLookup = spy(TopicLookup.class);
doReturn(false).when(destLookup).isRequestHttps();
destLookup.setPulsar(pulsar);
doReturn("null").when(destLookup).clientAppId();
@@ -187,7 +187,7 @@ public class HttpTopicLookupv2Test {
doReturn(Optional.of(policies2)).when(policiesCache)
.get(AdminResource.path(POLICIES, property, cluster, ns2));
- TopicLookup destLookup = spy(new TopicLookup());
+ TopicLookup destLookup = spy(TopicLookup.class);
doReturn(false).when(destLookup).isRequestHttps();
destLookup.setPulsar(pulsar);
doReturn("null").when(destLookup).clientAppId();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/namespace/OwnerShipForCurrentServerTestBase.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/namespace/OwnerShipForCurrentServerTestBase.java
index 2efbb20..e8cccc4 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/namespace/OwnerShipForCurrentServerTestBase.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/namespace/OwnerShipForCurrentServerTestBase.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.namespace;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import com.google.common.util.concurrent.MoreExecutors;
import com.google.common.util.concurrent.ThreadFactoryBuilder;
import io.netty.channel.EventLoopGroup;
@@ -120,7 +121,7 @@ public class OwnerShipForCurrentServerTestBase {
conf.setWebServicePortTls(Optional.of(0));
serviceConfigurationList.add(conf);
- PulsarService pulsar = spy(new PulsarService(conf));
+ PulsarService pulsar =
spyWithClassAndConstructorArgs(PulsarService.class, conf);
setupBrokerMocks(pulsar);
pulsar.start();
@@ -135,7 +136,7 @@ public class OwnerShipForCurrentServerTestBase {
MockZooKeeperSession mockZooKeeperSession =
MockZooKeeperSession.newInstance(mockZooKeeper);
doReturn(new
ZKMetadataStore(mockZooKeeperSession)).when(pulsar).createLocalMetadataStore();
doReturn(new
ZKMetadataStore(mockZooKeeperSession)).when(pulsar).createConfigurationMetadataStore();
- Supplier<NamespaceService> namespaceServiceSupplier = () -> spy(new
NamespaceService(pulsar));
+ Supplier<NamespaceService> namespaceServiceSupplier = () ->
spyWithClassAndConstructorArgs(NamespaceService.class, pulsar);
doReturn(namespaceServiceSupplier).when(pulsar).getNamespaceServiceProvider();
SameThreadOrderedSafeExecutor executor = new
SameThreadOrderedSafeExecutor();
@@ -159,7 +160,7 @@ public class OwnerShipForCurrentServerTestBase {
}
public static NonClosableMockBookKeeper
createMockBookKeeper(OrderedExecutor executor) throws Exception {
- return spy(new NonClosableMockBookKeeper(executor));
+ return spyWithClassAndConstructorArgs(NonClosableMockBookKeeper.class,
executor);
}
// Prevent the MockBookKeeper instance from being closed when the broker
is restarted within a test
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentDispatcherFailoverConsumerTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentDispatcherFailoverConsumerTest.java
index 4b947aa..8a3870b 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentDispatcherFailoverConsumerTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentDispatcherFailoverConsumerTest.java
@@ -18,10 +18,10 @@
*/
package org.apache.pulsar.broker.service;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.matches;
import static org.mockito.ArgumentMatchers.same;
-import static org.mockito.Mockito.CALLS_REAL_METHODS;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
@@ -29,7 +29,6 @@ import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
-import static org.mockito.Mockito.withSettings;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNull;
import static org.testng.AssertJUnit.assertEquals;
@@ -119,9 +118,9 @@ public class PersistentDispatcherFailoverConsumerTest {
@BeforeMethod
public void setup() throws Exception {
executor =
OrderedExecutor.newBuilder().numThreads(1).name("persistent-dispatcher-failover-test").build();
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
svcConfig.setBrokerShutdownTimeoutMs(0L);
- pulsar = spy(new PulsarService(svcConfig));
+ pulsar = spyWithClassAndConstructorArgs(PulsarService.class,
svcConfig);
doReturn(svcConfig).when(pulsar).getConfiguration();
mlFactoryMock = mock(ManagedLedgerFactory.class);
@@ -145,7 +144,7 @@ public class PersistentDispatcherFailoverConsumerTest {
doReturn(configCacheService).when(pulsar).getConfigurationCache();
doReturn(zkCache).when(pulsar).getLocalZkCacheService();
- brokerService = spy(new BrokerService(pulsar, eventLoopGroup));
+ brokerService = spyWithClassAndConstructorArgs(BrokerService.class,
pulsar, eventLoopGroup);
doReturn(brokerService).when(pulsar).getBrokerService();
consumerChanges = new LinkedBlockingQueue<>();
@@ -171,9 +170,7 @@ public class PersistentDispatcherFailoverConsumerTest {
return null;
}).when(channelCtx).writeAndFlush(any(), any());
- serverCnx = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ serverCnx = spyWithClassAndConstructorArgs(ServerCnx.class, pulsar);
doReturn(true).when(serverCnx).isActive();
doReturn(true).when(serverCnx).isWritable();
doReturn(new InetSocketAddress("localhost",
1234)).when(serverCnx).clientAddress();
@@ -182,9 +179,7 @@ public class PersistentDispatcherFailoverConsumerTest {
doReturn(new PulsarCommandSenderImpl(null, serverCnx))
.when(serverCnx).getCommandSender();
- serverCnxWithOldVersion = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ serverCnxWithOldVersion =
spyWithClassAndConstructorArgs(ServerCnx.class, pulsar);
doReturn(true).when(serverCnxWithOldVersion).isActive();
doReturn(true).when(serverCnxWithOldVersion).isWritable();
doReturn(new InetSocketAddress("localhost", 1234))
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicConcurrentTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicConcurrentTest.java
index 7ba35b9..d9e5837 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicConcurrentTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicConcurrentTest.java
@@ -18,14 +18,15 @@
*/
package org.apache.pulsar.broker.service;
-import static org.mockito.Mockito.CALLS_REAL_METHODS;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.withSettings;
import static org.testng.Assert.assertFalse;
-
+import com.google.common.collect.Lists;
+import io.netty.channel.EventLoopGroup;
+import io.netty.channel.nio.NioEventLoopGroup;
import java.lang.reflect.Method;
import java.util.Collections;
import java.util.List;
@@ -34,9 +35,6 @@ import java.util.concurrent.CountDownLatch;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicBoolean;
-
-import io.netty.channel.EventLoopGroup;
-import io.netty.channel.nio.NioEventLoopGroup;
import lombok.Cleanup;
import org.apache.bookkeeper.mledger.ManagedCursor;
import org.apache.bookkeeper.mledger.ManagedLedger;
@@ -53,16 +51,14 @@ import
org.apache.pulsar.broker.service.persistent.PersistentTopic;
import org.apache.pulsar.broker.transaction.TransactionTestBase;
import org.apache.pulsar.common.api.proto.CommandSubscribe;
import org.apache.pulsar.common.api.proto.CommandSubscribe.InitialPosition;
-import org.apache.pulsar.common.naming.TopicName;
import org.apache.pulsar.common.naming.NamespaceBundle;
+import org.apache.pulsar.common.naming.TopicName;
import org.apache.pulsar.common.policies.data.InactiveTopicDeleteMode;
import org.apache.pulsar.common.util.collections.ConcurrentOpenHashMap;
import org.apache.zookeeper.ZooKeeper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.testng.annotations.BeforeMethod;
-
-import com.google.common.collect.Lists;
import org.testng.annotations.Test;
@Test(groups = "broker")
@@ -83,10 +79,10 @@ public class PersistentTopicConcurrentTest extends
MockedBookKeeperTestCase {
@BeforeMethod
public void setup(Method m) throws Exception {
super.setUp(m);
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
svcConfig.setBrokerShutdownTimeoutMs(0L);
@Cleanup
- PulsarService pulsar = spy(new PulsarService(svcConfig));
+ PulsarService pulsar =
spyWithClassAndConstructorArgs(PulsarService.class, svcConfig);
doReturn(svcConfig).when(pulsar).getConfiguration();
@Cleanup(value = "shutdownGracefully")
@@ -103,12 +99,10 @@ public class PersistentTopicConcurrentTest extends
MockedBookKeeperTestCase {
ZooKeeper mockZk = TransactionTestBase.createMockZooKeeper();
doReturn(mockZk).when(pulsar).getZkClient();
- brokerService = spy(new BrokerService(pulsar, eventLoopGroup));
+ brokerService = spyWithClassAndConstructorArgs(BrokerService.class,
pulsar, eventLoopGroup);
doReturn(brokerService).when(pulsar).getBrokerService();
- serverCnx = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ serverCnx = spyWithClassAndConstructorArgs(ServerCnx.class, pulsar);
doReturn(true).when(serverCnx).isActive();
NamespaceService nsSvc = mock(NamespaceService.class);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicTest.java
index 0a6c6b4..0480fa0 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/PersistentTopicTest.java
@@ -18,12 +18,12 @@
*/
package org.apache.pulsar.broker.service;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockBookKeeper;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockZooKeeper;
import static
org.apache.pulsar.broker.cache.ConfigurationCacheService.POLICIES;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyLong;
-import static org.mockito.Mockito.CALLS_REAL_METHODS;
import static org.mockito.Mockito.anyString;
import static org.mockito.Mockito.atLeast;
import static org.mockito.Mockito.doAnswer;
@@ -36,7 +36,6 @@ import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
-import static org.mockito.Mockito.withSettings;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNull;
@@ -181,10 +180,10 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
public void setup() throws Exception {
eventLoopGroup = new NioEventLoopGroup();
executor = OrderedExecutor.newBuilder().numThreads(1).build();
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
svcConfig.setAdvertisedAddress("localhost");
svcConfig.setBrokerShutdownTimeoutMs(0L);
- pulsar = spy(new PulsarService(svcConfig));
+ pulsar = spyWithClassAndConstructorArgs(PulsarService.class,
svcConfig);
doReturn(svcConfig).when(pulsar).getConfiguration();
doReturn(mock(Compactor.class)).when(pulsar).getCompactor();
@@ -221,17 +220,15 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
doReturn(executor).when(pulsar).getOrderedExecutor();
store = new ZKMetadataStore(mockZk);
- PulsarResources pulsarResources = spy(new PulsarResources(store,
store));
- NamespaceResources nsr = spy(new NamespaceResources(store, 30));
+ PulsarResources pulsarResources =
spyWithClassAndConstructorArgs(PulsarResources.class, store, store);
+ NamespaceResources nsr =
spyWithClassAndConstructorArgs(NamespaceResources.class, store, 30);
doReturn(nsr).when(pulsarResources).getNamespaceResources();
doReturn(pulsarResources).when(pulsar).getPulsarResources();
- brokerService = spy(new BrokerService(pulsar, eventLoopGroup));
+ brokerService = spyWithClassAndConstructorArgs(BrokerService.class,
pulsar, eventLoopGroup);
doReturn(brokerService).when(pulsar).getBrokerService();
- serverCnx = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ serverCnx = spyWithClassAndConstructorArgs(ServerCnx.class, pulsar);
doReturn(true).when(serverCnx).isActive();
doReturn(true).when(serverCnx).isWritable();
doReturn(new InetSocketAddress("localhost",
1234)).when(serverCnx).clientAddress();
@@ -372,7 +369,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testDispatcherMultiConsumerReadFailed() throws Exception {
- PersistentTopic topic = spy(new PersistentTopic(successTopicName,
ledgerMock, brokerService));
+ PersistentTopic topic =
spyWithClassAndConstructorArgs(PersistentTopic.class, successTopicName,
ledgerMock, brokerService);
ManagedCursor cursor = mock(ManagedCursor.class);
when(cursor.getName()).thenReturn("cursor");
PersistentDispatcherMultipleConsumers dispatcher = new
PersistentDispatcherMultipleConsumers(topic, cursor, null);
@@ -382,7 +379,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testDispatcherSingleConsumerReadFailed() throws Exception {
- PersistentTopic topic = spy(new PersistentTopic(successTopicName,
ledgerMock, brokerService));
+ PersistentTopic topic =
spyWithClassAndConstructorArgs(PersistentTopic.class, successTopicName,
ledgerMock, brokerService);
ManagedCursor cursor = mock(ManagedCursor.class);
when(cursor.getName()).thenReturn("cursor");
PersistentDispatcherSingleActiveConsumer dispatcher = new
PersistentDispatcherSingleActiveConsumer(cursor,
@@ -579,7 +576,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxProducersForBroker() throws Exception {
// set max clients
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(2).when(svcConfig).getMaxProducersPerTopic();
doReturn(svcConfig).when(pulsar).getConfiguration();
testMaxProducers();
@@ -587,7 +584,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxProducersForNamespace() throws Exception {
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(svcConfig).when(pulsar).getConfiguration();
// set max clients
Policies policies = new Policies();
@@ -603,9 +600,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
final String producerNameBase = "producer";
final String role = "appid1";
- ServerCnx cnx = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ ServerCnx cnx = spyWithClassAndConstructorArgs(ServerCnx.class,
pulsar);
doReturn(true).when(cnx).isActive();
doReturn(true).when(cnx).isWritable();
doReturn(new InetSocketAddress(address,
1234)).when(cnx).clientAddress();
@@ -619,7 +614,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxSameAddressProducers() throws Exception {
// set max clients
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(2).when(svcConfig).getMaxSameAddressProducersPerTopic();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -912,7 +907,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxConsumersSharedForBroker() throws Exception {
// set max clients
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(2).when(svcConfig).getMaxConsumersPerSubscription();
doReturn(3).when(svcConfig).getMaxConsumersPerTopic();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -922,7 +917,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxConsumersSharedForNamespace() throws Exception {
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(svcConfig).when(pulsar).getConfiguration();
// set max clients
@@ -1007,7 +1002,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxConsumersFailoverForBroker() throws Exception {
// set max clients
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(2).when(svcConfig).getMaxConsumersPerSubscription();
doReturn(3).when(svcConfig).getMaxConsumersPerTopic();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -1017,7 +1012,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxConsumersFailoverForNamespace() throws Exception {
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(svcConfig).when(pulsar).getConfiguration();
// set max clients
@@ -1036,9 +1031,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
final String consumerNameBase = "consumer";
final String role = "appid1";
- ServerCnx cnx = mock(ServerCnx.class, withSettings()
- .useConstructor(pulsar)
- .defaultAnswer(CALLS_REAL_METHODS));
+ ServerCnx cnx = spyWithClassAndConstructorArgs(ServerCnx.class,
pulsar);
doReturn(true).when(cnx).isActive();
doReturn(true).when(cnx).isWritable();
doReturn(new InetSocketAddress(address,
1234)).when(cnx).clientAddress();
@@ -1052,7 +1045,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testMaxSameAddressConsumers() throws Exception {
// set max clients
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(2).when(svcConfig).getMaxSameAddressConsumersPerTopic();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -2090,13 +2083,13 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
ConcurrentOpenHashMap<String, PersistentSubscription> subscriptions =
new ConcurrentOpenHashMap<>(16, 1);
// This subscription is connected by consumer.
- PersistentSubscription nonDeletableSubscription1 = spy(new
PersistentSubscription(topic, "nonDeletableSubscription1", cursorMock, false));
+ PersistentSubscription nonDeletableSubscription1 =
spyWithClassAndConstructorArgs(PersistentSubscription.class, topic,
"nonDeletableSubscription1", cursorMock, false);
subscriptions.put(nonDeletableSubscription1.getName(),
nonDeletableSubscription1);
// This subscription is not connected by consumer.
- PersistentSubscription deletableSubscription1 = spy(new
PersistentSubscription(topic, "deletableSubscription1", cursorMock, false));
+ PersistentSubscription deletableSubscription1 =
spyWithClassAndConstructorArgs(PersistentSubscription.class, topic,
"deletableSubscription1", cursorMock, false);
subscriptions.put(deletableSubscription1.getName(),
deletableSubscription1);
// This subscription is replicated.
- PersistentSubscription nonDeletableSubscription2 = spy(new
PersistentSubscription(topic, "nonDeletableSubscription2", cursorMock, true));
+ PersistentSubscription nonDeletableSubscription2 =
spyWithClassAndConstructorArgs(PersistentSubscription.class, topic,
"nonDeletableSubscription2", cursorMock, true);
subscriptions.put(nonDeletableSubscription2.getName(),
nonDeletableSubscription2);
Field field = topic.getClass().getDeclaredField("subscriptions");
@@ -2115,7 +2108,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
.get(AdminResource.path(POLICIES,
TopicName.get(successTopicName).getNamespace())))
.thenReturn(Optional.of(new Policies()));
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(5).when(svcConfig).getSubscriptionExpirationTimeMinutes();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -2130,7 +2123,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
@Test
public void testTopicFencingTimeout() throws Exception {
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(svcConfig).when(pulsar).getConfiguration();
PersistentTopic topic = new PersistentTopic(successTopicName,
ledgerMock, brokerService);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ServerCnxTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ServerCnxTest.java
index 63347a3..ff9c188 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ServerCnxTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/ServerCnxTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.service;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockBookKeeper;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockZooKeeper;
import static
org.apache.pulsar.broker.cache.ConfigurationCacheService.POLICIES;
@@ -155,9 +156,9 @@ public class ServerCnxTest {
public void setup() throws Exception {
eventLoopGroup = new NioEventLoopGroup();
executor = OrderedExecutor.newBuilder().numThreads(1).build();
- svcConfig = spy(new ServiceConfiguration());
+ svcConfig = spy(ServiceConfiguration.class);
svcConfig.setBrokerShutdownTimeoutMs(0L);
- pulsar = spy(new PulsarService(svcConfig));
+ pulsar = spyWithClassAndConstructorArgs(PulsarService.class,
svcConfig);
doReturn(new
DefaultSchemaRegistryService()).when(pulsar).getSchemaRegistryService();
svcConfig.setKeepAliveIntervalSeconds(inSec(1, TimeUnit.SECONDS));
@@ -189,7 +190,7 @@ public class ServerCnxTest {
doReturn(configCacheService).when(pulsar).getConfigurationCache();
doReturn(zkCache).when(pulsar).getLocalZkCacheService();
- brokerService = spy(new BrokerService(pulsar, eventLoopGroup));
+ brokerService = spyWithClassAndConstructorArgs(BrokerService.class,
pulsar, eventLoopGroup);
BrokerInterceptor interceptor = mock(BrokerInterceptor.class);
doReturn(interceptor).when(brokerService).getInterceptor();
doReturn(brokerService).when(pulsar).getBrokerService();
@@ -515,13 +516,14 @@ public class ServerCnxTest {
doReturn(zkDataCache).when(configCacheService).policiesCache();
doReturn(CompletableFuture.completedFuture(Optional.empty())).when(zkDataCache).getAsync(matches(".*nonexistent.*"));
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, configCacheService));
+ AuthorizationService authorizationService =
spyWithClassAndConstructorArgs(AuthorizationService.class, svcConfig,
configCacheService);
doReturn(authorizationService).when(brokerService).getAuthorizationService();
doReturn(true).when(brokerService).isAuthorizationEnabled();
svcConfig.setAuthorizationEnabled(true);
Field providerField =
AuthorizationService.class.getDeclaredField("provider");
providerField.setAccessible(true);
- PulsarAuthorizationProvider authorizationProvider = spy(new
PulsarAuthorizationProvider(svcConfig, configCacheService));
+ PulsarAuthorizationProvider authorizationProvider =
spyWithClassAndConstructorArgs(PulsarAuthorizationProvider.class, svcConfig,
+ configCacheService);
providerField.set(authorizationService, authorizationProvider);
doReturn(CompletableFuture.completedFuture(false)).when(authorizationProvider).isSuperUser(Mockito.anyString(),
Mockito.any(), Mockito.any());
@@ -547,10 +549,11 @@ public class ServerCnxTest {
@Test(timeOut = 30000)
public void testClusterAccess() throws Exception {
svcConfig.setAuthorizationEnabled(true);
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, configCacheService));
+ AuthorizationService authorizationService =
spyWithClassAndConstructorArgs(AuthorizationService.class, svcConfig,
configCacheService);
Field providerField =
AuthorizationService.class.getDeclaredField("provider");
providerField.setAccessible(true);
- PulsarAuthorizationProvider authorizationProvider = spy(new
PulsarAuthorizationProvider(svcConfig, configCacheService));
+ PulsarAuthorizationProvider authorizationProvider =
spyWithClassAndConstructorArgs(PulsarAuthorizationProvider.class, svcConfig,
+ configCacheService);
providerField.set(authorizationService, authorizationProvider);
doReturn(authorizationService).when(brokerService).getAuthorizationService();
doReturn(true).when(brokerService).isAuthorizationEnabled();
@@ -577,12 +580,12 @@ public class ServerCnxTest {
@Test(timeOut = 30000)
public void testNonExistentTopicSuperUserAccess() throws Exception {
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, configCacheService));
+ AuthorizationService authorizationService =
spyWithClassAndConstructorArgs(AuthorizationService.class, svcConfig,
configCacheService);
doReturn(authorizationService).when(brokerService).getAuthorizationService();
doReturn(true).when(brokerService).isAuthorizationEnabled();
Field providerField =
AuthorizationService.class.getDeclaredField("provider");
providerField.setAccessible(true);
- PulsarAuthorizationProvider authorizationProvider = spy(new
PulsarAuthorizationProvider(svcConfig, configCacheService));
+ PulsarAuthorizationProvider authorizationProvider =
spyWithClassAndConstructorArgs(PulsarAuthorizationProvider.class, svcConfig,
configCacheService);
providerField.set(authorizationService, authorizationProvider);
doReturn(CompletableFuture.completedFuture(true)).when(authorizationProvider).isSuperUser(Mockito.anyString(),
Mockito.any(), Mockito.any());
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/TransactionMarkerDeleteTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/TransactionMarkerDeleteTest.java
index d6e9c73..4527798 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/TransactionMarkerDeleteTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/TransactionMarkerDeleteTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.service;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
@@ -87,8 +88,8 @@ public class TransactionMarkerDeleteTest extends
BrokerTestBase{
doReturn(false).when(configuration).isTransactionCoordinatorEnabled();
doReturn(managedLedger).when(topic).getManagedLedger();
ManagedCursor cursor = managedLedger.openCursor("test");
- PersistentSubscription persistentSubscription = spy(new
PersistentSubscription(topic, "test",
- cursor, false));
+ PersistentSubscription persistentSubscription =
spyWithClassAndConstructorArgs(PersistentSubscription.class, topic, "test",
+ cursor, false);
Position position = managedLedger.addEntry("test".getBytes());
persistentSubscription.acknowledgeMessage(Collections.singletonList(position),
AckType.Individual, Collections.emptyMap());
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/MessageDuplicationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/MessageDuplicationTest.java
index 1777a0a..56f833f 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/MessageDuplicationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/MessageDuplicationTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.service.persistent;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import io.netty.buffer.ByteBuf;
import io.netty.channel.EventLoopGroup;
import lombok.extern.slf4j.Slf4j;
@@ -73,7 +74,7 @@ public class MessageDuplicationTest {
doReturn(serviceConfiguration).when(pulsarService).getConfiguration();
PersistentTopic persistentTopic = mock(PersistentTopic.class);
ManagedLedger managedLedger = mock(ManagedLedger.class);
- MessageDeduplication messageDeduplication = spy(new
MessageDeduplication(pulsarService, persistentTopic, managedLedger));
+ MessageDeduplication messageDeduplication =
spyWithClassAndConstructorArgs(MessageDeduplication.class, pulsarService,
persistentTopic, managedLedger);
doReturn(true).when(messageDeduplication).isEnabled();
String producerName1 = "producer1";
@@ -166,7 +167,7 @@ public class MessageDuplicationTest {
serviceConfiguration.setBrokerDeduplicationProducerInactivityTimeoutMinutes(1);
doReturn(serviceConfiguration).when(pulsarService).getConfiguration();
- MessageDeduplication messageDeduplication = spy(new
MessageDeduplication(pulsarService, topic, managedLedger));
+ MessageDeduplication messageDeduplication =
spyWithClassAndConstructorArgs(MessageDeduplication.class, pulsarService,
topic, managedLedger);
doReturn(true).when(messageDeduplication).isEnabled();
Topic.PublishContext publishContext = mock(Topic.PublishContext.class);
@@ -243,7 +244,7 @@ public class MessageDuplicationTest {
doReturn(eventLoopGroup).when(brokerService).executor();
doReturn(pulsarService).when(brokerService).pulsar();
- PersistentTopic persistentTopic = spy(new PersistentTopic("topic-1",
brokerService, managedLedger, messageDeduplication));
+ PersistentTopic persistentTopic =
spyWithClassAndConstructorArgs(PersistentTopic.class, "topic-1", brokerService,
managedLedger, messageDeduplication);
String producerName1 = "producer1";
ByteBuf byteBuf1 = getMessage(producerName1, 0);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/PersistentSubscriptionTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/PersistentSubscriptionTest.java
index 884d2a0..c7f068c 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/PersistentSubscriptionTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/persistent/PersistentSubscriptionTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.service.persistent;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockBookKeeper;
import static
org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest.createMockZooKeeper;
import static org.mockito.ArgumentMatchers.any;
@@ -115,10 +116,10 @@ public class PersistentSubscriptionTest {
executor =
OrderedExecutor.newBuilder().numThreads(1).name("persistent-subscription-test").build();
eventLoopGroup = new NioEventLoopGroup();
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
svcConfig.setBrokerShutdownTimeoutMs(0L);
svcConfig.setTransactionCoordinatorEnabled(true);
- pulsarMock = spy(new PulsarService(svcConfig));
+ pulsarMock = spyWithClassAndConstructorArgs(PulsarService.class,
svcConfig);
doReturn(new
InMemTransactionBufferProvider()).when(pulsarMock).getTransactionBufferProvider();
doReturn(new TransactionPendingAckStoreProvider() {
@Override
@@ -192,7 +193,7 @@ public class PersistentSubscriptionTest {
doReturn(zkPoliciesDataCacheMock).when(zkCacheMock).policiesCache();
doReturn(zkCacheMock).when(pulsarMock).getLocalZkCacheService();
- brokerMock = spy(new BrokerService(pulsarMock, eventLoopGroup));
+ brokerMock = spyWithClassAndConstructorArgs(BrokerService.class,
pulsarMock, eventLoopGroup);
doNothing().when(brokerMock).unloadNamespaceBundlesGracefully();
doReturn(brokerMock).when(pulsarMock).getBrokerService();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/TransactionTestBase.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/TransactionTestBase.java
index 0a694c8..0e24f08 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/TransactionTestBase.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/TransactionTestBase.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.transaction;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
@@ -140,7 +141,7 @@ public abstract class TransactionTestBase extends
TestRetrySupport {
conf.setTopicLevelPoliciesEnabled(true);
serviceConfigurationList.add(conf);
- PulsarService pulsar = spy(new PulsarService(conf));
+ PulsarService pulsar =
spyWithClassAndConstructorArgs(PulsarService.class, conf);
setupBrokerMocks(pulsar);
pulsar.start();
@@ -156,7 +157,7 @@ public abstract class TransactionTestBase extends
TestRetrySupport {
MockZooKeeperSession mockZooKeeperSession =
MockZooKeeperSession.newInstance(mockZooKeeper);
doReturn(new
ZKMetadataStore(mockZooKeeperSession)).when(pulsar).createLocalMetadataStore();
doReturn(new
ZKMetadataStore(mockZooKeeperSession)).when(pulsar).createConfigurationMetadataStore();
- Supplier<NamespaceService> namespaceServiceSupplier = () -> spy(new
NamespaceService(pulsar));
+ Supplier<NamespaceService> namespaceServiceSupplier = () ->
spyWithClassAndConstructorArgs(NamespaceService.class, pulsar);
doReturn(namespaceServiceSupplier).when(pulsar).getNamespaceServiceProvider();
SameThreadOrderedSafeExecutor executor = new
SameThreadOrderedSafeExecutor();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/coordinator/TransactionMetaStoreTestBase.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/coordinator/TransactionMetaStoreTestBase.java
index 0457907..aba5b04 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/coordinator/TransactionMetaStoreTestBase.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/transaction/coordinator/TransactionMetaStoreTestBase.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.broker.transaction.coordinator;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import java.util.Optional;
import org.apache.pulsar.PulsarTransactionCoordinatorMetadataSetup;
import org.apache.pulsar.broker.PulsarService;
@@ -76,7 +77,7 @@ public abstract class TransactionMetaStoreTestBase extends
TestRetrySupport {
config.setTransactionCoordinatorEnabled(true);
configurations[i] = config;
- pulsarServices[i] = Mockito.spy(new PulsarService(config));
+ pulsarServices[i] =
spyWithClassAndConstructorArgs(PulsarService.class, config);
pulsarServices[i].start();
pulsarAdmins[i] = PulsarAdmin.builder()
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/web/WebServiceTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/web/WebServiceTest.java
index 48bf602..3dacdeb 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/web/WebServiceTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/web/WebServiceTest.java
@@ -18,8 +18,8 @@
*/
package org.apache.pulsar.broker.web;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertTrue;
import static org.testng.Assert.fail;
@@ -390,7 +390,7 @@ public class WebServiceTest {
config.setHttpRequestsLimitEnabled(true);
config.setHttpRequestsMaxPerSecond(rateLimit);
}
- pulsar = spy(new PulsarService(config));
+ pulsar = spyWithClassAndConstructorArgs(PulsarService.class, config);
// mock zk
MockZooKeeper mockZooKeeper =
MockedPulsarServiceBaseTest.createMockZooKeeper();
ZooKeeperClientFactory mockZooKeeperClientFactory = new
ZooKeeperClientFactory() {
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/ClientDeduplicationFailureTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/ClientDeduplicationFailureTest.java
index 538bb89..4d7fece 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/ClientDeduplicationFailureTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/ClientDeduplicationFailureTest.java
@@ -80,7 +80,7 @@ public class ClientDeduplicationFailureTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
config.setWebServicePort(Optional.of(0));
config.setZookeeperServers("127.0.0.1" + ":" +
bkEnsemble.getZookeeperPort());
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/DispatcherBlockConsumerTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/DispatcherBlockConsumerTest.java
index 00f52ad..cb1f44b 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/DispatcherBlockConsumerTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/DispatcherBlockConsumerTest.java
@@ -18,8 +18,8 @@
*/
package org.apache.pulsar.client.api;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotEquals;
import static org.testng.Assert.assertNotNull;
@@ -630,7 +630,7 @@ public class DispatcherBlockConsumerTest extends
ProducerConsumerBase {
// if broker unload bundle gracefully then cursor metadata recovered
from zk else from ledger
if (unloadBundleGracefully) {
// set clean namespace which will not let broker unload bundle
gracefully: stop broker
- Supplier<NamespaceService> namespaceServiceSupplier = () ->
spy(new NamespaceService(pulsar));
+ Supplier<NamespaceService> namespaceServiceSupplier = () ->
spyWithClassAndConstructorArgs(NamespaceService.class, pulsar);
doReturn(namespaceServiceSupplier).when(pulsar).getNamespaceServiceProvider();
}
stopBroker();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/BrokerClientIntegrationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/BrokerClientIntegrationTest.java
index d90ecb0..084d951 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/BrokerClientIntegrationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/BrokerClientIntegrationTest.java
@@ -20,6 +20,7 @@ package org.apache.pulsar.client.impl;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.UUID.randomUUID;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static
org.apache.pulsar.broker.service.BrokerService.BROKER_SERVICE_CONFIGURATION_PATH;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.atLeastOnce;
@@ -890,8 +891,9 @@ public class BrokerClientIntegrationTest extends
ProducerConsumerBase {
public void testJsonSchemaProducerConsumerWithSpecifiedReaderAndWriter()
throws PulsarClientException {
final String topicName = "persistent://my-property/my-ns/my-topic1";
ObjectMapper mapper = new ObjectMapper();
- SchemaReader<TestMessageObject> reader = Mockito.spy(new
JacksonJsonReader<>(mapper, TestMessageObject.class));
- SchemaWriter<TestMessageObject> writer = Mockito.spy(new
JacksonJsonWriter<>(mapper));
+ SchemaReader<TestMessageObject> reader =
+ spyWithClassAndConstructorArgs(JacksonJsonReader.class,
mapper, TestMessageObject.class);
+ SchemaWriter<TestMessageObject> writer =
spyWithClassAndConstructorArgs(JacksonJsonWriter.class, mapper);
SchemaDefinition<TestMessageObject> schemaDefinition = new
SchemaDefinitionBuilderImpl<TestMessageObject>()
.withPojo(TestMessageObject.class)
@@ -926,10 +928,10 @@ public class BrokerClientIntegrationTest extends
ProducerConsumerBase {
private static final class TestMessageObject{
private String value;
}
-
+
/**
* It validates pooled message consumption for batch and non-batch
messages.
- *
+ *
* @throws Exception
*/
@Test(dataProvider = "booleanFlagProvider")
@@ -980,10 +982,10 @@ public class BrokerClientIntegrationTest extends
ProducerConsumerBase {
consumer.close();
producer.close();
}
-
+
/**
* It verifies that expiry/redelivery of messages relesaes the messages
without leak.
- *
+ *
* @param isBatchingEnabled
* @throws Exception
*/
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ConnectionPoolTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ConnectionPoolTest.java
index 1e97550..0193f59 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ConnectionPoolTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/client/impl/ConnectionPoolTest.java
@@ -18,6 +18,7 @@
*/
package org.apache.pulsar.client.impl;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import com.google.common.collect.Lists;
import io.netty.channel.EventLoopGroup;
import io.netty.util.concurrent.DefaultThreadFactory;
@@ -58,7 +59,7 @@ public class ConnectionPoolTest extends
MockedPulsarServiceBaseTest {
public void testSingleIpAddress() throws Exception {
ClientConfigurationData conf = new ClientConfigurationData();
EventLoopGroup eventLoop = EventLoopUtil.newEventLoopGroup(1, false,
new DefaultThreadFactory("test"));
- ConnectionPool pool = Mockito.spy(new ConnectionPool(conf, eventLoop));
+ ConnectionPool pool =
spyWithClassAndConstructorArgs(ConnectionPool.class, conf, eventLoop);
conf.setServiceUrl(serviceUrl);
PulsarClientImpl client = new PulsarClientImpl(conf, eventLoop, pool);
@@ -78,7 +79,7 @@ public class ConnectionPoolTest extends
MockedPulsarServiceBaseTest {
ClientConfigurationData conf = new ClientConfigurationData();
EventLoopGroup eventLoop = EventLoopUtil.newEventLoopGroup(1, false,
new DefaultThreadFactory("test"));
- ConnectionPool pool = Mockito.spy(new ConnectionPool(conf, eventLoop));
+ ConnectionPool pool =
spyWithClassAndConstructorArgs(ConnectionPool.class, conf, eventLoop);
conf.setServiceUrl(serviceUrl);
PulsarClientImpl client = new PulsarClientImpl(conf, eventLoop, pool);
@@ -101,7 +102,7 @@ public class ConnectionPoolTest extends
MockedPulsarServiceBaseTest {
ClientConfigurationData conf = new ClientConfigurationData();
conf.setConnectionsPerBroker(0);
EventLoopGroup eventLoop = EventLoopUtil.newEventLoopGroup(8, false,
new DefaultThreadFactory("test"));
- ConnectionPool pool = Mockito.spy(new ConnectionPool(conf, eventLoop));
+ ConnectionPool pool =
spyWithClassAndConstructorArgs(ConnectionPool.class, conf, eventLoop);
InetSocketAddress brokerAddress =
InetSocketAddress.createUnresolved("127.0.0.1",
pulsar.getBrokerListenPort().get());
@@ -123,7 +124,7 @@ public class ConnectionPoolTest extends
MockedPulsarServiceBaseTest {
ClientConfigurationData conf = new ClientConfigurationData();
conf.setConnectionsPerBroker(5);
EventLoopGroup eventLoop = EventLoopUtil.newEventLoopGroup(8, false,
new DefaultThreadFactory("test"));
- ConnectionPool pool = Mockito.spy(new ConnectionPool(conf, eventLoop));
+ ConnectionPool pool =
spyWithClassAndConstructorArgs(ConnectionPool.class, conf, eventLoop);
InetSocketAddress brokerAddress =
InetSocketAddress.createUnresolved("127.0.0.1",
pulsar.getBrokerListenPort().get());
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionE2ESecurityTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionE2ESecurityTest.java
index e7173a2..c399cd8 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionE2ESecurityTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionE2ESecurityTest.java
@@ -125,7 +125,7 @@ public class PulsarFunctionE2ESecurityTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
Set<String> superUsers = Sets.newHashSet(ADMIN_SUBJECT);
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionLocalRunTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionLocalRunTest.java
index 232f7dc..8e95bd9 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionLocalRunTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionLocalRunTest.java
@@ -190,7 +190,7 @@ public class PulsarFunctionLocalRunTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName(CLUSTER);
Set<String> superUsers = Sets.newHashSet("superUser", "admin");
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionPublishTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionPublishTest.java
index 3b704b6..e9847b0 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionPublishTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarFunctionPublishTest.java
@@ -121,7 +121,7 @@ public class PulsarFunctionPublishTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
Set<String> superUsers = Sets.newHashSet("superUser", "admin");
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarWorkerAssignmentTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarWorkerAssignmentTest.java
index 019ca0b..1b66099 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarWorkerAssignmentTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/functions/worker/PulsarWorkerAssignmentTest.java
@@ -89,7 +89,7 @@ public class PulsarWorkerAssignmentTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
final Set<String> superUsers = Sets.newHashSet("superUser", "admin");
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/io/AbstractPulsarE2ETest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/io/AbstractPulsarE2ETest.java
index 1df4be1..93cdd79 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/io/AbstractPulsarE2ETest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/io/AbstractPulsarE2ETest.java
@@ -114,7 +114,7 @@ public abstract class AbstractPulsarE2ETest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
Set<String> superUsers = Sets.newHashSet("superUser", "admin");
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionAdminTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionAdminTest.java
index ddb1fb3..ffab0b7 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionAdminTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionAdminTest.java
@@ -96,7 +96,7 @@ public class PulsarFunctionAdminTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
Set<String> superUsers = Sets.newHashSet("superUser", "admin");
config.setSuperUserRoles(superUsers);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionTlsTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionTlsTest.java
index fcb2a44..31fe8e7 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionTlsTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/io/PulsarFunctionTlsTest.java
@@ -106,7 +106,7 @@ public class PulsarFunctionTlsTest {
bkEnsemble = new LocalBookkeeperEnsemble(3, 0, () -> 0);
bkEnsemble.start();
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setBrokerShutdownTimeoutMs(0L);
config.setClusterName("use");
Set<String> superUsers = Sets.newHashSet("superUser", "admin");
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthenticationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthenticationTest.java
index 5741a5e..764d4a5 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthenticationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthenticationTest.java
@@ -19,26 +19,22 @@
package org.apache.pulsar.websocket.proxy;
import static java.util.concurrent.Executors.newFixedThreadPool;
-import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
-import static org.mockito.ArgumentMatchers.anyString;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
-
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doReturn;
import com.google.common.collect.Sets;
-
import java.net.URI;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-
import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Invocation;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
-
import lombok.Cleanup;
import org.apache.pulsar.client.api.ProducerConsumerBase;
import org.apache.pulsar.metadata.impl.ZKMetadataStore;
@@ -86,7 +82,7 @@ public class ProxyAuthenticationTest extends
ProducerConsumerBase {
config.setAnonymousUserRole("anonymousUser");
}
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
proxyServer = new ProxyServer(config);
WebSocketServiceStarter.start(proxyServer, service);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthorizationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthorizationTest.java
index 78f3370..561fdd7 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthorizationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyAuthorizationTest.java
@@ -18,19 +18,16 @@
*/
package org.apache.pulsar.websocket.proxy;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertTrue;
-
import com.google.common.collect.Sets;
-
import java.util.EnumSet;
import java.util.Optional;
import java.util.Set;
-
import org.apache.pulsar.broker.auth.MockedPulsarServiceBaseTest;
import org.apache.pulsar.broker.authorization.AuthorizationService;
import org.apache.pulsar.common.naming.TopicName;
@@ -68,7 +65,7 @@ public class ProxyAuthorizationTest extends
MockedPulsarServiceBaseTest {
config.setClusterName("c1");
config.setWebServicePort(Optional.of(0));
config.setConfigurationStoreServers(GLOBAL_DUMMY_VALUE);
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
service.start();
}
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyConfigurationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyConfigurationTest.java
index 3848d6e..aeeab2a 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyConfigurationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyConfigurationTest.java
@@ -18,14 +18,12 @@
*/
package org.apache.pulsar.websocket.proxy;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
import static org.testng.Assert.assertEquals;
-
import java.util.Optional;
-
import org.apache.pulsar.client.api.ProducerConsumerBase;
import org.apache.pulsar.client.impl.PulsarClientImpl;
import org.apache.pulsar.metadata.impl.ZKMetadataStore;
@@ -65,7 +63,7 @@ public class ProxyConfigurationTest extends
ProducerConsumerBase {
public void configTest(int numIoThreads, int connectionsPerBroker) throws
Exception {
config.setWebSocketNumIoThreads(numIoThreads);
config.setWebSocketConnectionsPerBroker(connectionsPerBroker);
- WebSocketService service = spy(new WebSocketService(config));
+ WebSocketService service =
spyWithClassAndConstructorArgs(WebSocketService.class, config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
service.start();
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTest.java
index b12f670..177f6f8 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTest.java
@@ -19,20 +19,20 @@
package org.apache.pulsar.websocket.proxy;
import static java.util.concurrent.Executors.newFixedThreadPool;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertTrue;
import static org.testng.Assert.fail;
-
import com.google.common.collect.Sets;
import com.google.gson.Gson;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
import com.google.gson.reflect.TypeToken;
-
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.List;
@@ -43,7 +43,6 @@ import java.util.UUID;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
@@ -51,7 +50,6 @@ import javax.ws.rs.client.Invocation;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
-
import lombok.Cleanup;
import org.apache.pulsar.broker.BrokerTestUtil;
import org.apache.pulsar.client.api.Producer;
@@ -81,8 +79,6 @@ import org.slf4j.LoggerFactory;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
-import com.google.gson.JsonObject;
-import com.google.gson.JsonPrimitive;
@Test(groups = "websocket")
public class ProxyPublishConsumeTest extends ProducerConsumerBase {
@@ -104,7 +100,7 @@ public class ProxyPublishConsumeTest extends
ProducerConsumerBase {
config.setWebServicePort(Optional.of(0));
config.setClusterName("test");
config.setConfigurationStoreServers(GLOBAL_DUMMY_VALUE);
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
proxyServer = new ProxyServer(config);
WebSocketServiceStarter.start(proxyServer, service);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTlsTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTlsTest.java
index 4c82615..b870252 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTlsTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeTlsTest.java
@@ -19,18 +19,16 @@
package org.apache.pulsar.websocket.proxy;
import static java.util.concurrent.Executors.newFixedThreadPool;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
-
import java.net.URI;
import java.security.GeneralSecurityException;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-
import lombok.Cleanup;
import org.apache.pulsar.client.api.TlsProducerConsumerBase;
import org.apache.pulsar.client.impl.auth.AuthenticationTls;
@@ -76,7 +74,7 @@ public class ProxyPublishConsumeTlsTest extends
TlsProducerConsumerBase {
config.setBrokerClientAuthenticationParameters("tlsCertFile:" +
TLS_CLIENT_CERT_FILE_PATH + ",tlsKeyFile:" + TLS_CLIENT_KEY_FILE_PATH);
config.setBrokerClientAuthenticationPlugin(AuthenticationTls.class.getName());
config.setConfigurationStoreServers(GLOBAL_DUMMY_VALUE);
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
proxyServer = new ProxyServer(config);
WebSocketServiceStarter.start(proxyServer, service);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeWithoutZKTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeWithoutZKTest.java
index 485f23b..1fb1264 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeWithoutZKTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/ProxyPublishConsumeWithoutZKTest.java
@@ -19,17 +19,15 @@
package org.apache.pulsar.websocket.proxy;
import static java.util.concurrent.Executors.newFixedThreadPool;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
-
import java.net.URI;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-
import lombok.Cleanup;
import org.apache.pulsar.client.api.ProducerConsumerBase;
import org.apache.pulsar.metadata.impl.ZKMetadataStore;
@@ -63,7 +61,7 @@ public class ProxyPublishConsumeWithoutZKTest extends
ProducerConsumerBase {
config.setClusterName("test");
config.setServiceUrl(pulsar.getSafeWebServiceAddress());
config.setServiceUrlTls(pulsar.getWebServiceAddressTls());
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeper)).when(service).createMetadataStore(anyString(),
anyInt());
proxyServer = new ProxyServer(config);
WebSocketServiceStarter.start(proxyServer, service);
diff --git
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/v1/V1_ProxyAuthenticationTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/v1/V1_ProxyAuthenticationTest.java
index d315a10..842aced 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/v1/V1_ProxyAuthenticationTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/websocket/proxy/v1/V1_ProxyAuthenticationTest.java
@@ -19,26 +19,22 @@
package org.apache.pulsar.websocket.proxy.v1;
import static java.util.concurrent.Executors.newFixedThreadPool;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
-
import com.google.common.collect.Sets;
-
import java.net.URI;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
-
import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Invocation;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
-
import lombok.Cleanup;
import org.apache.pulsar.client.api.v1.V1_ProducerConsumerBase;
import org.apache.pulsar.metadata.impl.ZKMetadataStore;
@@ -88,7 +84,7 @@ public class V1_ProxyAuthenticationTest extends
V1_ProducerConsumerBase {
config.setAnonymousUserRole("anonymousUser");
}
- service = spy(new WebSocketService(config));
+ service = spyWithClassAndConstructorArgs(WebSocketService.class,
config);
doReturn(new
ZKMetadataStore(mockZooKeeperGlobal)).when(service).createMetadataStore(anyString(),
anyInt());
proxyServer = new ProxyServer(config);
WebSocketServiceStarter.start(proxyServer, service);
diff --git
a/pulsar-client/src/test/java/org/apache/pulsar/client/api/MessageRouterTest.java
b/pulsar-client/src/test/java/org/apache/pulsar/client/api/MessageRouterTest.java
index 48f5816..0c29ecb 100644
---
a/pulsar-client/src/test/java/org/apache/pulsar/client/api/MessageRouterTest.java
+++
b/pulsar-client/src/test/java/org/apache/pulsar/client/api/MessageRouterTest.java
@@ -32,7 +32,7 @@ import org.testng.annotations.Test;
*/
public class MessageRouterTest {
- private static class TestMessageRouter implements MessageRouter {
+ public static class TestMessageRouter implements MessageRouter {
@Override
public int choosePartition(Message<?> msg) {
@@ -43,7 +43,7 @@ public class MessageRouterTest {
@SuppressWarnings("deprecation")
@Test
public void testChoosePartition() {
- MessageRouter router = spy(new TestMessageRouter());
+ MessageRouter router = spy(TestMessageRouter.class);
Message<?> mockedMsg = mock(Message.class);
TopicMetadata mockedMetadata = mock(TopicMetadata.class);
diff --git
a/pulsar-functions/worker/src/test/java/org/apache/pulsar/functions/worker/FunctionRuntimeManagerTest.java
b/pulsar-functions/worker/src/test/java/org/apache/pulsar/functions/worker/FunctionRuntimeManagerTest.java
index 93e0e37..02e67a2 100644
---
a/pulsar-functions/worker/src/test/java/org/apache/pulsar/functions/worker/FunctionRuntimeManagerTest.java
+++
b/pulsar-functions/worker/src/test/java/org/apache/pulsar/functions/worker/FunctionRuntimeManagerTest.java
@@ -946,7 +946,7 @@ public class FunctionRuntimeManagerTest {
WorkerConfig workerConfig = new WorkerConfig();
workerConfig.setKubernetesContainerFactory(kubernetesContainerFactory);
- KubernetesRuntimeFactory mockedKubernetesRuntimeFactory = spy(new
KubernetesRuntimeFactory());
+ KubernetesRuntimeFactory mockedKubernetesRuntimeFactory =
spy(KubernetesRuntimeFactory.class);
doNothing().when(mockedKubernetesRuntimeFactory).initialize(
any(WorkerConfig.class),
any(AuthenticationConfig.class),
@@ -1111,7 +1111,7 @@ public class FunctionRuntimeManagerTest {
WorkerConfig.KubernetesContainerFactory kubernetesContainerFactory
= new WorkerConfig.KubernetesContainerFactory();
workerConfig.setKubernetesContainerFactory(kubernetesContainerFactory);
- KubernetesRuntimeFactory mockedKubernetesRuntimeFactory = spy(new
KubernetesRuntimeFactory());
+ KubernetesRuntimeFactory mockedKubernetesRuntimeFactory =
spy(KubernetesRuntimeFactory.class);
doNothing().when(mockedKubernetesRuntimeFactory).initialize(
any(WorkerConfig.class),
any(AuthenticationConfig.class),
diff --git
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarConnector.java
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarConnector.java
index 5d9f363..a91290c 100644
---
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarConnector.java
+++
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarConnector.java
@@ -347,7 +347,7 @@ public abstract class TestPulsarConnector {
public static PulsarMetadata mockColumnMetadata() {
ConnectorContext prestoConnectorContext = new
TestingConnectorContext();
- PulsarConnectorConfig pulsarConnectorConfig = spy(new
PulsarConnectorConfig());
+ PulsarConnectorConfig pulsarConnectorConfig =
spy(PulsarConnectorConfig.class);
pulsarConnectorConfig.setMaxEntryReadBatchSize(1);
pulsarConnectorConfig.setMaxSplitEntryQueueSize(10);
pulsarConnectorConfig.setMaxSplitMessageQueueSize(100);
@@ -436,7 +436,7 @@ public abstract class TestPulsarConnector {
@BeforeMethod
public void setup() throws Exception {
- this.pulsarConnectorConfig = spy(new PulsarConnectorConfig());
+ this.pulsarConnectorConfig = spy(PulsarConnectorConfig.class);
this.pulsarConnectorConfig.setMaxEntryReadBatchSize(1);
this.pulsarConnectorConfig.setMaxSplitEntryQueueSize(10);
this.pulsarConnectorConfig.setMaxSplitMessageQueueSize(100);
diff --git
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarRecordCursor.java
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarRecordCursor.java
index 60d3fed..3b853b7 100644
---
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarRecordCursor.java
+++
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/TestPulsarRecordCursor.java
@@ -446,7 +446,7 @@ public class TestPulsarRecordCursor extends
TestPulsarConnector {
PulsarAdmin pulsarAdmin = Mockito.mock(PulsarAdmin.class);
Schemas schemas = Mockito.mock(Schemas.class);
Mockito.when(pulsarAdmin.schemas()).thenReturn(schemas);
- PulsarConnectorConfig connectorConfig = spy(new
PulsarConnectorConfig());
+ PulsarConnectorConfig connectorConfig =
spy(PulsarConnectorConfig.class);
Mockito.when(connectorConfig.getPulsarAdmin()).thenReturn(pulsarAdmin);
PulsarRecordCursor pulsarRecordCursor = spy(new PulsarRecordCursor(
new ArrayList<>(), pulsarSplit, connectorConfig,
Mockito.mock(ManagedLedgerFactory.class),
diff --git
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/decoder/AbstractDecoderTester.java
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/decoder/AbstractDecoderTester.java
index cd4fcaf..5cd4683 100644
---
a/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/decoder/AbstractDecoderTester.java
+++
b/pulsar-sql/presto-pulsar/src/test/java/org/apache/pulsar/sql/presto/decoder/AbstractDecoderTester.java
@@ -62,7 +62,7 @@ public abstract class AbstractDecoderTester {
protected void init() {
ConnectorContext prestoConnectorContext = new
TestingConnectorContext();
this.decoderFactory = new
PulsarDispatchingRowDecoderFactory(prestoConnectorContext.getTypeManager());
- this.pulsarConnectorConfig = spy(new PulsarConnectorConfig());
+ this.pulsarConnectorConfig = spy(PulsarConnectorConfig.class);
this.pulsarConnectorConfig.setMaxEntryReadBatchSize(1);
this.pulsarConnectorConfig.setMaxSplitEntryQueueSize(10);
this.pulsarConnectorConfig.setMaxSplitMessageQueueSize(100);
