[pulsar] branch master updated: [improve][test] Setup broker client for TokenOauth2AuthenticatedProducerConsumerTest (#18313)

[zixuan]

This is an automated email from the ASF dual-hosted git repository.

zixuan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/pulsar.git


The following commit(s) were added to refs/heads/master by this push:
     new fd246a93b64 [improve][test] Setup broker client for 
TokenOauth2AuthenticatedProducerConsumerTest (#18313)
fd246a93b64 is described below

commit fd246a93b6401a3b4f608a93e207995c3d50bbdb
Author: Zixuan Liu <node...@gmail.com>
AuthorDate: Fri Nov 4 00:16:15 2022 +0800

    [improve][test] Setup broker client for 
TokenOauth2AuthenticatedProducerConsumerTest (#18313)
    
    Signed-off-by: Zixuan Liu <node...@gmail.com>
---
 .../api/TokenOauth2AuthenticatedProducerConsumerTest.java  | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git 
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/TokenOauth2AuthenticatedProducerConsumerTest.java
 
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/TokenOauth2AuthenticatedProducerConsumerTest.java
index 1a62eee9be6..22834b2e0c9 100644
--- 
a/pulsar-broker/src/test/java/org/apache/pulsar/client/api/TokenOauth2AuthenticatedProducerConsumerTest.java
+++ 
b/pulsar-broker/src/test/java/org/apache/pulsar/client/api/TokenOauth2AuthenticatedProducerConsumerTest.java
@@ -35,6 +35,7 @@ import 
org.apache.pulsar.broker.authentication.AuthenticationProviderToken;
 import org.apache.pulsar.client.admin.PulsarAdmin;
 import org.apache.pulsar.client.impl.ProducerImpl;
 import org.apache.pulsar.client.impl.auth.oauth2.AuthenticationFactoryOAuth2;
+import org.apache.pulsar.client.impl.auth.oauth2.AuthenticationOAuth2;
 import org.apache.pulsar.common.policies.data.ClusterData;
 import org.apache.pulsar.common.policies.data.TenantInfoImpl;
 import org.awaitility.Awaitility;
@@ -61,6 +62,8 @@ public class TokenOauth2AuthenticatedProducerConsumerTest 
extends ProducerConsum
 
     // Credentials File, which contains "client_id" and "client_secret"
     private final String CREDENTIALS_FILE = 
"./src/test/resources/authentication/token/credentials_file.json";
+    private final String ISSUER_URL = "https://dev-kt-aa9ne.us.auth0.com";;
+    private final String AUDIENCE = 
"https://dev-kt-aa9ne.us.auth0.com/api/v2/";;
 
     @BeforeMethod(alwaysRun = true)
     @Override
@@ -77,6 +80,13 @@ public class TokenOauth2AuthenticatedProducerConsumerTest 
extends ProducerConsum
         providers.add(AuthenticationProviderToken.class.getName());
         conf.setAuthenticationProviders(providers);
 
+        
conf.setBrokerClientAuthenticationPlugin(AuthenticationOAuth2.class.getName());
+        conf.setBrokerClientAuthenticationParameters("{\n"
+                + "  \"privateKey\": \"" + CREDENTIALS_FILE + "\",\n"
+                + "  \"issuerUrl\": \"" + ISSUER_URL + "\",\n"
+                + "  \"audience\": \"" + AUDIENCE + "\",\n"
+                + "}\n");
+
         conf.setClusterName("test");
 
         // Set provider domain name
@@ -94,9 +104,9 @@ public class TokenOauth2AuthenticatedProducerConsumerTest 
extends ProducerConsum
 
         // AuthenticationOAuth2
         Authentication authentication = 
AuthenticationFactoryOAuth2.clientCredentials(
-                new URL("https://dev-kt-aa9ne.us.auth0.com";),
+                new URL(ISSUER_URL),
                 path.toUri().toURL(),  // key file path
-                "https://dev-kt-aa9ne.us.auth0.com/api/v2/";
+                AUDIENCE
         );
 
         admin = spy(PulsarAdmin.builder().serviceHttpUrl(brokerUrl.toString())