power_manager_integration_test_lib/mocks/
admin.rs

1// Copyright 2022 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 fidl::endpoints::ServerEnd;
6use fidl_fuchsia_hardware_power_statecontrol::{self as fpower, RebootOptions};
7use fidl_fuchsia_io::DirectoryMarker;
8use fuchsia_async as fasync;
9use fuchsia_component::server::ServiceFs;
10use fuchsia_component_test::LocalComponentHandles;
11use futures::channel::mpsc;
12use futures::lock::Mutex;
13use futures::{StreamExt, TryStreamExt};
14use log::*;
15use std::sync::Arc;
16
17/// Mocks the fuchsia.hardware.power.statecontrol.Admin service to be used in integration tests.
18pub struct MockStateControlAdminService {
19    shutdown_received_sender: Mutex<mpsc::Sender<RebootOptions>>,
20    shutdown_received_receiver: Mutex<mpsc::Receiver<RebootOptions>>,
21}
22
23impl MockStateControlAdminService {
24    pub fn new() -> Arc<MockStateControlAdminService> {
25        let (sender, receiver) = mpsc::channel(1);
26        Arc::new(Self {
27            shutdown_received_sender: Mutex::new(sender),
28            shutdown_received_receiver: Mutex::new(receiver),
29        })
30    }
31
32    /// Runs the mock using the provided `LocalComponentHandles`.
33    ///
34    /// The mock intentionally does not complete reboot requests in order to prevent further calls
35    /// that may cause extra send.
36    ///
37    /// Expected usage is to call this function from a closure for the
38    /// `local_component_implementation` parameter to `RealmBuilder.add_local_child`.
39    ///
40    /// For example:
41    ///     let mock_admin_service = MockStateControlAdminService::new();
42    ///     let admin_service_child = realm_builder
43    ///         .add_local_child(
44    ///             "admin_service",
45    ///             move |handles| {
46    ///                 Box::pin(mock_admin_service.clone().run(handles))
47    ///             },
48    ///             ChildOptions::new(),
49    ///         )
50    ///         .await
51    ///         .unwrap();
52    ///
53    pub async fn run(self: Arc<Self>, handles: LocalComponentHandles) -> Result<(), anyhow::Error> {
54        self.run_inner(handles.outgoing_dir).await
55    }
56
57    async fn run_inner(
58        self: Arc<Self>,
59        outgoing_dir: ServerEnd<DirectoryMarker>,
60    ) -> Result<(), anyhow::Error> {
61        let mut fs = ServiceFs::new();
62        fs.dir("svc").add_fidl_service(move |mut stream: fpower::AdminRequestStream| {
63            let this = self.clone();
64            fasync::Task::local(async move {
65                info!("MockStateControlAdminService: new connection Admin");
66                while let Some(fpower::AdminRequest::PerformReboot { responder: _, options }) =
67                    stream.try_next().await.unwrap()
68                {
69                    info!("MockStateControlAdminService: received Reboot request");
70                    this.shutdown_received_sender
71                        .lock()
72                        .await
73                        .try_send(options)
74                        .expect("Failed to notify shutdown");
75                }
76            })
77            .detach();
78        });
79
80        fs.serve_connection(outgoing_dir).unwrap();
81        fs.collect::<()>().await;
82
83        Ok(())
84    }
85
86    /// Waits for the mock to receive a fidl.fuchsia.SystemController/Shutdown request.
87    pub async fn wait_for_shutdown_request(&self) -> RebootOptions {
88        self.shutdown_received_receiver
89            .lock()
90            .await
91            .next()
92            .await
93            .expect("Failed to wait for shutdown request")
94    }
95}
96
97#[cfg(test)]
98mod tests {
99    use super::*;
100    use fuchsia_component::client::connect_to_protocol_at_dir_svc;
101
102    #[fuchsia::test]
103    async fn test_shutdown() {
104        // Create and serve the mock service
105        let (dir, outgoing_dir) = fidl::endpoints::create_proxy::<DirectoryMarker>();
106        let mock = MockStateControlAdminService::new();
107        let _task = fasync::Task::local(mock.clone().run_inner(outgoing_dir));
108
109        // Connect to the mock server
110        let controller_client =
111            connect_to_protocol_at_dir_svc::<fpower::AdminMarker>(&dir).unwrap();
112
113        // Call the server's `shutdown` method and verify the mock sees the request
114        let _task = fuchsia_async::Task::local(controller_client.perform_reboot(&RebootOptions {
115            reasons: Some(vec![fpower::RebootReason2::HighTemperature]),
116            ..Default::default()
117        }));
118        mock.wait_for_shutdown_request().await;
119    }
120}
power_manager_integration_test_lib/mocks/admin.rs

power_manager_integration_test_lib/mocks/
admin.rs
