[CTS] Add location mode check
Bug: 220356755
Test: cts ConnectedNetworkScorerTest
Change-Id: I2264e585d95659319fbe2d08433f465b39bf8b93
diff --git a/tests/tests/wifi/src/android/net/wifi/cts/ConnectedNetworkScorerTest.java b/tests/tests/wifi/src/android/net/wifi/cts/ConnectedNetworkScorerTest.java
index 542456e1..94e95b1 100644
--- a/tests/tests/wifi/src/android/net/wifi/cts/ConnectedNetworkScorerTest.java
+++ b/tests/tests/wifi/src/android/net/wifi/cts/ConnectedNetworkScorerTest.java
@@ -47,6 +47,7 @@
import android.annotation.NonNull;
import android.app.UiAutomation;
import android.content.Context;
+import android.location.LocationManager;
import android.net.ConnectivityManager;
import android.net.DhcpOption;
import android.net.wifi.WifiConfiguration;
@@ -81,6 +82,7 @@
import java.util.ArrayList;
import java.util.List;
import java.util.NoSuchElementException;
+import java.util.Objects;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.Executors;
@@ -116,6 +118,9 @@
mWifiManager = mContext.getSystemService(WifiManager.class);
assertThat(mWifiManager).isNotNull();
+ // Location mode must be enabled, otherwise the connection info will be redacted.
+ assertThat(Objects.requireNonNull(mContext.getSystemService(LocationManager.class))
+ .isLocationEnabled()).isTrue();
mConnectivityManager = mContext.getSystemService(ConnectivityManager.class);
