Improve unit tests.

Thisara-Welmilla · Sep 26, 2024 · a8cea2d · a8cea2d
1 parent d4a99a1
commit a8cea2d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...g.wso2.carbon.idp.mgt/src/test/java/org/wso2/carbon/idp/mgt/dao/IdPManagementDAOTest.java b/...g.wso2.carbon.idp.mgt/src/test/java/org/wso2/carbon/idp/mgt/dao/IdPManagementDAOTest.java
@@ -71,10 +71,7 @@
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.mockStatic;
 import static org.mockito.Mockito.when;
-import static org.testng.Assert.assertEquals;
-import static org.testng.Assert.assertNotEquals;
-import static org.testng.Assert.assertNull;
-import static org.testng.Assert.assertThrows;
+import static org.testng.Assert.*;
 import static org.wso2.carbon.idp.mgt.util.IdPManagementConstants.RESET_PROVISIONING_ENTITIES_ON_CONFIG_UPDATE;
 
 /**
@@ -1032,6 +1029,7 @@ public void testGetIdPByAuthenticatorPropertyValue(int tenantId, String idpName,
                     value, authenticator, tenantId, TENANT_DOMAIN);
             if (isExist) {
                 assertEquals(idpResult.getIdentityProviderName(), idpName);
+                assertNotNull(idpResult.getDefaultAuthenticatorConfig().getDefinedByType());
             } else {
                 assertNull(idpResult);
             }
@@ -1084,6 +1082,7 @@ public void testGetIdPByAuthenticatorPropertyWithoutAuthenticatorData(int tenant
 
             if (isExist) {
                 assertEquals(idpResult.getIdentityProviderName(), idpName);
+                assertNotNull(idpResult.getDefaultAuthenticatorConfig().getDefinedByType());
             } else {
                 assertNull(idpResult);
             }
@@ -1748,6 +1747,7 @@ private void addTestIdps() throws IdentityProviderManagementException {
         property2.setConfidential(false);
         federatedAuthenticatorConfig.setProperties(new Property[]{property1, property2});
         idp1.setFederatedAuthenticatorConfigs(new FederatedAuthenticatorConfig[]{federatedAuthenticatorConfig});
+        idp1.setDefaultAuthenticatorConfig(federatedAuthenticatorConfig);
 
         ProvisioningConnectorConfig provisioningConnectorConfig1 = new ProvisioningConnectorConfig();
         provisioningConnectorConfig1.setName("ProvisiningConfig1");