bt_test_harness/
bootstrap.rs

1// Copyright 2021 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::{Context, Error};
6use fidl_fuchsia_bluetooth_sys::{BootstrapMarker, BootstrapProxy};
7use fuchsia_bluetooth::expectation::asynchronous::{expectable, Expectable};
8use futures::future::{self, BoxFuture, FutureExt};
9use std::ops::{Deref, DerefMut};
10use std::sync::Arc;
11use test_harness::{SharedState, TestHarness, SHARED_STATE_TEST_COMPONENT_INDEX};
12
13use crate::core_realm::{CoreRealm, SHARED_STATE_INDEX};
14use crate::host_watcher::ActivatedFakeHost;
15
16#[derive(Clone)]
17pub struct BootstrapHarness(Expectable<(), BootstrapProxy>);
18
19impl Deref for BootstrapHarness {
20    type Target = Expectable<(), BootstrapProxy>;
21
22    fn deref(&self) -> &Self::Target {
23        &self.0
24    }
25}
26
27impl DerefMut for BootstrapHarness {
28    fn deref_mut(&mut self) -> &mut Self::Target {
29        &mut self.0
30    }
31}
32
33impl TestHarness for BootstrapHarness {
34    type Env = (ActivatedFakeHost, Arc<CoreRealm>);
35    type Runner = future::Pending<Result<(), Error>>;
36
37    fn init(
38        shared_state: &Arc<SharedState>,
39    ) -> BoxFuture<'static, Result<(Self, Self::Env, Self::Runner), Error>> {
40        let shared_state = shared_state.clone();
41        async move {
42            let test_component: Arc<String> = shared_state
43                .get(SHARED_STATE_TEST_COMPONENT_INDEX)
44                .expect("SharedState must have TEST-COMPONENT")?;
45            let inserter = move || CoreRealm::create(test_component.to_string());
46            let realm = shared_state.get_or_insert_with(SHARED_STATE_INDEX, inserter).await?;
47            let fake_host = ActivatedFakeHost::new(realm.clone()).await?;
48            let proxy = realm
49                .instance()
50                .connect_to_protocol_at_exposed_dir::<BootstrapMarker>()
51                .context("Failed to connect to bootstrap service")?;
52            Ok((
53                BootstrapHarness(expectable(Default::default(), proxy)),
54                (fake_host, realm),
55                future::pending(),
56            ))
57        }
58        .boxed()
59    }
60
61    fn terminate((emulator, realm): Self::Env) -> BoxFuture<'static, Result<(), Error>> {
62        // The realm must be kept alive in order for emulator.release() to work properly.
63        async move {
64            let _realm = realm;
65            emulator.release().await
66        }
67        .boxed()
68    }
69}
bt_test_harness/bootstrap.rs

bt_test_harness/
bootstrap.rs
