mock_health_verification/
lib.rs

1// Copyright 2025 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_fuchsia_update_verify as fidl;
6use fuchsia_async::Task;
7use futures::{future, FutureExt as _, StreamExt as _};
8use std::sync::Arc;
9
10pub trait Hook: Send + Sync {
11    fn query_health_checks(&self) -> future::BoxFuture<'static, zx::Status>;
12}
13
14impl<F> Hook for F
15where
16    F: Fn() -> zx::Status + Send + Sync,
17{
18    fn query_health_checks(&self) -> future::BoxFuture<'static, zx::Status> {
19        future::ready(self()).boxed()
20    }
21}
22
23pub struct MockHealthVerificationService {
24    call_hook: Box<dyn Hook>,
25}
26
27impl MockHealthVerificationService {
28    /// Creates a new MockHealthVerificationService with a given callback to run per call to the service.
29    pub fn new(hook: impl Hook + 'static) -> Self {
30        Self { call_hook: Box::new(hook) }
31    }
32
33    pub fn spawn_health_verification_service(
34        self: Arc<Self>,
35    ) -> (fidl::HealthVerificationProxy, Task<()>) {
36        let (proxy, stream) =
37            ::fidl::endpoints::create_proxy_and_stream::<fidl::HealthVerificationMarker>();
38
39        let task = Task::spawn(self.run_health_verification_service(stream));
40
41        (proxy, task)
42    }
43
44    /// Serves fuchsia.update.verify/HealthVerification.QueryHealthChecks
45    pub async fn run_health_verification_service(
46        self: Arc<Self>,
47        stream: fidl::HealthVerificationRequestStream,
48    ) {
49        let Self { call_hook } = &*self;
50        stream
51            .for_each(|request| match request.expect("received verifier request") {
52                fidl::HealthVerificationRequest::QueryHealthChecks { responder } => call_hook
53                    .query_health_checks()
54                    .map(|res| responder.send(res.into_raw()).expect("sent verifier response")),
55            })
56            .await
57    }
58}
59
60#[cfg(test)]
61mod tests {
62    use super::*;
63    use fuchsia_async as fasync;
64
65    #[fasync::run_singlethreaded(test)]
66    async fn test_mock_verifier() {
67        let mock = Arc::new(MockHealthVerificationService::new(|| zx::Status::OK));
68        let (proxy, _server) = mock.spawn_health_verification_service();
69
70        let verify_result = proxy.query_health_checks().await.expect("made fidl call");
71
72        assert_eq!(verify_result, 0);
73    }
74}
mock_health_verification/lib.rs

mock_health_verification/
lib.rs
