732cdb42b2
- The AirplaneSafeNetworksSliceTest should restore Airplane Mode state after tear down - Create AirplaneModeRule to backup/restore Airplane Mode state automaticlly - Plan to reuse the AirplaneModeRule for other test case Bug: 173413889 Test: atest -c AirplaneSafeNetworksSliceTest Change-Id: Id142cf6302a188e6d82b44fcf29a395dc21f6013
77 lines
2.5 KiB
Java
77 lines
2.5 KiB
Java
/*
|
|
* Copyright (C) 2020 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package com.android.settings.testutils;
|
|
|
|
import android.content.Context;
|
|
import android.content.Intent;
|
|
import android.os.UserHandle;
|
|
import android.provider.Settings;
|
|
import android.util.Log;
|
|
|
|
import androidx.test.core.app.ApplicationProvider;
|
|
|
|
import org.junit.rules.ExternalResource;
|
|
|
|
/** A test rule that is used to manager the Airplane Mode resource for testing. */
|
|
public final class AirplaneModeRule extends ExternalResource {
|
|
|
|
private static final String TAG = "AirplaneModeRule";
|
|
|
|
private Context mContext;
|
|
private boolean mBackupValue;
|
|
private boolean mShouldRestore;
|
|
|
|
@Override
|
|
protected void before() throws Throwable {
|
|
mContext = ApplicationProvider.getApplicationContext();
|
|
}
|
|
|
|
@Override
|
|
protected void after() {
|
|
if (mShouldRestore) {
|
|
Log.d(TAG, "Restore Airplane Mode value:" + mBackupValue);
|
|
setAirplaneMode(mContext, mBackupValue);
|
|
}
|
|
}
|
|
|
|
public void setAirplaneMode(boolean enable) {
|
|
if (!mShouldRestore && isAirplaneModeOn() != enable) {
|
|
mShouldRestore = true;
|
|
mBackupValue = isAirplaneModeOn();
|
|
Log.d(TAG, "Backup Airplane Mode value:" + mBackupValue);
|
|
}
|
|
Log.d(TAG, "Set Airplane Mode enable:" + enable);
|
|
setAirplaneMode(mContext, enable);
|
|
}
|
|
|
|
public boolean isAirplaneModeOn() {
|
|
return Settings.Global.getInt(mContext.getContentResolver(),
|
|
Settings.Global.AIRPLANE_MODE_ON, 0) != 0;
|
|
}
|
|
|
|
private static void setAirplaneMode(Context context, boolean enable) {
|
|
// Change the system setting
|
|
Settings.Global.putInt(context.getContentResolver(), Settings.Global.AIRPLANE_MODE_ON,
|
|
enable ? 1 : 0);
|
|
|
|
// Post the intent
|
|
final Intent intent = new Intent(Intent.ACTION_AIRPLANE_MODE_CHANGED);
|
|
intent.putExtra("state", enable);
|
|
context.sendBroadcastAsUser(intent, UserHandle.ALL);
|
|
}
|
|
}
|