Improve unit tests.

Thisara-Welmilla · Sep 26, 2024 · c0c2106 · c0c2106
1 parent d4a99a1
commit c0c2106
Showing 1 changed file with 3 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);
             }