@@ -38,55 +38,56 @@ public void startBrowser() {
3838
3939 @ Test
4040 void acquireTokenInteractive_ManagedUser () {
41- LabResponse labResponse = LabUserHelper . getDefaultUserMultiTenantAppPublicClient ();
42- LabUser user = labResponse .getUser ();
41+ LabResponse labResponse = LabConfigHelper . getMultiTenantAppPublicClientConfig ();
42+ UserConfig user = labResponse .getUser ();
4343
4444 assertAcquireTokenCommon (user , labResponse .getApp ().getAppId (), labResponse .getApp ().getAuthority ()+ "common" , TestConstants .GRAPH_DEFAULT_SCOPE );
4545 }
4646
4747 @ Test
4848 void acquireTokenInteractive_Arlington () {
49- LabResponse labResponse = LabUserHelper . getArlingtonUser ();
50- LabUser user = labResponse .getUser ();
49+ LabResponse labResponse = LabConfigHelper . getArlingtonConfig ();
50+ UserConfig user = labResponse .getUser ();
5151
5252 assertAcquireTokenCommon (user , labResponse .getApp ().getAppId (), labResponse .getApp ().getAuthority ()+ "common" , TestConstants .GRAPH_DEFAULT_SCOPE );
5353 }
5454
5555 @ Test ()
5656 @ DisabledIfSystemProperty (named = "adfs.disabled" , matches = "true" )
5757 void acquireTokenInteractive_ADFSv2022 () {
58- LabResponse labResponse = LabUserHelper . getDefaultAdfsUser ();
58+ LabResponse labResponse = LabConfigHelper . getAdfsConfig ();
5959
60- LabUser user = labResponse .getUser ();
60+ UserConfig user = labResponse .getUser ();
6161 assertAcquireTokenCommon (user , labResponse .getApp ().getAppId (), labResponse .getApp ().getAuthority () + "organizations/" , TestConstants .ADFS_SCOPE );
6262 }
6363
6464 @ Test
6565 void acquireTokenWithAuthorizationCode_B2C_Local () {
66- LabResponse labResponse = LabUserHelper . getB2CLocalAccount ();
67- LabUser user = labResponse .getUser ();
68- assertAcquireTokenB2C (user , TestConstants .B2C_AUTHORITY );
66+ LabResponse labResponse = LabConfigHelper . getB2CConfig ();
67+ UserConfig user = labResponse .getUser ();
68+ assertAcquireTokenB2C (user , TestConstants .B2C_AUTHORITY , labResponse . getApp (). getAppId () );
6969 }
7070
7171 @ Test
7272 void acquireTokenWithAuthorizationCode_B2C_LegacyFormat () {
73- LabResponse labResponse = LabUserHelper . getB2CLocalAccount ();
74- LabUser user = labResponse .getUser ();
75- assertAcquireTokenB2C (user , TestConstants .B2C_AUTHORITY_LEGACY_FORMAT );
73+ LabResponse labResponse = LabConfigHelper . getB2CConfig ();
74+ UserConfig user = labResponse .getUser ();
75+ assertAcquireTokenB2C (user , TestConstants .B2C_AUTHORITY_LEGACY_FORMAT , labResponse . getApp (). getAppId () );
7676 }
7777
7878 @ Test
7979 void acquireTokenInteractive_ManagedUser_InstanceAware () {
80- LabResponse labResponse = LabUserHelper .getArlingtonUser ();
81- LabUser user = labResponse .getUser ();
80+ LabResponse labResponse = LabConfigHelper .getArlingtonConfig ();
81+ UserConfig user = labResponse .getUser ();
82+
8283 assertAcquireTokenInstanceAware (user , labResponse .getApp ().getAppId (), labResponse .getLab ().getTenantId ());
8384 }
8485
8586 @ Test
8687 void acquireTokenInteractive_Ciam () {
87- LabResponse labResponse = LabUserHelper . getCiamCudUser ();
88- LabUser user = labResponse .getUser ();
89- LabApp app = labResponse .getApp ();
88+ LabResponse labResponse = LabConfigHelper . getCiamConfig ();
89+ UserConfig user = labResponse .getUser ();
90+ AppConfig app = labResponse .getApp ();
9091
9192 Map <String , String > extraQueryParameters = new HashMap <>();
9293
@@ -128,7 +129,7 @@ void acquireTokenInteractive_Ciam() {
128129 assertEquals (user .getUpn (), result .account ().username ());
129130 }
130131
131- private void assertAcquireTokenCommon (LabUser user , String appId , String authority , String scope ) {
132+ private void assertAcquireTokenCommon (UserConfig user , String appId , String authority , String scope ) {
132133 PublicClientApplication pca = IntegrationTestHelper .createPublicApp (appId , authority );
133134
134135 IAuthenticationResult result = acquireTokenInteractive (
@@ -140,23 +141,23 @@ private void assertAcquireTokenCommon(LabUser user, String appId, String authori
140141 assertEquals (user .getUpn (), result .account ().username ());
141142 }
142143
143- private void assertAcquireTokenB2C (LabUser user , String authority ) {
144+ private void assertAcquireTokenB2C (UserConfig user , String authority , String appId ) {
144145
145146 PublicClientApplication pca ;
146147 try {
147148 pca = PublicClientApplication .builder (
148- user . getAppId () ).
149+ appId ).
149150 b2cAuthority (authority + TestConstants .B2C_SIGN_IN_POLICY ).
150151 build ();
151152 } catch (MalformedURLException ex ) {
152153 throw new RuntimeException (ex .getMessage ());
153154 }
154155
155- IAuthenticationResult result = acquireTokenInteractive (user , pca , user . getAppId () );
156+ IAuthenticationResult result = acquireTokenInteractive (user , pca , appId );
156157 IntegrationTestHelper .assertAccessAndIdTokensNotNull (result );
157158 }
158159
159- private void assertAcquireTokenInstanceAware (LabUser user , String appId , String tenantId ) {
160+ private void assertAcquireTokenInstanceAware (UserConfig user , String appId , String tenantId ) {
160161 PublicClientApplication pca = IntegrationTestHelper .createPublicApp (appId , TestConstants .MICROSOFT_AUTHORITY_HOST + tenantId );
161162
162163 IAuthenticationResult result = acquireTokenInteractive_instanceAware (user , pca , TestConstants .GRAPH_DEFAULT_SCOPE );
@@ -188,7 +189,7 @@ private IAuthenticationResult acquireTokenSilently(IPublicClientApplication pca,
188189 }
189190
190191 private IAuthenticationResult acquireTokenInteractive (
191- LabUser user ,
192+ UserConfig user ,
192193 PublicClientApplication pca ,
193194 String scope ) {
194195
@@ -218,7 +219,7 @@ private IAuthenticationResult acquireTokenInteractive(
218219 }
219220
220221 private IAuthenticationResult acquireTokenInteractive_instanceAware (
221- LabUser user ,
222+ UserConfig user ,
222223 PublicClientApplication pca ,
223224 String scope ) {
224225
@@ -249,10 +250,10 @@ private IAuthenticationResult acquireTokenInteractive_instanceAware(
249250
250251 class SeleniumOpenBrowserAction implements OpenBrowserAction {
251252
252- private LabUser user ;
253+ private UserConfig user ;
253254 private PublicClientApplication pca ;
254255
255- SeleniumOpenBrowserAction (LabUser user , PublicClientApplication pca ) {
256+ SeleniumOpenBrowserAction (UserConfig user , PublicClientApplication pca ) {
256257 this .user = user ;
257258 this .pca = pca ;
258259 }
0 commit comments