fidl_fuchsia_update_ext/
commit.rs

Help
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 anyhow::anyhow;
6use fidl_fuchsia_update::CommitStatusProviderProxy;
7use zx::{self as zx, AsHandleRef};
8
9/// Whether the current system version is pending commit.
10#[derive(Copy, Clone, Debug, PartialEq)]
11pub enum CommitStatus {
12    /// The current system version is pending commit.
13    Pending,
14    /// The current system version is already committed.
15    Committed,
16}
17
18/// Queries the commit status using `provider`.
19pub async fn query_commit_status(
20    provider: &CommitStatusProviderProxy,
21) -> Result<CommitStatus, anyhow::Error> {
22    let event_pair = provider.is_current_system_committed().await?;
23    match event_pair
24        .wait_handle(zx::Signals::USER_0, zx::MonotonicInstant::INFINITE_PAST)
25        .to_result()
26    {
27        Ok(_) => Ok(CommitStatus::Committed),
28        Err(zx::Status::TIMED_OUT) => Ok(CommitStatus::Pending),
29        Err(status) => Err(anyhow!("unexpected status while asserting signal: {:?}", status)),
30    }
31}
32
33#[cfg(test)]
34mod tests {
35    use super::*;
36    use fidl::endpoints::create_proxy_and_stream;
37    use fidl_fuchsia_update::{CommitStatusProviderMarker, CommitStatusProviderRequest};
38    use fuchsia_async::{self as fasync};
39    use futures::StreamExt;
40    use zx::{HandleBased, Peered};
41
42    // Verifies that query_commit_status returns the expected CommitStatus.
43    #[fasync::run_singlethreaded(test)]
44    async fn test_query_commit_status() {
45        let (proxy, mut stream) = create_proxy_and_stream::<CommitStatusProviderMarker>();
46        let (p0, p1) = zx::EventPair::create();
47
48        let _fidl_server = fasync::Task::local(async move {
49            while let Some(Ok(req)) = stream.next().await {
50                let CommitStatusProviderRequest::IsCurrentSystemCommitted { responder } = req;
51                let () = responder.send(p1.duplicate_handle(zx::Rights::BASIC).unwrap()).unwrap();
52            }
53        });
54
55        // When no signals are asserted, we should report Pending.
56        assert_eq!(query_commit_status(&proxy).await.unwrap(), CommitStatus::Pending);
57
58        // When USER_0 is asserted, we should report Committed.
59        let () = p0.signal_peer(zx::Signals::NONE, zx::Signals::USER_0).unwrap();
60        assert_eq!(query_commit_status(&proxy).await.unwrap(), CommitStatus::Committed,);
61    }
62}
fidl_fuchsia_update_ext/commit.rs

fidl_fuchsia_update_ext/
commit.rs
