Enhance aad multi tenant it (Azure#18586)

Author: yiliuTo

sdk/spring/azure-spring-boot-test-aad/src/test/java/com/azure/test/aad/selenium/AADSeleniumITHelper.java

@@ -1,20 +1,21 @@
 package com.azure.test.aad.selenium;
 
+import com.azure.test.aad.common.SeleniumITHelper;
+import org.junit.Assert;
+import org.openqa.selenium.By;
+import org.openqa.selenium.Keys;
+import org.openqa.selenium.support.ui.ExpectedConditions;
+
+import java.util.HashMap;
+import java.util.Map;
+
 import static com.azure.spring.test.EnvironmentVariable.AAD_SINGLE_TENANT_CLIENT_ID;
 import static com.azure.spring.test.EnvironmentVariable.AAD_SINGLE_TENANT_CLIENT_SECRET;
 import static com.azure.spring.test.EnvironmentVariable.AAD_TENANT_ID_1;
 import static com.azure.spring.test.EnvironmentVariable.AAD_USER_NAME_1;
 import static com.azure.spring.test.EnvironmentVariable.AAD_USER_PASSWORD_1;
 import static org.openqa.selenium.support.ui.ExpectedConditions.presenceOfElementLocated;
 
-import com.azure.test.aad.common.SeleniumITHelper;
-import java.util.HashMap;
-import java.util.Map;
-import org.junit.Assert;
-import org.openqa.selenium.By;
-import org.openqa.selenium.Keys;
-import org.openqa.selenium.support.ui.ExpectedConditions;
-
 public class AADSeleniumITHelper extends SeleniumITHelper {
 
     private String username;
@@ -31,9 +32,13 @@ public static Map<String, String> createDefaultProperties() {
     }
 
     public AADSeleniumITHelper(Class<?> appClass, Map<String, String> properties) {
+        this(appClass, properties, AAD_USER_NAME_1, AAD_USER_PASSWORD_1);
+    }
+
+    public AADSeleniumITHelper(Class<?> appClass, Map<String, String> properties, String username, String password) {
         super(appClass, properties);
-        username = AAD_USER_NAME_1;
-        password = AAD_USER_PASSWORD_1;
+        this.username = username;
+        this.password = password;
     }
 
     public void logIn() {

sdk/spring/azure-spring-boot-test-aad/src/test/java/com/azure/test/aad/selenium/multitenant/AADMultipleTenantIT.java

@@ -0,0 +1,61 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.test.aad.selenium.multitenant;
+
+import com.azure.test.aad.selenium.AADSeleniumITHelper;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.http.ResponseEntity;
+import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity;
+import org.springframework.security.oauth2.client.authentication.OAuth2AuthenticationToken;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.security.Principal;
+import java.util.HashMap;
+import java.util.Map;
+
+import static com.azure.spring.test.EnvironmentVariable.AAD_MULTI_TENANT_CLIENT_ID;
+import static com.azure.spring.test.EnvironmentVariable.AAD_MULTI_TENANT_CLIENT_SECRET;
+import static com.azure.spring.test.EnvironmentVariable.AAD_USER_NAME_2;
+import static com.azure.spring.test.EnvironmentVariable.AAD_USER_PASSWORD_2;
+
+public class AADMultipleTenantIT {
+    private static final Logger LOGGER = LoggerFactory.getLogger(AADMultipleTenantIT.class);
+    private AADSeleniumITHelper aadSeleniumITHelper;
+
+    @Test
+    public void multipleTenantTest() {
+        Map<String, String> properties = new HashMap<>();
+        properties.put("azure.activedirectory.client-id", AAD_MULTI_TENANT_CLIENT_ID);
+        properties.put("azure.activedirectory.client-secret", AAD_MULTI_TENANT_CLIENT_SECRET);
+        aadSeleniumITHelper = new AADSeleniumITHelper(DumbApp.class, properties,
+            AAD_USER_NAME_2, AAD_USER_PASSWORD_2);
+        aadSeleniumITHelper.logIn();
+
+        String httpResponse = aadSeleniumITHelper.httpGet("api/home");
+        Assert.assertTrue(httpResponse.contains("home"));
+    }
+
+    @After
+    public void destroy() {
+        aadSeleniumITHelper.destroy();
+    }
+
+    @EnableGlobalMethodSecurity(securedEnabled = true, prePostEnabled = true)
+    @SpringBootApplication
+    @RestController
+    public static class DumbApp {
+
+        @GetMapping(value = "/api/home")
+        public ResponseEntity<String> home(Principal principal) {
+            LOGGER.info(((OAuth2AuthenticationToken) principal).getAuthorities().toString());
+            return ResponseEntity.ok("home");
+        }
+    }
+}