src/settings/service/src/tests/do_not_disturb_tests.rs

// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use crate::agent::storage::storage_factory::testing::InMemoryStorageFactory;
use crate::do_not_disturb::types::DoNotDisturbInfo;
use crate::ingress::fidl::Interface;
use crate::EnvironmentBuilder;
use fidl_fuchsia_settings::{DoNotDisturbMarker, DoNotDisturbProxy, DoNotDisturbSettings};
use std::sync::Arc;

const ENV_NAME: &str = "settings_service_do_not_disturb_test_environment";

#[fuchsia_async::run_until_stalled(test)]
async fn test_do_not_disturb() {
    // Prepopulate initial value.
    let storage_factory =
        InMemoryStorageFactory::with_initial_data(&DoNotDisturbInfo::new(true, false));

    let env = EnvironmentBuilder::new(Arc::new(storage_factory))
        .fidl_interfaces(&[Interface::DoNotDisturb])
        .spawn_and_get_nested_environment(ENV_NAME)
        .await
        .unwrap();

    let dnd_proxy = env.connect_to_protocol::<DoNotDisturbMarker>().expect("connected to service");

    verify_dnd_watch(&dnd_proxy, DoNotDisturbInfo::new(true, false)).await;

    set_dnd(&dnd_proxy, None, Some(true)).await;

    verify_dnd_watch(&dnd_proxy, DoNotDisturbInfo::new(true, true)).await;

    set_dnd(&dnd_proxy, Some(false), None).await;

    verify_dnd_watch(&dnd_proxy, DoNotDisturbInfo::new(false, true)).await;

    set_dnd(&dnd_proxy, Some(false), Some(false)).await;

    verify_dnd_watch(&dnd_proxy, DoNotDisturbInfo::new(false, false)).await;

    set_dnd(&dnd_proxy, Some(true), Some(true)).await;

    verify_dnd_watch(&dnd_proxy, DoNotDisturbInfo::new(true, true)).await;
}

async fn set_dnd(
    dnd_proxy: &DoNotDisturbProxy,
    user_dnd: Option<bool>,
    night_mode_dnd: Option<bool>,
) {
    let mut dnd_settings = DoNotDisturbSettings::EMPTY;
    dnd_settings.user_initiated_do_not_disturb = user_dnd;
    dnd_settings.night_mode_initiated_do_not_disturb = night_mode_dnd;
    dnd_proxy.set(dnd_settings).await.expect("set completed").expect("set successful");
}

async fn verify_dnd_watch(dnd_proxy: &DoNotDisturbProxy, expected_dnd: DoNotDisturbInfo) {
    let settings = dnd_proxy.watch().await.expect("watch completed");
    assert_eq!(settings.user_initiated_do_not_disturb, expected_dnd.user_dnd);
    assert_eq!(settings.night_mode_initiated_do_not_disturb, expected_dnd.night_mode_dnd);
}