MyDomainLoginDiscoveryHandler インターフェース

1// This sample class contains the default logic for My Domain login discovery by password. 
2// You can customize the code to ensure it meets your needs. The requestAttributes parameter 
3// provides additional information you can use in the discovery logic. Attributes include MyDomainUrl, 
4// IpAddress, UserAgent, and location information (such as Country and City). 
5// Use Auth.DiscoveryCustomErrorException to throw custom errors which will be shown on login page. 
6 
7global class MyDomainDiscLoginDefaultHandler implements Auth.MyDomainLoginDiscoveryHandler {
8 
9  global PageReference login(String identifier, String startUrl, Map<String, String> requestAttributes) {
10   if (identifier != null) {
11        // Search for user by email 
12        List<User> users = [SELECT Id FROM User WHERE Email = :identifier AND IsActive = TRUE];
13        if (!users.isEmpty() && users.size() == 1) {
14            return discoveryResult(users[0], startUrl, requestAttributes);
15        } else {
16            throw new Auth.LoginDiscoveryException('No unique user found. User count=' + users.size());
17        }
18    }
19    throw new Auth.LoginDiscoveryException('Invalid Identifier');
20  }
21  
22  private PageReference getSsoRedirect(User user, String startUrl, Map<String, String> requestAttributes) {
23    // You can look up if the user should log in with SAML or an Auth Provider and return the URL to initialize SSO. For example : 
24    // SamlSsoConfig SSO = [select Id from SamlSsoConfig where DeveloperName='SamlTest' limit 1];
25    // String ssoUrl = Auth.AuthConfiguration.getSamlSsoUrl(requestAttributes.get('MyDomainUrl'), startUrl, SSO.Id);
26    // return new PageReference(ssoUrl);
27    return null;
28  }
29 
30  private PageReference discoveryResult(User user, String startUrl, Map<String, String> requestAttributes) {
31     PageReference ssoRedirect = getSsoRedirect(user, startUrl, requestAttributes);
32      if (ssoRedirect != null) {
33        return ssoRedirect;
34      } else {
35        return Auth.SessionManagement.finishLoginDiscovery(Auth.LoginDiscoveryMethod.password, user.Id);     
36      }      
37   }
38}

1// Test class for MyDomainDiscLoginDefaultHandler
2@isTest
3class MyDomainDiscLoginDefaultHandlerTest {
4    
5    /* Test Discoverable handler login. 
6       Create a user with specific email identifier and invoke login. 
7       Expected : User should be discovered and pagereference should be returned. 
8     */
9    @isTest static void testLogin() {
10        // Create user
11        String identifierEmail = getUniqueName() + '@test.org';
12        createTestUser(identifierEmail);
13        Map<String, String> requestAttributes = new Map<String, String>();
14        String startUrl = '';
15        MyDomainDiscLoginDefaultHandler myDomainDiscLoginDefaultHandler = new MyDomainDiscLoginDefaultHandler();
16        // Invoke login method from handler with the email of user created
17        PageReference  pageReference = myDomainDiscLoginDefaultHandler.login(identifierEmail, startUrl, requestAttributes);
18        // Asser page reference is returned
19        System.assertNotEquals(null, pageReference, 'Page reference was not returned');
20    }   
21    
22    /* Test Discoverable handler login with invalid (non-existing) user. 
23        Expected : Auth.LoginDiscoveryException  
24     */
25    @isTest static void testLoginWithInvalidUser() {
26        try {
27            Map<String, String> requestAttributes = new Map<String, String>();
28            String startUrl = '';
29            String uniqueName = getUniqueName();
30            String email = uniqueName + '@test.org';
31            MyDomainDiscLoginDefaultHandler myDomainDiscLoginDefaultHandler = new MyDomainDiscLoginDefaultHandler();
32            // Invoke login method from handler with non-existing user
33            myDomainDiscLoginDefaultHandler.login(email, startUrl, requestAttributes);
34        }catch (Auth.LoginDiscoveryException loginDiscoveryException) {
35            // Assert exception message
36            System.assert(loginDiscoveryException.getMessage().contains('No unique user found'), 'message=' + loginDiscoveryException.getMessage());
37        }
38    }
39    
40    /*
41       Generate a random name
42     */
43    private static String getUniqueName() { 
44        String orgId = UserInfo.getOrganizationId();
45        String dateString = String.valueof(Datetime.now()).replace(' ','').replace(':','').replace('-','');
46        Integer randomInt = Integer.valueOf(math.rint(math.random()*1000000));
47        String uniqueName = orgId + dateString + randomInt;
48        return uniqueName;
49    }
50    
51    /*
52      Create user with given email.
53     */
54    private static void createTestUser(String identifierEmail)
55    {
56        String uniqueName = getUniqueName();
57        Profile pf = [SELECT Id FROM Profile WHERE Name='Standard User'];
58        String profileID = pf.Id;
59        String fName = 'fname';
60        String lName = uniqueName + '-lname';            
61        User tuser = new User(  firstname = fName,
62                                lastName = lName,
63                                email = identifierEmail,
64                                Username = uniqueName + '@test.org',
65                                EmailEncodingKey = 'ISO-8859-1',
66                                Alias = uniqueName.substring(18, 23),
67                                TimeZoneSidKey = 'America/Los_Angeles',
68                                LocaleSidKey = 'en_US',
69                                LanguageLocaleKey = 'en_US',
70                                ProfileId = profileID);
71        insert tuser;
72    }
73}