@W-21933885: [MSDK Android] App Attestation Implementation (Fix AuthConfigUtilTest Timeout By Mocking Network Calls)

Author: JohnsonEricAtSalesforce

libs/test/SalesforceSDKTest/src/com/salesforce/androidsdk/util/AuthConfigUtilTest.java

@@ -35,9 +35,19 @@
 import androidx.test.ext.junit.runners.AndroidJUnit4;
 import androidx.test.filters.SmallTest;
 import com.salesforce.androidsdk.app.SalesforceSDKManager;
+import com.salesforce.androidsdk.auth.HttpAccess;
+import java.io.IOException;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Future;
 import java.util.concurrent.CompletableFuture;
+import okhttp3.Call;
+import okhttp3.OkHttpClient;
+import okhttp3.Protocol;
+import okhttp3.Request;
+import okhttp3.Response;
+import okhttp3.ResponseBody;
+import okhttp3.MediaType;
+import org.json.JSONObject;
 import org.junit.Assert;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -58,6 +68,45 @@ public class AuthConfigUtilTest {
     private static final String SANDBOX_ENDPOINT = "https://test.salesforce.com";
     private static final String FORWARD_SLASH = "/";
 
+    /**
+     * Mock HttpAccess that returns controlled responses without making real network calls.
+     */
+    private static class MockHttpAccess extends HttpAccess {
+        private final boolean shouldSucceed;
+        private final String responseBody;
+
+        public MockHttpAccess(boolean shouldSucceed, String responseBody) {
+            super(null, "MockUserAgent");
+            this.shouldSucceed = shouldSucceed;
+            this.responseBody = responseBody;
+        }
+
+        @Override
+        public OkHttpClient getOkHttpClient() {
+            return new OkHttpClient.Builder()
+                    .addInterceptor(chain -> {
+                        Request request = chain.request();
+                        Response.Builder responseBuilder = new Response.Builder()
+                                .request(request)
+                                .protocol(Protocol.HTTP_1_1)
+                                .message(shouldSucceed ? "OK" : "Not Found")
+                                .code(shouldSucceed ? 200 : 404);
+
+                        if (shouldSucceed && responseBody != null) {
+                            responseBuilder.body(ResponseBody.create(
+                                    responseBody,
+                                    MediaType.get("application/json")
+                            ));
+                        } else {
+                            responseBuilder.body(ResponseBody.create("", MediaType.get("text/plain")));
+                        }
+
+                        return responseBuilder.build();
+                    })
+                    .build();
+        }
+    }
+
     private static class TestBroadcastReceiver extends BroadcastReceiver {
         private final CompletableFuture<Intent> intentFuture = new CompletableFuture<>();
 
@@ -128,13 +177,35 @@ public void testBroadcastFails() throws ExecutionException, InterruptedException
         testBroadcast(SANDBOX_ENDPOINT, false);
     }
 
+    /**
+     * Tests that AUTH_CONFIG_COMPLETE_INTENT_ACTION broadcast is sent with correct success flag.
+     *
+     * @param endpoint The endpoint URL to test
+     * @param expected Whether the auth config request should succeed (true) or fail (false)
+     */
     private void testBroadcast(String endpoint, Boolean expected) throws InterruptedException, ExecutionException {
         final TestBroadcastReceiver receiver = new TestBroadcastReceiver();
         ContextCompat.registerReceiver(SalesforceSDKManager.getInstance().getAppContext(), receiver,
                 new IntentFilter(AuthConfigUtil.AUTH_CONFIG_COMPLETE_INTENT_ACTION), ContextCompat.RECEIVER_NOT_EXPORTED);
 
         try {
-            AuthConfigUtil.getMyDomainAuthConfig(endpoint);
+            // Create mock response body for successful auth config requests
+            String mockAuthConfigJson = "{"
+                    + "\"MobileSDK\": {"
+                    + "  \"UseAndroidNativeBrowserForAuthentication\": true,"
+                    + "  \"shareBrowserSessionAndroid\": false"
+                    + "},"
+                    + "\"SamlProviders\": [{"
+                    + "  \"SsoUrl\": \"https://example.com/sso\""
+                    + "}]"
+                    + "}";
+
+            // Create mock HttpAccess to avoid real network calls that would timeout in test environment.
+            // Mock returns 200 with JSON body for expected=true, 404 for expected=false.
+            MockHttpAccess mockHttpAccess = new MockHttpAccess(expected, expected ? mockAuthConfigJson : null);
+
+            // Call with mocked HttpAccess instead of using DEFAULT to prevent network timeout
+            AuthConfigUtil.getMyDomainAuthConfig(mockHttpAccess, endpoint);
 
             final Intent intent = receiver.getIntent().get();
             Assert.assertTrue("The intent extra should be set", intent.hasExtra(AuthConfigUtil.WAS_REQUEST_SUCCESSFUL_EXTRA));