Adjust Runtime::test function

src/main.rs

@@ -7,7 +7,7 @@ use std::rc::Rc;
 use std::time::Duration;
 
 use actix::prelude::*;
-use anyhow::Context;
+use anyhow::{anyhow, Context};
 use chrono::Utc;
 use clap::Parser;
 use futures::prelude::*;
@@ -243,7 +243,11 @@ async fn run<RUNTIME: process::Runtime + Clone + Unpin + 'static>(
             io::stdout().write_all(offer_template.as_bytes())?;
             return Ok(());
         }
-        Command::Test => return RUNTIME::test(&runtime_config),
+        Command::Test => {
+            return RUNTIME::test(runtime_config.clone())
+                .await
+                .map_err(|e| anyhow!("Testing runtime failed: {e}"))
+        }
     };
 
     let runtime_config = Box::pin(runtime_config);

src/process.rs

@@ -32,7 +32,7 @@ pub struct Usage {
 }
 
 #[async_trait]
-pub(crate) trait Runtime: Sized {
+pub(crate) trait Runtime: Sized + Send {
     type CONFIG: RuntimeConfig;
 
     fn parse_config(config: &Option<Value>) -> anyhow::Result<Self::CONFIG> {
@@ -48,10 +48,9 @@ pub(crate) trait Runtime: Sized {
 
     async fn wait(&mut self) -> std::io::Result<ExitStatus>;
 
-    fn test(config: &Self::CONFIG) -> anyhow::Result<()> {
-        gpu_detection(config).map_err(|err| {
-            anyhow::anyhow!("Testing runtime failed. Unable to detect GPU. Error: {err}")
-        })?;
+    async fn test(config: Self::CONFIG) -> anyhow::Result<()> {
+        gpu_detection(&config)
+            .map_err(|err| anyhow::anyhow!("Unable to detect GPU. Error: {err}"))?;
         Ok(())
     }
 
@@ -66,7 +65,7 @@ pub(crate) trait Runtime: Sized {
     }
 }
 
-pub(crate) trait RuntimeConfig: DeserializeOwned + Default + Debug + Clone {
+pub(crate) trait RuntimeConfig: DeserializeOwned + Default + Debug + Clone + Send {
     fn gpu_uuid(&self) -> Option<String>;
 }
 

src/process/automatic.rs

@@ -15,6 +15,7 @@ use tokio::{
     time::timeout,
 };
 
+use crate::offer_template::gpu_detection;
 use std::{
     path::PathBuf,
     process::{ExitStatus, Stdio},
@@ -60,6 +61,13 @@ impl Runtime for Automatic {
         })
     }
 
+    async fn test(config: Self::CONFIG) -> anyhow::Result<()> {
+        gpu_detection(&config)
+            .map_err(|err| anyhow::anyhow!("Unable to detect GPU. Error: {err}"))?;
+
+        Ok(())
+    }
+
     async fn stop(&mut self) -> anyhow::Result<()> {
         log::info!("Stopping Automatic server");
         let client = reqwest::Client::new();

src/process/dummy.rs

@@ -86,7 +86,7 @@ impl Runtime for Dummy {
         child.wait().await
     }
 
-    fn test(_config: &Self::CONFIG) -> anyhow::Result<()> {
+    async fn test(_config: Self::CONFIG) -> anyhow::Result<()> {
         Ok(())
     }