src/diagnostics/sampler/tests/test_topology.rs

// Copyright 2021 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 anyhow::*;
use fidl::endpoints::create_endpoints;
use fuchsia_component::client::connect_to_protocol;
use realm_client::{extend_namespace, InstalledNamespace};

pub(crate) const SAMPLER_NAME: &str = "sampler";
pub(crate) const COUNTER_NAME: &str = "single_counter";
pub(crate) const COBALT_NAME: &str = "cobalt";
pub(crate) const ARCHIVIST_NAME: &str = "test_archivist";

pub(crate) async fn create_realm() -> Result<InstalledNamespace, Error> {
    inner_create_realm(fidl_test_sampler::RealmOptions {
        sampler_component_name: Some(SAMPLER_NAME.into()),
        single_counter_name: Some(COUNTER_NAME.into()),
        mock_cobalt_name: Some(COBALT_NAME.into()),
        test_archivist_name: Some(ARCHIVIST_NAME.into()),
        ..Default::default()
    })
    .await
}

async fn inner_create_realm(
    options: fidl_test_sampler::RealmOptions,
) -> Result<InstalledNamespace, Error> {
    let realm_factory = connect_to_protocol::<fidl_test_sampler::RealmFactoryMarker>()?;
    let (dict_client, dict_server) = create_endpoints();
    realm_factory
        .create_realm(options, dict_server)
        .await?
        .map_err(realm_client::Error::OperationError)?;
    let ns = extend_namespace(realm_factory, dict_client).await?;
    Ok(ns)
}