This is an automated email from the ASF dual-hosted git repository.
mattisonchao pushed a commit to branch branch-2.9
in repository https://gitbox.apache.org/repos/asf/pulsar.git
The following commit(s) were added to refs/heads/branch-2.9 by this push:
new dfbc09f5eee [branch-2.9] [Tests] Fix flaky WrongTypeOfReturnValue
Mockito misuse issue in broker tests. (#15738)
dfbc09f5eee is described below
commit dfbc09f5eee0a2d4217663cd9cb56126a0b7283f
Author: Qiang Zhao <[email protected]>
AuthorDate: Wed May 25 09:11:26 2022 +0800
[branch-2.9] [Tests] Fix flaky WrongTypeOfReturnValue Mockito misuse issue
in broker tests. (#15738)
---
.../mledger/impl/OffloadPrefixReadTest.java | 4 +-
.../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 | 8 +--
.../pulsar/broker/admin/ResourceGroupsTest.java | 2 +-
.../broker/auth/MockedPulsarServiceBaseTest.java | 6 ++-
.../loadbalance/LeaderElectionServiceTest.java | 6 ++-
.../loadbalance/SimpleLoadManagerImplTest.java | 5 +-
.../broker/lookup/http/HttpTopicLookupv2Test.java | 8 +--
.../broker/lookup/http/v2/TopicLookupTest.java | 4 +-
.../OwnerShipForCurrentServerTestBase.java | 8 +--
.../PersistentDispatcherFailoverConsumerTest.java | 15 +++---
.../service/PersistentTopicConcurrentTest.java | 11 ++---
.../pulsar/broker/service/PersistentTopicTest.java | 57 ++++++++++++----------
.../pulsar/broker/service/ServerCnxTest.java | 20 +++++---
.../service/persistent/MessageDuplicationTest.java | 11 +++--
.../persistent/PersistentSubscriptionTest.java | 7 +--
.../broker/transaction/TransactionTestBase.java | 6 ++-
.../coordinator/TransactionMetaStoreTestBase.java | 4 +-
.../apache/pulsar/broker/web/WebServiceTest.java | 3 +-
.../client/api/ClientDeduplicationFailureTest.java | 2 +-
.../client/api/DispatcherBlockConsumerTest.java | 4 +-
.../client/impl/BrokerClientIntegrationTest.java | 6 ++-
.../pulsar/client/impl/ConnectionPoolTest.java | 10 ++--
.../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 | 3 +-
.../websocket/proxy/ProxyAuthorizationTest.java | 3 +-
.../websocket/proxy/ProxyConfigurationTest.java | 3 +-
.../websocket/proxy/ProxyPublishConsumeTest.java | 4 +-
.../proxy/ProxyPublishConsumeTlsTest.java | 3 +-
.../proxy/ProxyPublishConsumeWithoutZKTest.java | 3 +-
.../proxy/v1/V1_ProxyAuthenticationTest.java | 5 +-
.../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 +-
44 files changed, 160 insertions(+), 125 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 426a00db25e..e761b04b79b 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/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 cba06863831..ea0b5c2a155 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
@@ -81,7 +81,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 82e5b90b54f..a2f04927de9 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
@@ -130,17 +130,17 @@ public class AdminTest extends
MockedPulsarServiceBaseTest {
conf.setClusterName(configClusterName);
super.internalSetup();
- 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("test").when(namespaces).clientAppId();
@@ -149,7 +149,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();
@@ -157,7 +157,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("test").when(persistentTopics).clientAppId();
@@ -167,11 +167,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);
@@ -180,7 +180,7 @@ public class AdminTest extends MockedPulsarServiceBaseTest {
doReturn("test").when(persistentTopics).clientAppId();
doReturn(mock(AuthenticationDataHttps.class)).when(persistentTopics).clientAuthData();
- schemasResource = spy(new SchemasResource(mockClock));
+ 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 16ae98f0b43..69bede376c8 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
@@ -155,7 +155,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(false).when(namespaces).isRequestHttps();
@@ -1089,7 +1089,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();
@@ -1253,7 +1253,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 2baf5b0f469..4591aebc67d 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;
@@ -116,7 +117,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(false).when(persistentTopics).isRequestHttps();
@@ -126,7 +127,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);
namespaceResources = mock(NamespaceResources.class);
@@ -139,7 +140,8 @@ public class PersistentTopicsTest extends
MockedPulsarServiceBaseTest {
PulsarResources resources =
spy(new PulsarResources(pulsar.getLocalMetadataStore(),
pulsar.getConfigurationMetadataStore()));
- doReturn(spy(new
TopicResources(pulsar.getLocalMetadataStore()))).when(resources).getTopicResources();
+ doReturn(spyWithClassAndConstructorArgs(TopicResources.class,
+
pulsar.getLocalMetadataStore())).when(resources).getTopicResources();
Whitebox.setInternalState(pulsar, "pulsarResources", resources);
admin.clusters().createCluster("use",
ClusterData.builder().serviceUrl("http://broker-use.com:8080";).build());
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 b6510a18995..4910f6798ba 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(false).when(resourcegroups).isRequestHttps();
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 c50b477b5d3..c731d310d99 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;
@@ -327,7 +328,8 @@ 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();
@@ -391,7 +393,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 d252b1cf326..ec19480ae06 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
@@ -44,6 +44,8 @@ import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
+
@Slf4j
@Test(groups = "broker")
public class LeaderElectionServiceTest {
@@ -75,7 +77,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
@@ -135,7 +137,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 059191b7876..d08699998e7 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;
@@ -108,7 +109,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());
@@ -336,7 +337,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 b65b084ccdc..47db73d39f7 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
@@ -73,7 +73,7 @@ public class HttpTopicLookupv2Test {
pulsar = mock(PulsarService.class);
ns = mock(NamespaceService.class);
auth = mock(AuthorizationService.class);
- config = spy(new ServiceConfiguration());
+ config = spy(ServiceConfiguration.class);
config.setClusterName("use");
clusters = new TreeSet<>();
clusters.add("use");
@@ -102,7 +102,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();
@@ -132,7 +132,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();
@@ -170,7 +170,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/lookup/http/v2/TopicLookupTest.java
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/v2/TopicLookupTest.java
index 317b320a9a4..21e7abded47 100644
---
a/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/v2/TopicLookupTest.java
+++
b/pulsar-broker/src/test/java/org/apache/pulsar/broker/lookup/http/v2/TopicLookupTest.java
@@ -46,7 +46,7 @@ public class TopicLookupTest extends PulsarWebResourceTest {
@Override
protected ResourceConfig configure() {
- resource = spy(new TestableTopicLookup());
+ resource = spy(TestableTopicLookup.class);
return new ResourceConfig().register(resource);
}
@@ -70,7 +70,7 @@ public class TopicLookupTest extends PulsarWebResourceTest {
assertEquals(resource.actualListenerName, "query");
}
- private static class TestableTopicLookup extends TopicLookup {
+ public static class TestableTopicLookup extends TopicLookup {
private String actualListenerName;
@Override
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 9a8021db28c..a6604d96397 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 static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
@@ -118,7 +119,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();
@@ -133,7 +134,8 @@ 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();
@@ -157,7 +159,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 b886adee15e..663a1023543 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,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.ArgumentMatchers.matches;
import static org.mockito.ArgumentMatchers.same;
@@ -122,9 +123,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);
@@ -141,7 +142,7 @@ public class PersistentDispatcherFailoverConsumerTest {
PulsarResources pulsarResources = new PulsarResources(store, store);
doReturn(pulsarResources).when(pulsar).getPulsarResources();
- brokerService = spy(new BrokerService(pulsar, eventLoopGroup));
+ brokerService = spyWithClassAndConstructorArgs(BrokerService.class,
pulsar, eventLoopGroup);
doReturn(brokerService).when(pulsar).getBrokerService();
consumerChanges = new LinkedBlockingQueue<>();
@@ -167,9 +168,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();
@@ -178,9 +177,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 78c7e255739..9bd469a955b 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,6 +18,7 @@
*/
package org.apache.pulsar.broker.service;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static org.mockito.Mockito.CALLS_REAL_METHODS;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.doReturn;
@@ -83,10 +84,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")
@@ -100,12 +101,10 @@ public class PersistentTopicConcurrentTest extends
MockedBookKeeperTestCase {
mlFactoryMock = factory;
doReturn(mlFactoryMock).when(pulsar).getManagedLedgerFactory();
- 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 5cb945e436f..26004251399 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,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.mockito.ArgumentMatchers.any;
@@ -175,10 +176,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();
@@ -199,20 +200,18 @@ 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, store, 30));
+ PulsarResources pulsarResources =
spyWithClassAndConstructorArgs(PulsarResources.class, store, store);
+ NamespaceResources nsr =
spyWithClassAndConstructorArgs(NamespaceResources.class, store, store, 30);
doReturn(nsr).when(pulsarResources).getNamespaceResources();
doReturn(pulsarResources).when(pulsar).getPulsarResources();
doReturn(store).when(pulsar).getLocalMetadataStore();
doReturn(store).when(pulsar).getConfigurationMetadataStore();
- 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();
@@ -354,7 +353,8 @@ 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);
@@ -364,7 +364,8 @@ 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,
@@ -561,7 +562,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();
@@ -569,7 +570,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();
@@ -605,7 +606,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();
@@ -902,7 +903,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();
@@ -912,7 +913,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
@@ -1005,7 +1006,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();
@@ -1015,7 +1016,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
@@ -1037,9 +1038,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();
@@ -1053,7 +1052,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();
@@ -2106,13 +2105,19 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
.concurrencyLevel(1)
.build();
// 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");
@@ -2131,7 +2136,7 @@ public class PersistentTopicTest extends
MockedBookKeeperTestCase {
NamespaceName ns =
TopicName.get(successTopicName).getNamespaceObject();
doReturn(Optional.of(new Policies())).when(nsr).getPolicies(ns);
- ServiceConfiguration svcConfig = spy(new ServiceConfiguration());
+ ServiceConfiguration svcConfig = spy(ServiceConfiguration.class);
doReturn(5).when(svcConfig).getSubscriptionExpirationTimeMinutes();
doReturn(svcConfig).when(pulsar).getConfiguration();
@@ -2146,7 +2151,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 8881ce90812..1a5efb19f3e 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.mockito.ArgumentMatchers.any;
@@ -152,9 +153,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));
@@ -176,14 +177,14 @@ public class ServerCnxTest {
doReturn(store).when(pulsar).getLocalMetadataStore();
doReturn(store).when(pulsar).getConfigurationMetadataStore();
- 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();
doReturn(executor).when(pulsar).getOrderedExecutor();
- PulsarResources pulsarResources = spy(new PulsarResources(store,
store));
- namespaceResources = spy(new NamespaceResources(store, store, 30));
+ PulsarResources pulsarResources =
spyWithClassAndConstructorArgs(PulsarResources.class, store, store);
+ namespaceResources =
spyWithClassAndConstructorArgs(NamespaceResources.class, store, store, 30);
doReturn(namespaceResources).when(pulsarResources).getNamespaceResources();
doReturn(pulsarResources).when(pulsar).getPulsarResources();
@@ -502,7 +503,8 @@ public class ServerCnxTest {
@Test(timeOut = 30000)
public void testNonExistentTopic() throws Exception {
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, pulsar.getPulsarResources()));
+ AuthorizationService authorizationService =
+ spyWithClassAndConstructorArgs(AuthorizationService.class,
svcConfig, pulsar.getPulsarResources());
doReturn(authorizationService).when(brokerService).getAuthorizationService();
doReturn(true).when(brokerService).isAuthorizationEnabled();
svcConfig.setAuthorizationEnabled(true);
@@ -535,7 +537,8 @@ public class ServerCnxTest {
@Test(timeOut = 30000)
public void testClusterAccess() throws Exception {
svcConfig.setAuthorizationEnabled(true);
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, pulsar.getPulsarResources()));
+ AuthorizationService authorizationService =
+ spyWithClassAndConstructorArgs(AuthorizationService.class,
svcConfig, pulsar.getPulsarResources());
Field providerField =
AuthorizationService.class.getDeclaredField("provider");
providerField.setAccessible(true);
PulsarAuthorizationProvider authorizationProvider = spy(new
PulsarAuthorizationProvider(svcConfig,
@@ -566,7 +569,8 @@ public class ServerCnxTest {
@Test(timeOut = 30000)
public void testNonExistentTopicSuperUserAccess() throws Exception {
- AuthorizationService authorizationService = spy(new
AuthorizationService(svcConfig, pulsar.getPulsarResources()));
+ AuthorizationService authorizationService =
+ spyWithClassAndConstructorArgs(AuthorizationService.class,
svcConfig, pulsar.getPulsarResources());
doReturn(authorizationService).when(brokerService).getAuthorizationService();
doReturn(true).when(brokerService).isAuthorizationEnabled();
Field providerField =
AuthorizationService.class.getDeclaredField("provider");
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 4dc7f7f4232..117c9dd1582 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
@@ -37,6 +37,7 @@ import org.testng.annotations.Test;
import java.lang.reflect.Field;
import java.util.Map;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
import static
org.apache.pulsar.common.protocol.Commands.serializeMetadataAndPayload;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
@@ -71,7 +72,8 @@ 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";
@@ -164,7 +166,8 @@ 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);
@@ -238,7 +241,9 @@ 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 b72fe76f14c..b9304cb5fb8 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;
@@ -109,10 +110,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);
PulsarResources pulsarResources = mock(PulsarResources.class);
doReturn(pulsarResources).when(pulsarMock).getPulsarResources();
NamespaceResources namespaceResources = mock(NamespaceResources.class);
@@ -182,7 +183,7 @@ public class PersistentSubscriptionTest {
doReturn(store).when(pulsarMock).getLocalMetadataStore();
doReturn(store).when(pulsarMock).getConfigurationMetadataStore();
- 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 7cae6ca3ec3..2829aa9f913 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;
@@ -171,7 +172,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();
@@ -187,7 +188,8 @@ 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 04579078ad4..b012dfa87d2 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
@@ -34,6 +34,8 @@ import org.slf4j.LoggerFactory;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
+
public abstract class TransactionMetaStoreTestBase extends TestRetrySupport {
private static final Logger log =
LoggerFactory.getLogger(TransactionMetaStoreTestBase.class);
@@ -76,7 +78,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 23a3db91692..4173d5152ea 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,6 +18,7 @@
*/
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;
@@ -424,7 +425,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 a2b4b5c3dbf..2ddb9e8c8a3 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
@@ -81,7 +81,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 0f21e211d92..867ea1592eb 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,6 +18,7 @@
*/
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;
@@ -630,7 +631,8 @@ 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 2be6067c7b7..911a23b1bb8 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.mockito.Mockito.any;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.doAnswer;
@@ -822,8 +823,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)
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 1e97550322b..235bd7167a5 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
@@ -36,6 +36,8 @@ import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.stream.IntStream;
+import static
org.apache.pulsar.broker.BrokerTestUtil.spyWithClassAndConstructorArgs;
+
@Test(groups = "broker-impl")
public class ConnectionPoolTest extends MockedPulsarServiceBaseTest {
@@ -58,7 +60,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 +80,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 +103,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 +125,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 e7173a23e39..c399cd8b734 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 13ac623aa1d..c48de5257b3 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
@@ -193,7 +193,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 9f5e525d6b7..d985241e290 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
@@ -119,7 +119,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 019ca0bd577..1b66099c14c 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 1df4be1c71d..93cdd79ad01 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 ddb1fb3d736..ffab0b78a27 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 5867bb78404..15ee27dc3a5 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
@@ -105,7 +105,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 5741a5eb0e6..b848fa76d54 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,6 +19,7 @@
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.Mockito.doReturn;
import static org.mockito.Mockito.spy;
import static org.mockito.ArgumentMatchers.anyString;
@@ -86,7 +87,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 78f33706d53..a2758b72a4e 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,6 +18,7 @@
*/
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;
@@ -68,7 +69,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 3848d6ecc1d..ec4937bdd21 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,6 +18,7 @@
*/
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;
@@ -65,7 +66,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 b12f670b8b2..1e74cdae787 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,10 +19,10 @@
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;
@@ -104,7 +104,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 b780fde760e..cdc2eb58d9a 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,6 +19,7 @@
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;
@@ -76,7 +77,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 485f23bdeb1..5baaacd52d9 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,6 +19,7 @@
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;
@@ -63,7 +64,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 d315a10c46a..03227e9587d 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,11 +19,10 @@
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;
@@ -88,7 +87,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 48f5816e2d2..0c29ecb7fef 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 79871db8784..be832236aac 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),
@@ -1112,7 +1112,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 fdfde36cb28..7db32f59148 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
@@ -361,7 +361,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);
@@ -451,7 +451,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 2ea2616b6f1..880c2fb585b 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
@@ -461,7 +461,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 98b7d8b6f69..e5ceb321aae 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
@@ -63,7 +63,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);
