Add tests.

Author: wsx-antithesis

lib/tests/assert_guidance.rs

@@ -0,0 +1,58 @@
+use antithesis_sdk::{antithesis_init, assert_always_greater_than, LOCAL_OUTPUT};
+use serde_json::json;
+
+mod common;
+use common::SDKInput;
+
+use crate::common::{AntithesisGuidance, GuidanceType};
+
+#[test]
+fn assert_guidance() {
+    let output_file = "/tmp/antithesis-assert-guidance.json";
+    let prev_v = common::env::set_var(LOCAL_OUTPUT, output_file);
+    antithesis_init();
+
+    for i in 0..10 {
+        let x = if i % 2 == 0 { i } else { -i };
+        assert_always_greater_than!(x, 0, "Positive x", &json!({"x": x}));
+    }
+
+    match common::read_jsonl_tags(output_file) {
+        Ok(x) => {
+            let mut did_register = false;
+            let mut did_hit = false;
+            for obj in x.iter() {
+                if let SDKInput::AntithesisGuidance(AntithesisGuidance {
+                    guidance_type,
+                    hit,
+                    id,
+                    message,
+                    location,
+                    ..
+                }) = obj
+                {
+                    if *hit {
+                        did_hit = true;
+                    } else {
+                        did_register = true;
+                    };
+                    assert_eq!(*guidance_type, GuidanceType::Numeric);
+                    assert_eq!(message, "Positive x");
+                    assert_eq!(id, message);
+                    assert!(location.begin_line > 0);
+                    assert!(location.begin_column >= 0);
+                    assert_eq!(location.class, "assert_guidance");
+                    assert!(location.function.ends_with("::assert_guidance"));
+                    assert!(location
+                        .file
+                        .ends_with("/tests/assert_guidance.rs"));
+                }
+                println!("{:?}", obj);
+            }
+            assert!(did_register);
+            assert!(did_hit);
+        }
+        Err(e) => println!("{}", e),
+    }
+    common::env::restore_var(LOCAL_OUTPUT, prev_v);
+}

lib/tests/common/mod.rs

@@ -47,11 +47,23 @@ pub struct AntithesisAssert {
     pub details: Value,
 }
 
+#[derive(Deserialize, Debug)]
+pub struct AntithesisGuidance {
+    pub guidance_type: GuidanceType,
+    pub message: String,
+    pub id: String,
+    pub location: Location,
+    pub maximize: bool,
+    pub guidance_data: Value,
+    pub hit: bool,
+}
+
 #[derive(Deserialize, Debug)]
 #[serde(rename_all = "snake_case")]
 pub enum SDKInput {
     AntithesisSdk(AntithesisSdk),
     AntithesisAssert(AntithesisAssert),
+    AntithesisGuidance(AntithesisGuidance),
     AntithesisSetup(AntithesisSetup),
     SendEvent { event_name: String, details: Value },
 }
@@ -64,6 +76,14 @@ pub enum AssertType {
     Reachability,
 }
 
+#[derive(Deserialize, Debug, PartialEq)]
+#[serde(rename_all = "snake_case")]
+pub enum GuidanceType {
+    Numeric,
+    Boolean,
+    JSON,
+}
+
 fn parse_lines(lines: Vec<&str>) -> Result<Vec<SDKInput>, Box<dyn std::error::Error>> {
     let mut result = Vec::new();
 

simple/src/main.rs

@@ -79,6 +79,10 @@ fn assert_demo() {
     // ant_unreachable
     let details = json!({"impossible!": {"name": "trouble", "weights": [100,200,300]}});
     assert_unreachable!("Impossible to get here", &details);
+
+    assert_always_greater_than!(3, 100, "not right", &json!({}));
+
+    assert_sometimes_all!({a: true, b: false}, "not all right", &json!({}));
 }
 
 pub fn main() {