cm_stress_tests_lib/
lib.rs

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::{format_err, Context, Error};
6use fidl::endpoints::Proxy;
7use fidl_test_componentmanager_stresstests as fstresstests;
8use fuchsia_component_test::ScopedInstance;
9pub struct Child {
10    pub instance: ScopedInstance,
11    pub realm: fstresstests::ChildRealmProxy,
12}
13
14pub async fn create_child(collection: &str, url: &str) -> Result<Child, Error> {
15    let instance = ScopedInstance::new(collection.to_string(), url.to_string())
16        .await
17        .context(format_err!("Cannot create child for '{}:{}'", collection, url))?;
18    let realm =
19        instance.connect_to_protocol_at_exposed_dir::<fstresstests::ChildRealmMarker>().context(
20            format_err!("Cannot connect to child realm service for '{}'", instance.child_name()),
21        )?;
22    Ok(Child { instance, realm })
23}
24
25pub async fn stop_child(child: Child) -> Result<(), Error> {
26    child
27        .realm
28        .stop()
29        .context(format_err!("Error calling stop for '{}'", child.instance.child_name()))?;
30    child.realm.on_closed().await.context(format_err!(
31        "Error waiting for child to stop '{}'",
32        child.instance.child_name()
33    ))?;
34    Ok(())
35}