1 files changed, 11 insertions, 1 deletions
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiBackupRestoreTest.java b/tests/wifitests/src/com/android/server/wifi/WifiBackupRestoreTest.java
index a2a1a4595..6b01f8e45 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiBackupRestoreTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiBackupRestoreTest.java
@@ -832,11 +832,13 @@ public class WifiBackupRestoreTest extends WifiBaseTest {
         List<WifiConfiguration> configurations = new ArrayList<>();
 
         WifiConfiguration wepNetwork = WifiConfigurationTestUtil.createWepNetwork();
+        wepNetwork.allowedAuthAlgorithms.set(WifiConfiguration.AuthAlgorithm.SHARED);
         wepNetwork.setIpConfiguration(
                 WifiConfigurationTestUtil.createDHCPIpConfigurationWithPacProxy());
         configurations.add(wepNetwork);
 
         WifiConfiguration pskNetwork = WifiConfigurationTestUtil.createPskNetwork();
+        pskNetwork.allowedAuthAlgorithms.set(WifiConfiguration.AuthAlgorithm.OPEN);
         pskNetwork.setIpConfiguration(
                 WifiConfigurationTestUtil.createStaticIpConfigurationWithPacProxy());
         configurations.add(pskNetwork);
@@ -1113,10 +1115,10 @@ public class WifiBackupRestoreTest extends WifiBaseTest {
             throws IOException {
         out.write("network={\n");
         out.write("        " + "ssid=" + configuration.SSID + "\n");
-        String allowedKeyManagement = "";
         if (configuration.hiddenSSID) {
             out.write("        " + "scan_ssid=1" + "\n");
         }
+        String allowedKeyManagement = "";
         if (configuration.allowedKeyManagement.get(WifiConfiguration.KeyMgmt.NONE)) {
             allowedKeyManagement += "NONE";
         }
@@ -1130,6 +1132,14 @@ public class WifiBackupRestoreTest extends WifiBaseTest {
             allowedKeyManagement += "IEEE8021X ";
         }
         out.write("        " + "key_mgmt=" + allowedKeyManagement + "\n");
+        String allowedAuthAlgorithm = "";
+        if (configuration.allowedAuthAlgorithms.get(WifiConfiguration.AuthAlgorithm.OPEN)) {
+            allowedAuthAlgorithm += "OPEN ";
+        }
+        if (configuration.allowedAuthAlgorithms.get(WifiConfiguration.AuthAlgorithm.SHARED)) {
+            allowedAuthAlgorithm += "SHARED ";
+        }
+        out.write("        " + "auth_alg=" + allowedAuthAlgorithm + "\n");
         if (configuration.preSharedKey != null) {
             out.write("        " + "psk=" + configuration.preSharedKey + "\n");
         }