diff --git a/tests/cts/net/src/android/net/cts/NetworkRequestTest.java b/tests/cts/net/src/android/net/cts/NetworkRequestTest.java index bca4456fc9..637ed26747 100644 --- a/tests/cts/net/src/android/net/cts/NetworkRequestTest.java +++ b/tests/cts/net/src/android/net/cts/NetworkRequestTest.java @@ -30,6 +30,7 @@ import static android.net.NetworkCapabilities.TRANSPORT_WIFI; import static junit.framework.Assert.fail; +import static org.junit.Assert.assertArrayEquals; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNotEquals; @@ -456,6 +457,21 @@ public class NetworkRequestTest { } } + @Test @IgnoreUpTo(Build.VERSION_CODES.R) + public void testGetCapabilities() { + final int[] netCapabilities = new int[] { + NET_CAPABILITY_INTERNET, + NET_CAPABILITY_NOT_ROAMING }; + final NetworkCapabilities.Builder builder = NetworkCapabilities.Builder + .withoutDefaultCapabilities(); + for (int capability : netCapabilities) builder.addCapability(capability); + final NetworkRequest nr = new NetworkRequest.Builder() + .clearCapabilities() + .setCapabilities(builder.build()) + .build(); + assertArrayEquals(netCapabilities, nr.getCapabilities()); + } + @Test public void testBuildRequestFromExistingRequestWithBuilder() { assumeTrue(TestUtils.shouldTestSApis());