From 363aa6a67b34c1a06a05221d64609c03db6bb195 Mon Sep 17 00:00:00 2001 From: Savindu Dimal Date: Wed, 9 Oct 2024 12:55:38 +0530 Subject: [PATCH] Fix unit tests --- .../carbon/apimgt/impl/utils/APIUtilTierTest.java | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/components/apimgt/org.wso2.carbon.apimgt.impl/src/test/java/org/wso2/carbon/apimgt/impl/utils/APIUtilTierTest.java b/components/apimgt/org.wso2.carbon.apimgt.impl/src/test/java/org/wso2/carbon/apimgt/impl/utils/APIUtilTierTest.java index ff1ce9d9a1d0..cfc2c362f882 100644 --- a/components/apimgt/org.wso2.carbon.apimgt.impl/src/test/java/org/wso2/carbon/apimgt/impl/utils/APIUtilTierTest.java +++ b/components/apimgt/org.wso2.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(