Fix test

wso2 · Oct 31, 2023 · b4a293c · b4a293c
1 parent d9066ec
commit b4a293c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/...wso2.apk.enforcer/src/main/java/org/wso2/apk/enforcer/server/RevokedTokenRedisClient.java b/...wso2.apk.enforcer/src/main/java/org/wso2/apk/enforcer/server/RevokedTokenRedisClient.java
@@ -223,6 +223,10 @@ public static Set<String> getRevokedTokens() {
         return revokedTokensStatic;
     }
 
+    public static void setRevokedTokens(Set<String>revokedTokensSet) {
+        revokedTokensStatic = revokedTokensSet;
+    }
+
     private static SSLSocketFactory createSslSocketFactory(String redisCaCertPath) throws EnforcerException {
         try {
             KeyStore trustStore = TLSUtils.getDefaultCertTrustStore();

diff --git a/...orcer/org.wso2.apk.enforcer/src/test/java/org/wso2/apk/enforcer/jwt/JWTValidatorTest.java b/...orcer/org.wso2.apk.enforcer/src/test/java/org/wso2/apk/enforcer/jwt/JWTValidatorTest.java
@@ -19,13 +19,15 @@
 import com.google.common.cache.LoadingCache;
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Map;
 import java.util.UUID;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.junit.Assert;
+import org.junit.Before;
 import org.junit.Test;
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
@@ -52,10 +54,16 @@
 import org.wso2.apk.enforcer.security.KeyValidator;
 import org.wso2.apk.enforcer.security.jwt.JWTAuthenticator;
 import org.wso2.apk.enforcer.security.jwt.validator.JWTValidator;
+import org.wso2.apk.enforcer.server.RevokedTokenRedisClient;
 import org.wso2.apk.enforcer.subscription.SubscriptionDataStoreImpl;
 
 public class JWTValidatorTest {
 
+    @Before
+    public void setup() {
+        RevokedTokenRedisClient.setRevokedTokens(new HashSet<>());
+    }
+
     @Test
     public void testJWTValidator() throws APISecurityException, EnforcerException {
         String organization = "org1";