if

tests/common.rs

@@ -0,0 +1,24 @@
+#[macro_export]
+macro_rules! run_test {
+    ($name:ident, $code:expr) => {
+        #[test]
+        fn $name() {
+            let code = $code;
+            let output = _run_test(code);
+            insta::assert_debug_snapshot!(output);
+        }
+    };
+}
+
+#[doc(hidden)]
+pub fn _run_test(code: &str) -> String {
+    let mut stdout = Vec::<u8>::new();
+    let mut cfg = dilaria::Config {
+        debug: false,
+        stdout: &mut stdout,
+    };
+
+    dilaria::run_program(code, &mut cfg);
+
+    String::from_utf8(stdout).unwrap()
+}

tests/control_flow.rs

@@ -0,0 +1,53 @@
+mod common;
+use crate::common::_run_test;
+
+run_test!(
+    single_if,
+    r#"
+if true {
+    print "true!";
+}
+
+if false {
+    print "WRONG";
+}
+"#
+);
+
+run_test!(
+    if_else,
+    r#"
+if true {
+    print "true!";
+} else {
+    print "WRONG";
+}
+
+if false {
+    print "WRONG";
+} else {
+    print "true!";
+}
+"#
+);
+
+run_test!(
+    if_else_if,
+    r#"
+if false {
+    print "WRONG";
+} else if true {
+    print "true!";
+} else {
+    print "WRONG";
+}
+
+if false {
+    print "WRONG";
+} else if false {
+    print "WRONG";
+} else {
+    print "true!";
+}
+"#
+);

tests/snapshots/control_flow__if_else.snap

@@ -0,0 +1,7 @@
+---
+source: tests/control_flow.rs
+assertion_line: 17
+expression: output
+
+---
+"true!\ntrue!\n"

tests/snapshots/control_flow__if_else_if.snap

@@ -0,0 +1,7 @@
+---
+source: tests/control_flow.rs
+assertion_line: 34
+expression: output
+
+---
+"true!\ntrue!\n"

tests/snapshots/control_flow__single_if.snap

@@ -0,0 +1,7 @@
+---
+source: tests/control_flow.rs
+assertion_line: 4
+expression: output
+
+---
+"true!\n"