From 2ac691c021713f0e3f738a1a18f50b196595783a Mon Sep 17 00:00:00 2001
From: David Su <dysu@google.com>
Date: Thu, 5 Nov 2020 12:53:56 -0800
Subject: [PATCH] Add mainline TEST_MAPPING to run S module on R base image

Bug: 169443762
Test: presubmit
Change-Id: Id22217964450c14b4ef09b76a4366e66191c347d
---
 wifi/TEST_MAPPING          | 13 +++++++++++++
 wifi/tests/Android.bp      |  2 +-
 wifi/tests/AndroidTest.xml |  2 ++
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/wifi/TEST_MAPPING b/wifi/TEST_MAPPING
index fde3a6aa993c2..8c515109a3095 100644
--- a/wifi/TEST_MAPPING
+++ b/wifi/TEST_MAPPING
@@ -8,5 +8,18 @@
         }
       ]
     }
+  ],
+  "mainline-presubmit": [
+    {
+      "name": "FrameworksWifiApiTests[com.google.android.wifi.apex]"
+    },
+    {
+      "name": "CtsWifiTestCases[com.google.android.wifi.apex]",
+      "options": [
+        {
+          "exclude-annotation": "android.net.wifi.cts.VirtualDeviceNotSupported"
+        }
+      ]
+    }
   ]
 }
diff --git a/wifi/tests/Android.bp b/wifi/tests/Android.bp
index 6a39959e8cfd5..cb9693d3869e0 100644
--- a/wifi/tests/Android.bp
+++ b/wifi/tests/Android.bp
@@ -44,7 +44,7 @@ android_test {
     ],
 
     test_suites: [
-        "device-tests",
+        "general-tests",
         "mts",
     ],
 }
diff --git a/wifi/tests/AndroidTest.xml b/wifi/tests/AndroidTest.xml
index 34e2e3af9cdac..2cdaddb590184 100644
--- a/wifi/tests/AndroidTest.xml
+++ b/wifi/tests/AndroidTest.xml
@@ -20,6 +20,8 @@
 
     <option name="test-suite-tag" value="apct" />
     <option name="test-tag" value="FrameworksWifiApiTests" />
+    <option name="config-descriptor:metadata" key="mainline-param"
+            value="com.google.android.wifi.apex" />
     <test class="com.android.tradefed.testtype.AndroidJUnitTest" >
         <option name="package" value="android.net.wifi.test" />
         <option name="runner" value="androidx.test.runner.AndroidJUnitRunner" />