Add Chaos support to CLI

crates/parser/src/lib.rs

@@ -666,6 +666,12 @@ pub fn parse(unparsed_file: &str) -> Result<ast::TestDefinition, errors::ParserE
                 assertions.push(assertion);
             }
             Rule::EOI | Rule::comment => (),
+            Rule::setter => {
+                println!("Setter!!");
+            }
+            Rule::remover => {
+                println!("Remover!!");
+            }
             _ => {
                 return Err(errors::ParserError::InvalidRule(record.as_str().to_owned()));
             }

crates/parser/src/zombienet.pest

@@ -27,6 +27,7 @@ span_id = { ASCII_HEX_DIGIT{32} }
 block_height = { "blockheight" | "block height" | "best block" }
 finalized_height = { "finalised height" | "finalised block" }
 peers_count = { "peers count" | "peers" }
+chaos_type = { "latency" | "jitter" | "correlation" }
 
 metric_name = @{ (block_height | finalized_height | peers_count | (ASCII_ALPHANUMERIC | "_" )+) ~ ("{" ~ metric_key_value ~ "}")? }
 square_brackets_strings = {
@@ -51,6 +52,7 @@ math_ops    = { plus | minus }
 // commons
 node_name = { name ~ colon }
 seconds = _{ "seconds"|"secs"|"s" }
+miliseconds = _{ "miliseconds"|"ms"|"msecs" }
 within = { "within" ~ int+ ~ seconds }
 parachain = { "parachain" ~ int+ }
 
@@ -83,6 +85,10 @@ pause = { node_name ~ "pause" }
 resume = { node_name ~ "resume" }
 restart = { node_name ~ "restart" ~ ("after" ~ int+ ~ seconds)? }
 
+// CHAOS COMMANDS
+setter = { node_name ~ "set" ~ chaos_type ~ "by" ~ int+ ~ (seconds | miliseconds)}
+remover = { node_name ~ "remove" ~ chaos_type }
+
 /// COMMENTS
 comment = ${ ("#" | "//") ~ (!NEWLINE ~ ANY)* }
 
@@ -111,5 +117,7 @@ file = { SOI ~ (
     sleep |
     pause |
     resume |
-    restart
+    restart | 
+    setter |
+    remover
     )* ~ NEWLINE* ~ EOI }

tests/chaos/0001-delay.zndsl

@@ -6,4 +6,5 @@ Creds: config
 # alice to bob take at least 2s
 alice: run ./delay_check.sh with "bob,9615,2" within 30 seconds
 # bob to alice take at least 1s
-bob: run ./delay_check.sh within 30 seconds
+bob: run ./delay_check.sh within 30 seconds
+bob: remove latency

tests/chaos/delay_check.sh

@@ -3,7 +3,8 @@
 POD=$1
 PORT=$2 # 9615 for prometheus
 DELAY=$3
-TAKE=$(/cfg/curl  -w "%{time_connect}"  -so /dev/null  http://$POD:$PORT/ |awk -F "." '{print $1}')
+## For mac the `confi/curl` needs to be replaced with `curl` 
+TAKE=$(curl  -w "%{time_connect}"  -so /dev/null  http://$POD:$PORT/ |awk -F "." '{print $1}')
 if [[ $TAKE -ge $DELAY ]]; then
     exit 0;
 fi