From bad438386ee004a19e1f4a651d29c8fc58f45309 Mon Sep 17 00:00:00 2001
From: Roshan Pius <rpius@google.com>
Date: Tue, 3 Mar 2020 12:19:50 -0800
Subject: [PATCH] WifiNetworkSpecifierTest: Support for more network types

Bug: 150236894
Test: atest android.net.wifi.cts.WifiNetworkSpecifierTest
Change-Id: I244c3624a097c3214642923be18a3b511c9b468f
---
 .../net/src/android/net/wifi/cts/WifiNetworkSpecifierTest.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/cts/net/src/android/net/wifi/cts/WifiNetworkSpecifierTest.java b/tests/cts/net/src/android/net/wifi/cts/WifiNetworkSpecifierTest.java
index 3dd207ef8c..638991f742 100644
--- a/tests/cts/net/src/android/net/wifi/cts/WifiNetworkSpecifierTest.java
+++ b/tests/cts/net/src/android/net/wifi/cts/WifiNetworkSpecifierTest.java
@@ -363,9 +363,12 @@ public class WifiNetworkSpecifierTest extends AndroidTestCase {
             } else {
                 fail("Unsupported security type found in saved networks");
             }
+        } else if (!mTestNetwork.allowedKeyManagement.get(WifiConfiguration.KeyMgmt.OWE)) {
+            specifierBuilder.setIsEnhancedOpen(false);
         } else if (!mTestNetwork.allowedKeyManagement.get(WifiConfiguration.KeyMgmt.NONE)) {
             fail("Unsupported security type found in saved networks");
         }
+        specifierBuilder.setIsHiddenSsid(mTestNetwork.hiddenSSID);
         return specifierBuilder;
     }