src/sys/run_test_suite/src/artifact.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 futures::{channel::mpsc, SinkExt};

pub enum Artifact {
    SuiteStdoutMessage(String),
    SuiteStderrMessage(String),
    SuiteLogMessage(String),
}

impl Artifact {
    fn suite_stdout_message<S: Into<String>>(s: S) -> Self {
        Self::SuiteStdoutMessage(s.into())
    }

    fn suite_stderr_message<S: Into<String>>(s: S) -> Self {
        Self::SuiteStderrMessage(s.into())
    }

    fn suite_log_message<S: Into<String>>(s: S) -> Self {
        Self::SuiteLogMessage(s.into())
    }
}

#[derive(Clone)]
pub struct ArtifactSender(mpsc::Sender<Artifact>);

impl ArtifactSender {
    pub async fn send_test_stdout_msg<S: Into<String>>(
        &mut self,
        s: S,
    ) -> Result<(), mpsc::SendError> {
        self.0.send(Artifact::suite_stdout_message(s)).await
    }

    pub async fn send_test_stderr_msg<S: Into<String>>(
        &mut self,
        s: S,
    ) -> Result<(), mpsc::SendError> {
        self.0.send(Artifact::suite_stderr_message(s)).await
    }

    pub async fn send_suite_log_msg<S: Into<String>>(
        &mut self,
        s: S,
    ) -> Result<(), mpsc::SendError> {
        self.0.send(Artifact::suite_log_message(s)).await
    }
}

impl From<mpsc::Sender<Artifact>> for ArtifactSender {
    fn from(s: mpsc::Sender<Artifact>) -> Self {
        Self(s)
    }
}