src/settings/service/src/factory_reset/factory_reset_fidl_handler.rs

// Copyright 2020 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::base::{SettingInfo, SettingType};
use crate::handler::base::{Request, Response};
use crate::ingress::{request, watch, Scoped};
use crate::job::source::ErrorResponder;
use crate::job::Job;
use fidl::prelude::*;
use fidl_fuchsia_settings::{
    Error, FactoryResetRequest, FactoryResetSetResponder, FactoryResetSetResult,
    FactoryResetSettings, FactoryResetWatchResponder,
};

use crate::job::source::Error as JobError;

impl ErrorResponder for FactoryResetSetResponder {
    fn id(&self) -> &'static str {
        "FactoryReset_Set"
    }

    fn respond(self: Box<Self>, error: fidl_fuchsia_settings::Error) -> Result<(), fidl::Error> {
        self.send(Err(error))
    }
}

impl From<Response> for Scoped<FactoryResetSetResult> {
    fn from(response: Response) -> Self {
        Scoped(response.map_or(Err(Error::Failed), |_| Ok(())))
    }
}

impl request::Responder<Scoped<FactoryResetSetResult>> for FactoryResetSetResponder {
    fn respond(self, Scoped(response): Scoped<FactoryResetSetResult>) {
        let _ = self.send(response);
    }
}

impl watch::Responder<FactoryResetSettings, fuchsia_zircon::Status> for FactoryResetWatchResponder {
    fn respond(self, response: Result<FactoryResetSettings, fuchsia_zircon::Status>) {
        match response {
            Ok(settings) => {
                let _ = self.send(&settings);
            }
            Err(error) => {
                self.control_handle().shutdown_with_epitaph(error);
            }
        }
    }
}

impl TryFrom<FactoryResetRequest> for Job {
    type Error = JobError;
    fn try_from(item: FactoryResetRequest) -> Result<Self, Self::Error> {
        #[allow(unreachable_patterns)]
        match item {
            FactoryResetRequest::Set { settings, responder } => match to_request(settings) {
                Some(request) => {
                    Ok(request::Work::new(SettingType::FactoryReset, request, responder).into())
                }
                None => Err(JobError::InvalidInput(Box::new(responder))),
            },
            FactoryResetRequest::Watch { responder } => {
                Ok(watch::Work::new_job(SettingType::FactoryReset, responder))
            }
            _ => {
                tracing::warn!("Received a call to an unsupported API: {:?}", item);
                Err(JobError::Unsupported)
            }
        }
    }
}

impl From<SettingInfo> for FactoryResetSettings {
    fn from(response: SettingInfo) -> Self {
        if let SettingInfo::FactoryReset(info) = response {
            FactoryResetSettings {
                is_local_reset_allowed: Some(info.is_local_reset_allowed),
                ..Default::default()
            }
        } else {
            panic!("incorrect value sent to factory_reset");
        }
    }
}

fn to_request(settings: FactoryResetSettings) -> Option<Request> {
    settings.is_local_reset_allowed.map(Request::SetLocalResetAllowed)
}
#[cfg(test)]
mod tests {
    use super::*;
    use crate::job::{execution, work};
    use assert_matches::assert_matches;
    use fidl_fuchsia_settings::{FactoryResetMarker, FactoryResetRequestStream};
    use futures::StreamExt;

    #[fuchsia::test]
    fn to_request_maps_correctly() {
        let result = to_request(FactoryResetSettings {
            is_local_reset_allowed: Some(true),
            ..Default::default()
        });
        assert_matches!(result, Some(Request::SetLocalResetAllowed(true)));
    }

    #[fuchsia::test(allow_stalls = false)]
    async fn try_from_set_handles_missing_params() {
        let (proxy, server) = fidl::endpoints::create_proxy::<FactoryResetMarker>()
            .expect("should be able to create proxy");
        let _fut = proxy.set(&FactoryResetSettings::default());
        let mut request_stream: FactoryResetRequestStream =
            server.into_stream().expect("should be able to convert to stream");
        let request = request_stream
            .next()
            .await
            .expect("should have on request before stream is closed")
            .expect("should have gotten a request");
        let job = Job::try_from(request);
        assert_matches!(job, Err(crate::job::source::Error::InvalidInput(_)));
    }

    #[fuchsia::test(allow_stalls = false)]
    async fn try_from_set_converts_supplied_params() {
        let (proxy, server) = fidl::endpoints::create_proxy::<FactoryResetMarker>()
            .expect("should be able to create proxy");
        let _fut = proxy.set(&FactoryResetSettings {
            is_local_reset_allowed: Some(true),
            ..Default::default()
        });
        let mut request_stream: FactoryResetRequestStream =
            server.into_stream().expect("should be able to convert to stream");
        let request = request_stream
            .next()
            .await
            .expect("should have on request before stream is closed")
            .expect("should have gotten a request");
        let job = Job::try_from(request);
        let job = job.as_ref();
        assert_matches!(job.map(|j| j.workload()), Ok(work::Load::Independent(_)));
        assert_matches!(job.map(|j| j.execution_type()), Ok(execution::Type::Independent));
    }

    #[fuchsia::test(allow_stalls = false)]
    async fn try_from_watch_converts_supplied_params() {
        let (proxy, server) = fidl::endpoints::create_proxy::<FactoryResetMarker>()
            .expect("should be able to create proxy");
        let _fut = proxy.watch();
        let mut request_stream: FactoryResetRequestStream =
            server.into_stream().expect("should be able to convert to stream");
        let request = request_stream
            .next()
            .await
            .expect("should have on request before stream is closed")
            .expect("should have gotten a request");
        let job = Job::try_from(request);
        let job = job.as_ref();
        assert_matches!(job.map(|j| j.workload()), Ok(work::Load::Sequential(_, _)));
        assert_matches!(job.map(|j| j.execution_type()), Ok(execution::Type::Sequential(_)));
    }
}