From 1ccef6fefe605899484106e615b38add25da6f5d Mon Sep 17 00:00:00 2001
From: Roshan Pius <rpius@google.com>
Date: Fri, 20 Mar 2020 11:58:08 -0700
Subject: [PATCH] WifiMigration: Add test for the config store APIs

This only tests that the API does exist, there isn't any other
feasible tests for this API surface.

Bug: 150973073
Test: atest android.net.wifi.cts.WifiMigrationTest
Change-Id: I094e716027a6eb0995b6c20ed351ee12b91fce85
---
 .../net/wifi/cts/WifiMigrationTest.java       | 57 ++++++++++++++++++-
 1 file changed, 56 insertions(+), 1 deletion(-)

diff --git a/tests/cts/net/src/android/net/wifi/cts/WifiMigrationTest.java b/tests/cts/net/src/android/net/wifi/cts/WifiMigrationTest.java
index 7d94ad3ff8..c74c177039 100644
--- a/tests/cts/net/src/android/net/wifi/cts/WifiMigrationTest.java
+++ b/tests/cts/net/src/android/net/wifi/cts/WifiMigrationTest.java
@@ -16,7 +16,10 @@
 
 package android.net.wifi.cts;
 
+import android.app.ActivityManager;
 import android.net.wifi.WifiMigration;
+import android.os.UserHandle;
+import android.os.UserManager;
 import android.test.AndroidTestCase;
 
 public class WifiMigrationTest extends AndroidTestCase {
@@ -75,9 +78,61 @@ public class WifiMigrationTest extends AndroidTestCase {
      */
     public void testWifiMigrationSettings() throws Exception {
         try {
-            // ensure that this does not crash.
             WifiMigration.loadFromSettings(getContext());
         } catch (Exception ignore) {
         }
     }
+
+    /**
+     * Tests {@link WifiMigration#convertAndRetrieveSharedConfigStoreFile(int)},
+     * {@link WifiMigration#convertAndRetrieveUserConfigStoreFile(int, UserHandle)},
+     * {@link WifiMigration#removeSharedConfigStoreFile(int)} and
+     * {@link WifiMigration#removeUserConfigStoreFile(int, UserHandle)}.
+     */
+    public void testWifiMigrationConfigStore() throws Exception {
+        try {
+            WifiMigration.convertAndRetrieveSharedConfigStoreFile(
+                    WifiMigration.STORE_FILE_SHARED_GENERAL);
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.convertAndRetrieveSharedConfigStoreFile(
+                    WifiMigration.STORE_FILE_SHARED_SOFTAP);
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.convertAndRetrieveUserConfigStoreFile(
+                    WifiMigration.STORE_FILE_USER_GENERAL,
+                    UserHandle.of(ActivityManager.getCurrentUser()));
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.convertAndRetrieveUserConfigStoreFile(
+                    WifiMigration.STORE_FILE_USER_NETWORK_SUGGESTIONS,
+                    UserHandle.of(ActivityManager.getCurrentUser()));
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.removeSharedConfigStoreFile(
+                    WifiMigration.STORE_FILE_SHARED_GENERAL);
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.removeSharedConfigStoreFile(
+                    WifiMigration.STORE_FILE_SHARED_SOFTAP);
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.removeUserConfigStoreFile(
+                    WifiMigration.STORE_FILE_USER_GENERAL,
+                    UserHandle.of(ActivityManager.getCurrentUser()));
+        } catch (Exception ignore) {
+        }
+        try {
+            WifiMigration.removeUserConfigStoreFile(
+                    WifiMigration.STORE_FILE_USER_NETWORK_SUGGESTIONS,
+                    UserHandle.of(ActivityManager.getCurrentUser()));
+        } catch (Exception ignore) {
+        }
+    }
 }