tests/tests/voicesettings/src/android/voicesettings/cts/ZenModeTest.java

/*
 * Copyright (C) 2015 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 android.voicesettings.cts;

import static android.provider.Settings.ACTION_VOICE_CONTROL_DO_NOT_DISTURB_MODE;
import static android.provider.Settings.EXTRA_DO_NOT_DISTURB_MODE_ENABLED;
import static android.provider.Settings.EXTRA_DO_NOT_DISTURB_MODE_MINUTES;

import android.provider.Settings;
import android.provider.Settings.Global;
import android.util.Log;

import common.src.android.voicesettings.common.Utils;

public class ZenModeTest extends VoiceSettingsTestBase {
    static final String TAG = "ZenModeTest";

    // The following are hidden in frameworks/base/core/java/android/provider/Settings.java
    // If they weren't, we could have used them directly here.
    private static final String ZEN_MODE = "zen_mode";
    private static final int ZEN_MODE_IS_OFF = 0;
    private static final int ZEN_MODE_IS_ALARMS = 3;

    public ZenModeTest() {
        super();
    }

    public void testAll() throws Exception {
        if (!isIntentSupported(ACTION_VOICE_CONTROL_DO_NOT_DISTURB_MODE)) {
            return;
        }
        int mode;
        try {
            mode = getMode();
            Log.i(TAG, "Before testing, zen-mode is set to: " + mode);
        } catch (Settings.SettingNotFoundException e) {
            // if the mode is not supported, don't run the test.
            Log.i(TAG, "zen_mode is not found in Settings. Skipping ZenModeTest");
            return;
        }
        startTestActivity("ZEN_MODE");
        if (mode == ZEN_MODE_IS_OFF) {
            // mode is currently OFF.
            // run a test to turn it on.
            // After successful run of the test, run a test to turn it back off.
            if (!runTest(Utils.TestcaseType.ZEN_MODE_ON, ZEN_MODE_IS_ALARMS)) {
                // the test failed. don't test the next one.
                return;
            }
            runTest(Utils.TestcaseType.ZEN_MODE_OFF, ZEN_MODE_IS_OFF);
        } else {
            // mode is currently ON.
            // run a test to turn it off.
            // After successful run of the test, run a test to turn it back on.
            if (!runTest(Utils.TestcaseType.ZEN_MODE_OFF, ZEN_MODE_IS_OFF)) {
                // the test failed. don't test the next one.
                return;
            }
            runTest(Utils.TestcaseType.ZEN_MODE_ON, ZEN_MODE_IS_ALARMS);
        }
    }

    private boolean runTest(Utils.TestcaseType test, int expectedMode) throws Exception {
        if (!startTestAndWaitForBroadcast(test)) {
            return false;
        }

        // verify the test results
        int mode = getMode();
        Log.i(TAG, "After testing, zen-mode is set to: " + mode);
        assertEquals(expectedMode, mode);
        Log.i(TAG, "results_received: " + Utils.toBundleString(mResultExtras));
        assertNotNull(mResultExtras);
        if (expectedMode == ZEN_MODE_IS_ALARMS) {
            assertTrue(mResultExtras.getBoolean(EXTRA_DO_NOT_DISTURB_MODE_ENABLED));
            assertEquals(Utils.NUM_MINUTES_FOR_ZENMODE,
                    mResultExtras.getInt(EXTRA_DO_NOT_DISTURB_MODE_MINUTES));
        } else {
            assertFalse(mResultExtras.getBoolean(EXTRA_DO_NOT_DISTURB_MODE_ENABLED));
        }
        Log.i(TAG, "Successfully Tested: " + test);
        return true;
    }

    private int getMode() throws Settings.SettingNotFoundException {
        return Settings.Global.getInt(mContext.getContentResolver(), ZEN_MODE);
    }
}