From bf6a2ad00c25114bf07d12480d72e97cb115f06e Mon Sep 17 00:00:00 2001
From: Pavel Maltsev <pavelm@google.com>
Date: Mon, 2 Apr 2018 11:15:04 -0700
Subject: [PATCH] Add cts to test unwanted capabilities api

Test: atest android.net.cts.NetworkRequestTest
Bug: 74945408
Bug: 72828388
Change-Id: I19099a3d7a555f610c96eb1859493903152e10a3
---
 .../android/net/cts/NetworkRequestTest.java   | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/tests/cts/net/src/android/net/cts/NetworkRequestTest.java b/tests/cts/net/src/android/net/cts/NetworkRequestTest.java
index c862c77bc7..2a5ef31847 100644
--- a/tests/cts/net/src/android/net/cts/NetworkRequestTest.java
+++ b/tests/cts/net/src/android/net/cts/NetworkRequestTest.java
@@ -30,6 +30,35 @@ public class NetworkRequestTest extends AndroidTestCase {
                 .hasCapability(NET_CAPABILITY_MMS));
     }
 
+    public void testUnwantedCapabilities() {
+        assertTrue(new NetworkRequest.Builder()
+                .addUnwantedCapability(NET_CAPABILITY_MMS)
+                .build()
+                .hasUnwantedCapability(NET_CAPABILITY_MMS));
+        assertFalse(new NetworkRequest.Builder()
+                .removeCapability(NET_CAPABILITY_MMS)
+                .build()
+                .hasCapability(NET_CAPABILITY_MMS));
+    }
+
+    public void testCapabilityMutualExclusivity() {
+        NetworkRequest.Builder reqBuilder = new NetworkRequest.Builder()
+                .addCapability(NET_CAPABILITY_MMS);
+
+        assertTrue(reqBuilder.build().hasCapability(NET_CAPABILITY_MMS));
+        assertFalse(reqBuilder.build().hasUnwantedCapability(NET_CAPABILITY_MMS));
+
+        // Move capability to unwanted list
+        reqBuilder.addUnwantedCapability(NET_CAPABILITY_MMS);
+        assertFalse(reqBuilder.build().hasCapability(NET_CAPABILITY_MMS));
+        assertTrue(reqBuilder.build().hasUnwantedCapability(NET_CAPABILITY_MMS));
+
+        // Move it back to the list of capabilities
+        reqBuilder.addCapability(NET_CAPABILITY_MMS);
+        assertTrue(reqBuilder.build().hasCapability(NET_CAPABILITY_MMS));
+        assertFalse(reqBuilder.build().hasUnwantedCapability(NET_CAPABILITY_MMS));
+    }
+
     public void testTransports() {
         assertTrue(new NetworkRequest.Builder().addTransportType(TRANSPORT_BLUETOOTH).build()
                 .hasTransport(TRANSPORT_BLUETOOTH));