add test

AzureAD · Nov 30, 2023 · 2058c0c · 2058c0c
1 parent 0228bdb
commit 2058c0c
Showing 1 changed file with 36 additions and 1 deletion.
diff --git a/msal/src/test/java/com/microsoft/identity/client/CommandParametersTest.java b/msal/src/test/java/com/microsoft/identity/client/CommandParametersTest.java
@@ -26,6 +26,7 @@
 import android.content.Context;
 
 import androidx.annotation.NonNull;
+import androidx.annotation.Nullable;
 import androidx.test.core.app.ApplicationProvider;
 
 import com.microsoft.identity.client.claims.ClaimsRequest;
@@ -42,6 +43,7 @@
 import com.microsoft.identity.common.java.constants.FidoConstants;
 import com.microsoft.identity.common.java.providers.oauth2.OAuth2TokenCache;
 import com.microsoft.identity.common.java.exception.ClientException;
+import com.microsoft.identity.common.java.ui.PreferredAuthMethod;
 
 import org.junit.Assert;
 import org.junit.Before;
@@ -61,7 +63,7 @@
 import java.util.UUID;
 
 @RunWith(RobolectricTestRunner.class)
-@Ignore("Passkey feature in development.")
+//@Ignore("Passkey feature in development.")
 public class CommandParametersTest {
 
     private static final String AAD_CP1_CONFIG_FILE = "src/test/res/raw/aad_capabilities_cp1.json";
@@ -142,6 +144,28 @@ public void testAcquireTokenOperationWithCorrelationId() throws ClientException
         Assert.assertEquals(correlationId.toString(), commandParameters.getCorrelationId());
     }
 
+    @Test
+    public void testAcquireTokenOperationWithPreferredAuthMethod() throws ClientException {
+
+        InteractiveTokenCommandParameters commandParameters = CommandParametersAdapter.createInteractiveTokenCommandParameters(
+                getConfiguration(AAD_NONE_CONFIG_FILE),
+                getCache(),
+                getAcquireTokenParametersPreferredAuthMethod(PreferredAuthMethod.QR)
+        );
+        Assert.assertEquals(PreferredAuthMethod.QR, commandParameters.getPreferredAuthMethod());
+    }
+
+    @Test
+    public void testAcquireTokenOperationWithNoPreferredAuthMethod() throws ClientException {
+
+        InteractiveTokenCommandParameters commandParameters = CommandParametersAdapter.createInteractiveTokenCommandParameters(
+                getConfiguration(AAD_NONE_CONFIG_FILE),
+                getCache(),
+                getAcquireTokenParametersPreferredAuthMethod(null)
+        );
+        Assert.assertNull(commandParameters.getPreferredAuthMethod());
+    }
+
     @Test
     public void testAcquireTokenSilentOperationWithoutCorrelationId() throws ClientException {
         SilentTokenCommandParameters commandParameters = CommandParametersAdapter.createSilentTokenCommandParameters(getConfiguration(AAD_CP1_CONFIG_FILE), getCache(), getAcquireTokenSilentParametersWithoutCorrelationId());
@@ -336,6 +360,17 @@ private AcquireTokenSilentParameters getAcquireTokenSilentParametersWithCorrelat
         return parameters;
     }
 
+    private AcquireTokenParameters getAcquireTokenParametersPreferredAuthMethod(final @Nullable PreferredAuthMethod preferredAuthMethod) {
+        final AcquireTokenParameters.Builder parametersBuilder = new AcquireTokenParameters.Builder()
+                .withClaims(getAccessTokenClaimsRequest("device_id", ""))
+                .withScopes(new ArrayList<String>(Arrays.asList("User.Read")))
+                .startAuthorizationFromActivity(mActivity);
+        if (preferredAuthMethod != null) {
+            parametersBuilder.withPreferredAuthMethod(preferredAuthMethod);
+        }
+        return parametersBuilder.build();
+    }
+
     private AcquireTokenParameters getAcquireTokenParametersWithoutCorrelationId() {
         AcquireTokenParameters parameters = new AcquireTokenParameters.Builder()
                 .withClaims(getAccessTokenClaimsRequest("device_id", ""))