power_manager_integration_test_lib/client_connectors/
thermal_client.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 crate::TestEnv;
6use fidl_fuchsia_thermal as fthermal;
7
8/// Convenience type for interacting with the Power Manager's thermal client service.
9pub struct ThermalClient {
10    watcher_proxy: fthermal::ClientStateWatcherProxy,
11}
12
13impl ThermalClient {
14    pub fn new(test_env: &TestEnv, client_type: &str) -> Self {
15        let connector = test_env.connect_to_protocol::<fthermal::ClientStateConnectorMarker>();
16        let (watcher_proxy, watcher_remote) =
17            fidl::endpoints::create_proxy::<fthermal::ClientStateWatcherMarker>();
18        connector.connect(client_type, watcher_remote).expect("Failed to connect thermal client");
19        Self { watcher_proxy }
20    }
21
22    pub async fn get_thermal_state(&self) -> Result<u64, anyhow::Error> {
23        self.watcher_proxy.watch().await.map_err(|e| e.into())
24    }
25}