1 /*
2  * Copyright (C) 2020 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package android.net.wifi.cts;
18 
19 import android.location.LocationManager;
20 import android.os.Process;
21 import android.os.UserHandle;
22 import android.test.AndroidTestCase;
23 
24 import com.android.compatibility.common.util.ShellIdentityUtils;
25 
26 /**
27  * Base test for Wifi JUnit3 tests that enables/disables location
28  */
29 public abstract class WifiJUnit3TestBase extends AndroidTestCase {
30 
31     private LocationManager mLocationManager;
32     private boolean mWasLocationEnabledForTest = false;
33 
34     @Override
setUp()35     protected void setUp() throws Exception {
36         super.setUp();
37 
38         mLocationManager = mContext.getSystemService(LocationManager.class);
39         if (!mLocationManager.isLocationEnabled()) {
40             // Turn on location if it isn't on already
41             ShellIdentityUtils.invokeWithShellPermissions(() ->
42                 mLocationManager.setLocationEnabledForUser(
43                     true, UserHandle.getUserHandleForUid(Process.myUid())));
44 
45             mWasLocationEnabledForTest = true;
46         }
47     }
48 
49     @Override
tearDown()50     protected void tearDown() throws Exception {
51         if (mWasLocationEnabledForTest) {
52             mLocationManager.setLocationEnabledForUser(
53                     false, UserHandle.getUserHandleForUid(Process.myUid()));
54         }
55 
56         super.tearDown();
57     }
58 }
59