Merge pull request #3281 from golemfactory/provider/add-self-test-com…

…mand Add self-test command to provider cli
golemfactory · Jul 30, 2024 · 8483435 · 8483435
2 parents 83d4600 + 48e25db
commit 8483435
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/agent/provider/src/main.rs b/agent/provider/src/main.rs
@@ -13,7 +13,8 @@ async fn main() -> anyhow::Result<()> {
 
     let cli_args = StartupConfig::from_args();
     match &cli_args.commands {
-        Commands::Run(_) => (), // logging is handled by ProviderAgent
+        Commands::Run(_) => (),      // logging is handled by ProviderAgent
+        Commands::SelfTest(_) => (), // logging is handled by ProviderAgent
         _ => {
             ya_file_logging::start_logger("info", None, &[], false)?;
         }
@@ -42,6 +43,12 @@ async fn main() -> anyhow::Result<()> {
 
             Ok(())
         }
+        Commands::SelfTest(args) => {
+            let agent = ProviderAgent::new(args, config).await?.start();
+            agent.send(Shutdown).await??;
+            log::info!("Self-test finished. Exiting...");
+            Ok(())
+        }
         Commands::Config(config_cmd) => config_cmd.run(config),
         Commands::Preset(presets_cmd) => presets_cmd.run(config),
         Commands::PreInstall(preinstall_cmd) => preinstall_cmd.run(config),

diff --git a/agent/provider/src/startup_config.rs b/agent/provider/src/startup_config.rs
@@ -290,6 +290,8 @@ pub struct StartupConfig {
 pub enum Commands {
     /// Run provider agent
     Run(RunConfig),
+    /// Dry run provider agent to check configuration and test runtimes.
+    SelfTest(RunConfig),
     /// Configure provider agent
     Config(ConfigConfig),
     /// Manage offer presets