chore(tests): checks function return type

Author: JakubOnderka

src/compiler/test_util.rs

@@ -135,6 +135,10 @@ macro_rules! test_function {
                         assert!(got_value == want, "assertion failed for `{}` case:\n  got:    {:?}\n  wanted: {:?}", stringify!($case), got_value, want);
                         let got_tdef = expression.type_def(&state);
                         assert_eq!(got_tdef, $tdef);
+
+                        if let Ok(got_value) = got_value {
+                            assert!($func.return_kind() & got_value.kind_id() > 0, "assertion failed for `{}` return type, got {}", stringify!($case), got_value.kind_str());
+                        }
                     }
                     err@Err(_) => {
                         // Allow tests against compiler errors.

src/value/value.rs

@@ -81,6 +81,22 @@ impl Value {
         }
     }
 
+    #[cfg(test)]
+    pub const fn kind_id(&self) -> u16 {
+        use crate::compiler::value::kind;
+        match self {
+            Self::Bytes(_) => kind::BYTES,
+            Self::Regex(_) => kind::REGEX,
+            Self::Timestamp(_) => kind::TIMESTAMP,
+            Self::Integer(_) => kind::INTEGER,
+            Self::Float(_) => kind::FLOAT,
+            Self::Boolean(_) => kind::BOOLEAN,
+            Self::Object(_) => kind::OBJECT,
+            Self::Array(_) => kind::ARRAY,
+            Self::Null => kind::NULL,
+        }
+    }
+
     /// Merges `incoming` value into self.
     ///
     /// Will concatenate `Bytes` and overwrite the rest value kinds.