sl4f_lib/factory_reset/
facade.rs

1// Copyright 2020 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use crate::common_utils::common::LazyProxy;
6use crate::common_utils::common::macros::{fx_err_and_bail, with_line};
7use anyhow::Error;
8use fidl_fuchsia_recovery::{FactoryResetMarker, FactoryResetProxy};
9use log::info;
10
11/// Perform factory reset fidl operations.
12///
13/// Note this object is shared among all threads created by server.
14///
15#[derive(Debug)]
16pub struct FactoryResetFacade {
17    factory_reset_manager: LazyProxy<FactoryResetMarker>,
18}
19
20impl FactoryResetFacade {
21    pub fn new() -> FactoryResetFacade {
22        FactoryResetFacade { factory_reset_manager: Default::default() }
23    }
24
25    /// Returns the proxy provided on instantiation or establishes a new connection.
26    fn factory_reset_manager(&self) -> Result<FactoryResetProxy, Error> {
27        self.factory_reset_manager.get_or_connect()
28    }
29
30    /// Returns the pairing code from the FactoryDataManager proxy service.
31    pub async fn factory_reset(&self) -> Result<(), Error> {
32        let tag = "FactoryResetFacade::factory_reset";
33        info!("Executing factory reset");
34        match self.factory_reset_manager()?.reset().await {
35            Ok(_) => Ok(()),
36            Err(e) => fx_err_and_bail!(
37                &with_line!(tag),
38                format_err!("FIDL call failed with error: {}", e)
39            ),
40        }
41    }
42}
43
44#[cfg(test)]
45mod tests {
46    use super::*;
47    use fidl::endpoints::create_proxy_and_stream;
48    use fidl_fuchsia_recovery::FactoryResetRequest;
49    use fuchsia_async as fasync;
50    use futures::prelude::*;
51    use std::sync::LazyLock;
52
53    static RESULT: LazyLock<i32> = LazyLock::new(|| 0);
54
55    #[fasync::run_singlethreaded(test)]
56    async fn test_factory_reset() {
57        let (proxy, mut stream) = create_proxy_and_stream::<FactoryResetMarker>();
58        let facade = FactoryResetFacade::new();
59        facade.factory_reset_manager.set(proxy).unwrap();
60        let facade_fut = async move {
61            assert_eq!(facade.factory_reset().await.ok(), Some(()));
62        };
63        let stream_fut = async move {
64            match stream.try_next().await {
65                Ok(Some(FactoryResetRequest::Reset { responder })) => {
66                    responder.send((*RESULT).clone()).unwrap();
67                }
68                err => panic!("Error in request handler: {:?}", err),
69            }
70        };
71        future::join(facade_fut, stream_fut).await;
72    }
73}
sl4f_lib/factory_reset/facade.rs

sl4f_lib/factory_reset/
facade.rs
