Merge pull request #3 from SavinduDimal/api-key-mediator-tests

Fix unit tests
PasanT9 · Oct 9, 2024 · d1ca39e · d1ca39e
2 parents 2340201 + 363aa6a
commit d1ca39e
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/...2.carbon.apimgt.impl/src/test/java/org/wso2/carbon/apimgt/impl/utils/APIUtilTierTest.java b/...2.carbon.apimgt.impl/src/test/java/org/wso2/carbon/apimgt/impl/utils/APIUtilTierTest.java
@@ -439,7 +439,9 @@ public void testAddDefaultSuperTenantAdvancedThrottlePoliciesSubLevel() throws E
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_UNLIMITED,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_SUBSCRIPTIONLESS,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_GOLD, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_SILVER,
-                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED};
+                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED,
+                APIConstants.DEFAULT_SUB_POLICY_AI_GOLD, APIConstants.DEFAULT_SUB_POLICY_AI_SILVER,
+                APIConstants.DEFAULT_SUB_POLICY_AI_BRONZE};
 
         for (String policy : subPolicies) {
             Mockito.when(
@@ -472,7 +474,8 @@ public void testAddDefaultSuperTenantAdvancedThrottlePoliciesSubLevelAlreadyAdde
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_SILVER, APIConstants.DEFAULT_SUB_POLICY_ASYNC_BRONZE,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_UNLIMITED, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_GOLD,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_SILVER, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE,
-                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED};
+                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED, APIConstants.DEFAULT_SUB_POLICY_AI_GOLD,
+                APIConstants.DEFAULT_SUB_POLICY_AI_SILVER, APIConstants.DEFAULT_SUB_POLICY_AI_BRONZE};
 
         for (String policy : subPolicies) {
             Mockito.when(
@@ -622,7 +625,9 @@ public void testAddDefaultTenantAdvancedThrottlePoliciesSubLevel() throws Except
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_UNLIMITED,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_SUBSCRIPTIONLESS,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_GOLD, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_SILVER,
-                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED};
+                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED,
+                APIConstants.DEFAULT_SUB_POLICY_AI_GOLD, APIConstants.DEFAULT_SUB_POLICY_AI_SILVER,
+                APIConstants.DEFAULT_SUB_POLICY_AI_BRONZE};
 
         for (String policy : policies) {
             Mockito.when(
@@ -656,7 +661,9 @@ public void testAddDefaultTenantAdvancedThrottlePoliciesSubLevelAlreadyAdded() t
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_UNLIMITED,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_SUBSCRIPTIONLESS,
                 APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_GOLD, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_SILVER,
-                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED};
+                APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_BRONZE, APIConstants.DEFAULT_SUB_POLICY_ASYNC_WH_UNLIMITED,
+                APIConstants.DEFAULT_SUB_POLICY_AI_GOLD, APIConstants.DEFAULT_SUB_POLICY_AI_SILVER,
+                APIConstants.DEFAULT_SUB_POLICY_AI_BRONZE};
 
         for (String policy : policies) {
             Mockito.when(