diff --git a/tests/cts/hostside/app/src/com/android/cts/net/hostside/AbstractRestrictBackgroundNetworkTestCase.java b/tests/cts/hostside/app/src/com/android/cts/net/hostside/AbstractRestrictBackgroundNetworkTestCase.java index 08df9e1bbc..08c8fdbcee 100644 --- a/tests/cts/hostside/app/src/com/android/cts/net/hostside/AbstractRestrictBackgroundNetworkTestCase.java +++ b/tests/cts/hostside/app/src/com/android/cts/net/hostside/AbstractRestrictBackgroundNetworkTestCase.java @@ -71,6 +71,8 @@ abstract class AbstractRestrictBackgroundNetworkTestCase extends Instrumentation // Must be higher than NETWORK_TIMEOUT_MS private static final int ORDERED_BROADCAST_TIMEOUT_MS = NETWORK_TIMEOUT_MS * 4; + private static final boolean ASSERT_NETWORK_INFO_STATE = false; + protected Context mContext; protected Instrumentation mInstrumentation; protected ConnectivityManager mCm; @@ -220,15 +222,19 @@ abstract class AbstractRestrictBackgroundNetworkTestCase extends Instrumentation if (expectAvailable) { assertTrue("should be connected: " + connectionCheckDetails + " (network info: " + networkInfo + ")", connected); - assertEquals("wrong state for " + networkInfo, State.CONNECTED, state); - assertEquals("wrong detailed state for " + networkInfo, - DetailedState.CONNECTED, detailedState); + if (ASSERT_NETWORK_INFO_STATE) { + assertEquals("wrong state for " + networkInfo, State.CONNECTED, state); + assertEquals("wrong detailed state for " + networkInfo, + DetailedState.CONNECTED, detailedState); + } } else { assertFalse("should not be connected: " + connectionCheckDetails + " (network info: " + networkInfo + ")", connected); - assertEquals("wrong state for " + networkInfo, State.DISCONNECTED, state); - assertEquals("wrong detailed state for " + networkInfo, - DetailedState.BLOCKED, detailedState); + if (ASSERT_NETWORK_INFO_STATE) { + assertEquals("wrong state for " + networkInfo, State.DISCONNECTED, state); + assertEquals("wrong detailed state for " + networkInfo, + DetailedState.BLOCKED, detailedState); + } } }