Merge pull request #20 from trento-project/add-exit-codes

Add exit codes to the tool

Cargo.toml

@@ -9,3 +9,4 @@ clap = {version = "3.0.7", features = ["cargo"] }
 serde = { version = "1.0", features = ["derive"] }
 reqwest = { version = "0.11", features = ["json"] }
 tokio = { version = "1", features = ["full"] }
+exitcode = "1.1.2"

src/main.rs

@@ -1,4 +1,5 @@
 extern crate clap;
+extern crate exitcode;
 
 use clap::{App, Arg};
 
@@ -40,12 +41,19 @@ async fn main() {
 
     if options.subcommand_matches("list").is_some() {
         list::show_list(&scenarios);
+        std::process::exit(exitcode::OK)
     }
 
     if let Some(run_options) = options.subcommand_matches("run") {
         let scenario_label = run_options.value_of("SET").unwrap();
         let endpoint_url = run_options.value_of("url").unwrap();
         let api_key = run_options.value_of("API_KEY").unwrap();
-        run::run(endpoint_url, api_key, scenario_label.to_string(), scenarios).await;
+        match run::run(endpoint_url, api_key, scenario_label.to_string(), scenarios).await {
+            Ok(()) => std::process::exit(exitcode::OK),
+            Err(()) => std::process::exit(1)
+        }
     }
+
+    println!("Subcommand not provided. Available subcommands: list|run");
+    std::process::exit(exitcode::USAGE)
 }

src/run.rs

@@ -7,6 +7,7 @@ enum FixtureResult {
     Success,
     Retryable { file: String },
     Skippable,
+    Unauthorized,
 }
 
 struct Errored {
@@ -38,6 +39,10 @@ async fn post_fixture(
                         println!("Successfully POSTed file: {}", file);
                         Ok(FixtureResult::Success)
                     }
+                    StatusCode::UNAUTHORIZED => {
+                        println!("POST request unauthorized. Set the correct API_KEY as argument");
+                        Ok(FixtureResult::Unauthorized)
+                    }
                     StatusCode::BAD_REQUEST
                     | StatusCode::UNPROCESSABLE_ENTITY
                     | StatusCode::NOT_FOUND => Ok(FixtureResult::Retryable {
@@ -87,13 +92,16 @@ pub async fn run(
     api_key: &str,
     scenario_label: String,
     scenarios: Vec<Scenario>,
-) {
+) -> Result<(), ()> {
     let selected_scenario = scenarios
         .iter()
         .find(|current_scenario| current_scenario.label == scenario_label);
 
     match selected_scenario {
-        None => println!("Non-existing scenario!"),
+        None => {
+            println!("Non-existing scenario!");
+            return Err(())
+        },
         Some(scenario) => {
             let fixtures_in_directories: Vec<String> = scenario
                 .directories
@@ -113,6 +121,9 @@ pub async fn run(
                         retryable.push(FixtureResult::Retryable { file })
                     }
                     Ok(FixtureResult::Skippable | FixtureResult::Success) => (),
+                    Ok(FixtureResult::Unauthorized) => {
+                        return Err(())
+                    },
                     Err(Errored { file, reason }) => {
                         println!("An error occurred in loading fixture {}: {}", file, reason)
                     }
@@ -127,6 +138,8 @@ pub async fn run(
             }
         }
     }
+
+    Ok(())
 }
 
 fn extract_fixtures_from_directory(directory: &String) -> Option<Vec<String>> {