From db884c9a93de747c96d0665ebbbf9af2047f862f Mon Sep 17 00:00:00 2001
From: Nick Wille <nickwille@google.com>
Date: Thu, 30 Nov 2023 22:04:12 +0000
Subject: [PATCH] Revert "Update NetworkAgent#register to compatible with old
 NetworkAgentTest"

This reverts commit 1e7c4a494d89f9a1bd24f1eeca42f7e6a434fea8.

This change is a suspect for causing test breakages, it may or may not be reverted.

Bug: 314087796

Reason for revert: 314087796

Change-Id: If0a1c1e2d0e260319cf64b2d00828638abebb7fe
---
 framework/src/android/net/NetworkAgent.java   | 19 ++++-----------
 .../src/android/net/cts/NetworkAgentTest.kt   | 23 ++++++++++---------
 2 files changed, 17 insertions(+), 25 deletions(-)

diff --git a/framework/src/android/net/NetworkAgent.java b/framework/src/android/net/NetworkAgent.java
index 574ab2fe1b..4e9087cbd3 100644
--- a/framework/src/android/net/NetworkAgent.java
+++ b/framework/src/android/net/NetworkAgent.java
@@ -756,20 +756,11 @@ public abstract class NetworkAgent {
             }
             final ConnectivityManager cm = (ConnectivityManager) mInitialConfiguration.context
                     .getSystemService(Context.CONNECTIVITY_SERVICE);
-            if (mInitialConfiguration.localNetworkConfig == null) {
-                // Call registerNetworkAgent without localNetworkConfig argument to pass
-                // android.net.cts.NetworkAgentTest#testAgentStartsInConnecting in old cts
-                mNetwork = cm.registerNetworkAgent(new NetworkAgentBinder(mHandler),
-                        new NetworkInfo(mInitialConfiguration.info),
-                        mInitialConfiguration.properties, mInitialConfiguration.capabilities,
-                        mInitialConfiguration.score, mInitialConfiguration.config, providerId);
-            } else {
-                mNetwork = cm.registerNetworkAgent(new NetworkAgentBinder(mHandler),
-                        new NetworkInfo(mInitialConfiguration.info),
-                        mInitialConfiguration.properties, mInitialConfiguration.capabilities,
-                        mInitialConfiguration.localNetworkConfig, mInitialConfiguration.score,
-                        mInitialConfiguration.config, providerId);
-            }
+            mNetwork = cm.registerNetworkAgent(new NetworkAgentBinder(mHandler),
+                    new NetworkInfo(mInitialConfiguration.info),
+                    mInitialConfiguration.properties, mInitialConfiguration.capabilities,
+                    mInitialConfiguration.localNetworkConfig, mInitialConfiguration.score,
+                    mInitialConfiguration.config, providerId);
             mInitialConfiguration = null; // All this memory can now be GC'd
         }
         return mNetwork;
diff --git a/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt b/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt
index c7d65555e3..225408c4cc 100644
--- a/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt
+++ b/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt
@@ -131,6 +131,17 @@ import com.android.testutils.TestableNetworkCallback
 import com.android.testutils.assertThrows
 import com.android.testutils.runAsShell
 import com.android.testutils.tryTest
+import org.junit.After
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.ArgumentMatchers.any
+import org.mockito.ArgumentMatchers.argThat
+import org.mockito.ArgumentMatchers.eq
+import org.mockito.Mockito.doReturn
+import org.mockito.Mockito.mock
+import org.mockito.Mockito.timeout
+import org.mockito.Mockito.verify
 import java.io.Closeable
 import java.io.IOException
 import java.net.DatagramSocket
@@ -149,17 +160,6 @@ import kotlin.test.assertNotNull
 import kotlin.test.assertNull
 import kotlin.test.assertTrue
 import kotlin.test.fail
-import org.junit.After
-import org.junit.Before
-import org.junit.Test
-import org.junit.runner.RunWith
-import org.mockito.ArgumentMatchers.any
-import org.mockito.ArgumentMatchers.argThat
-import org.mockito.ArgumentMatchers.eq
-import org.mockito.Mockito.doReturn
-import org.mockito.Mockito.mock
-import org.mockito.Mockito.timeout
-import org.mockito.Mockito.verify
 
 private const val TAG = "NetworkAgentTest"
 // This test doesn't really have a constraint on how fast the methods should return. If it's
@@ -951,6 +951,7 @@ class NetworkAgentTest {
                 argThat<NetworkInfo> { it.detailedState == NetworkInfo.DetailedState.CONNECTING },
                 any(LinkProperties::class.java),
                 any(NetworkCapabilities::class.java),
+                any(), // LocalNetworkConfig TODO : specify when it's public
                 any(NetworkScore::class.java),
                 any(NetworkAgentConfig::class.java),
                 eq(NetworkProvider.ID_NONE))
-- 
GitLab