add color to output

Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+colored = "2.0.0"
 reqwest = "0.11.14"
 serde_json = "1.0.91"
 tokio = { version = "1.25.0", features = ["full"] }

src/interpreter.rs

@@ -121,6 +121,38 @@ pub async fn interpreter( certus_file: &str ){
                     return
                 }
             }
+            // println!("Here");
+            //Retrieve expected HTTP status
+            let mut read_status = false;
+            for line in content.lines(){
+                if line.trim()=="[EXPECT]" {
+                    read_status = true;
+                    continue;
+                }
+                if line.trim().starts_with("#") {
+                    continue;
+                }
+                if line.trim()==""{
+                    continue;
+                }
+                if line.trim().starts_with("HTTP") & read_status{
+                    let splitted : Vec<&str> = line.split(" ").collect();
+                    if splitted.len()==2{
+                        match u16::from_str(splitted[1])  {
+                            Ok(status)=>{
+                                certus_test.expected_status = status;
+                                break;
+                            },
+                            Err(_)=>{
+                                println!("Invalid HTTP status code {}", splitted[1]);
+                                return ;
+                            }
+                        }
+                    }
+                    println!("Failed to parse expected HTTP status code");
+                    return;
+                }
+            }
             run(certus_test).await;
         },
         Err(_)=>{

src/main.rs

@@ -1,3 +1,4 @@
+extern crate colored;
 mod interpreter;
 mod runner;
 mod utils;
@@ -6,6 +7,7 @@ use std::path::Path;
 use tokio;
 use utils::write_template;
 use interpreter::interpreter;
+use colored::*;
 
 #[tokio::main]
 async fn main() -> () {

src/runner.rs

@@ -1,5 +1,6 @@
 use crate::utils::CertusTest;
 use reqwest::Client;
+use colored::*;
 
 pub async fn run( test: CertusTest ){
     let reqwest_client = Client::new();
@@ -11,7 +12,13 @@ pub async fn run( test: CertusTest ){
     let response = certus_request.send().await;
     match response {
         Ok(response)=>{
-            let response_status = response.status();
+            let response_status = response.status().as_u16();
+            if response_status!=test.expected_status {
+                let error_message = std::format!("Test failed: Got HTTP {} but expected HTTP {}", response_status, test.expected_status).color("red");                
+                println!("{}", error_message);
+                return
+            }
+
             let response_headers = response.headers();
             let response_text = &response.text().await;
             match response_text {

src/utils.rs

@@ -23,12 +23,23 @@ pub struct CertusTest{
     pub method: Method,
     pub endpoint: String,
     pub headers: HeaderMap,
-    pub body: Value
+    pub body: Value,
+    pub expected_status: u16,
+    pub expected_headers: HeaderMap,
+    pub expected_body: Value,
 
 }
 
 impl Default for CertusTest {
     fn default() -> Self {
-        CertusTest { method: Method::from_str("GET").unwrap(), endpoint: "".to_string(), headers: HeaderMap::new(), body: Value::default() }
+        CertusTest { 
+            method: Method::from_str("GET").unwrap(), 
+            endpoint: "".to_string(), 
+            headers: HeaderMap::new(), 
+            body: Value::default(), 
+            expected_status:200,
+            expected_headers: HeaderMap::new(),
+            expected_body: Value::default() 
+        }
     }
 }