Add unit tests

wso2-extensions · Oct 3, 2024 · 14529e1 · 14529e1
1 parent da9ce97
commit 14529e1
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/...c/test/java/org/wso2/carbon/identity/recovery/signup/UserSelfRegistrationManagerTest.java b/...c/test/java/org/wso2/carbon/identity/recovery/signup/UserSelfRegistrationManagerTest.java
@@ -69,6 +69,7 @@
 import org.wso2.carbon.user.api.Claim;
 import org.wso2.carbon.user.api.UserRealm;
 import org.wso2.carbon.user.api.UserStoreManager;
+import org.wso2.carbon.user.core.UserStoreException;
 import org.wso2.carbon.user.core.service.RealmService;
 
 import java.sql.Timestamp;
@@ -80,6 +81,7 @@
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.when;
 import static org.testng.Assert.assertEquals;
 import static org.wso2.carbon.identity.auth.attribute.handler.AuthAttributeHandlerConstants.ErrorMessages.ERROR_CODE_AUTH_ATTRIBUTE_HANDLER_NOT_FOUND;
@@ -481,6 +483,20 @@ public void testAttributeVerificationFailures(String scenario, Property[] proper
         }
     }
 
+    @Test
+    public void registerUserInvalidOrganizationEmailDomain() {
+
+        try {
+            // Mock addUser to throw UserStoreException
+            doThrow(new UserStoreException("Simulated exception", "ORG-60091")).when(userStoreManager)
+                    .isExistingRole(anyString());
+            userSelfRegistrationManager.registerUser(new User(), "", new Claim[]{}, null);
+        } catch (Exception e) {
+            // Skip the exception as the required error handling flow related code is covered.
+        }
+
+    }
+
     /**
      * Sample consent manager class.
      */