src/sys/pkg/bin/omaha-client/src/cobalt.rs

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use fidl_fuchsia_cobalt::{SystemDataUpdaterMarker, SystemDataUpdaterProxy};
use fuchsia_component::client::connect_to_service;
use log::{error, info};
use omaha_client::common::AppSet;

pub async fn notify_cobalt_current_channel(app_set: AppSet) {
    info!("Notifying Cobalt about the current channel");
    let proxy = match connect_to_service::<SystemDataUpdaterMarker>() {
        Ok(proxy) => proxy,
        Err(e) => {
            error!("Failed to connect to cobalt: {}", e);
            return;
        }
    };
    notify_cobalt_current_channel_impl(proxy, app_set).await;
}

async fn notify_cobalt_current_channel_impl(proxy: SystemDataUpdaterProxy, app_set: AppSet) {
    let channel = app_set.get_current_channel().await;

    match proxy.set_channel(&channel).await {
        Ok(fidl_fuchsia_cobalt::Status::Ok) => {}
        error => {
            error!("SystemDataUpdater.SetChannel failed: {:?}", error);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use fidl::endpoints::create_proxy_and_stream;
    use fidl_fuchsia_cobalt::SystemDataUpdaterRequest;
    use fuchsia_async as fasync;
    use futures::prelude::*;
    use omaha_client::{common::App, protocol::Cohort};

    #[fasync::run_singlethreaded(test)]
    async fn test_notify_cobalt() {
        let app_set = AppSet::new(vec![App::new(
            "id",
            [1, 2],
            Cohort { name: Some("current-channel".to_string()), ..Cohort::default() },
        )]);

        let (proxy, mut stream) = create_proxy_and_stream::<SystemDataUpdaterMarker>().unwrap();
        let stream_fut = async move {
            match stream.next().await {
                Some(Ok(SystemDataUpdaterRequest::SetChannel { current_channel, responder })) => {
                    assert_eq!(current_channel, "current-channel");
                    responder.send(fidl_fuchsia_cobalt::Status::Ok).unwrap();
                }
                err => panic!("Err in request handler: {:?}", err),
            }
            assert!(stream.next().await.is_none());
        };
        future::join(notify_cobalt_current_channel_impl(proxy, app_set), stream_fut).await;
    }
}