run_test_suite_lib/
connector.rs

1// Copyright 2022 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 crate::outcome::ConnectionError;
6use fidl_fuchsia_test_manager::RunBuilderProxy;
7use futures::lock::Mutex;
8
9/// Implementing this trait allows configuring the number of suites to
10/// run on a single RunBuilder connection.
11/// This alleviates an issue where for n suites run on a single RunBuilder
12/// connection, n channels must be opened up front. This can cause some
13/// issues with resource limitations when a large number of tests is
14/// specified (see https://fxbug.dev/42062444).
15#[async_trait::async_trait]
16pub trait RunBuilderConnector {
17    /// Create a new connection to RunBuilder.
18    async fn connect(&self) -> Result<RunBuilderProxy, ConnectionError>;
19    /// Number of suites for which a connection produced by this connector
20    /// should be used.
21    fn batch_size(&self) -> usize;
22}
23
24/// A connector that produces a single proxy and instructs all suites to
25/// be executed using it.
26pub struct SingleRunConnector {
27    proxy: Mutex<Option<RunBuilderProxy>>,
28}
29
30impl SingleRunConnector {
31    pub fn new(proxy: RunBuilderProxy) -> Self {
32        Self { proxy: Mutex::new(Some(proxy)) }
33    }
34}
35
36#[async_trait::async_trait]
37impl RunBuilderConnector for SingleRunConnector {
38    async fn connect(&self) -> Result<RunBuilderProxy, ConnectionError> {
39        Ok(self.proxy.lock().await.take().expect("connect only called once"))
40    }
41
42    fn batch_size(&self) -> usize {
43        usize::MAX
44    }
45}
run_test_suite_lib/connector.rs

run_test_suite_lib/
connector.rs
