diff --git a/Cargo.lock b/Cargo.lock index a3b733486..e4edde420 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -387,6 +387,7 @@ dependencies = [ "num-bigint 0.3.2", "num-traits", "rustpython-parser", + "test-case", ] [[package]] @@ -825,6 +826,19 @@ dependencies = [ "winapi", ] +[[package]] +name = "test-case" +version = "1.2.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b114ece25254e97bf48dd4bfc2a12bad0647adacfe4cae1247a9ca6ad302cec" +dependencies = [ + "cfg-if", + "proc-macro2", + "quote", + "syn", + "version_check", +] + [[package]] name = "tiny-keccak" version = "2.0.2" diff --git a/nac3core/Cargo.toml b/nac3core/Cargo.toml index af84b2552..412e12347 100644 --- a/nac3core/Cargo.toml +++ b/nac3core/Cargo.toml @@ -11,3 +11,6 @@ inkwell = { git = "https://github.com/TheDan64/inkwell", branch = "master", feat rustpython-parser = { git = "https://github.com/RustPython/RustPython", branch = "master" } indoc = "1.0" +[dev-dependencies] +test-case = "1.2.0" +indoc = "1.0" \ No newline at end of file diff --git a/nac3core/src/lib.rs b/nac3core/src/lib.rs index ff64158d7..0057c2e60 100644 --- a/nac3core/src/lib.rs +++ b/nac3core/src/lib.rs @@ -1,6 +1,9 @@ #![warn(clippy::all)] #![allow(clippy::clone_double_ref)] +#[cfg(test)] +extern crate test_case; + extern crate num_bigint; extern crate inkwell; extern crate rustpython_parser; diff --git a/nac3core/src/typecheck/expression_inference.rs b/nac3core/src/typecheck/expression_inference.rs index 3fcbf10e4..7cb2027b6 100644 --- a/nac3core/src/typecheck/expression_inference.rs +++ b/nac3core/src/typecheck/expression_inference.rs @@ -22,14 +22,14 @@ impl<'a> ast::fold::Fold<()> for TypeInferencer<'a> { fn fold_expr(&mut self, node: ast::Expr<()>) -> Result, Self::Error> { - self.error_stack.push((node.node.name().into(), node.location)); + self.error_stack.push(("Checking ".to_string() + node.node.name(), node.location)); let expr = match &node.node { ast::ExprKind::ListComp { .. } => return self.fold_listcomp(node), _ => rustpython_parser::ast::fold::fold_expr(self, node)? }; - Ok(ast::Expr { + let ret = Ok(ast::Expr { // compute type info and store in the custom field custom: match &expr.node { ast::ExprKind::Constant {value, kind: _} => self.infer_constant(value), @@ -50,7 +50,9 @@ impl<'a> ast::fold::Fold<()> for TypeInferencer<'a> { }?, location: expr.location, node: expr.node - }) + }); + self.error_stack.pop(); + ret } } @@ -320,7 +322,7 @@ impl<'a> TypeInferencer<'a> { // some pre-folds need special handling fn fold_listcomp(&mut self, expr: ast::Expr<()>) -> Result>, String> { - self.error_stack.push(("folding list comprehension at ".into(), expr.location)); + self.error_stack.push(("list comprehension at ".into(), expr.location)); if let ast::Expr { location, @@ -343,7 +345,7 @@ impl<'a> TypeInferencer<'a> { is_async} = gen; let iter_folded = Box::new(self.fold_expr(*iter)?); - if let TypeEnum::ParametricType( + let ret = if let TypeEnum::ParametricType( primitives::LIST_TYPE, ls) = iter_folded .custom @@ -390,14 +392,17 @@ impl<'a> TypeInferencer<'a> { result } else { Err("iteration is supported for list only".into()) - } + }; + self.error_stack.pop(); + ret } else { panic!("this function is for list comprehensions only!"); } } fn infer_simple_binding(&mut self, name: &ast::Expr, ty: Type) -> Result<(), String> { - match &name.node { + self.error_stack.push(("resolving list comprehension variables".into(), name.location)); + let ret = match &name.node { ast::ExprKind::Name {id, ctx: _} => { if id == "_" { Ok(()) @@ -424,7 +429,9 @@ impl<'a> TypeInferencer<'a> { } } _ => Err("not supported".into()) - } + }; + self.error_stack.pop(); + ret } fn fold_expr(&mut self, node: ast::Expr<()>) -> Result>, String> { @@ -437,10 +444,14 @@ impl<'a> TypeInferencer<'a> { } } + pub mod test { + use std::vec; + use crate::typecheck::{symbol_resolver::SymbolResolver, symbol_resolver::*, location::*}; use rustpython_parser::ast::Expr; + use test_case::test_case; // FIXME use super::*; pub fn new_ctx<'a>() -> TypeInferencer<'a> { @@ -626,7 +637,7 @@ pub mod test { ); } - #[test] + #[test] fn test_mix() { let mut inf = new_ctx(); let ast1 = rustpython_parser::parser::parse_expression("False == [True or True, False][0]").unwrap(); @@ -642,7 +653,7 @@ pub mod test { let ast11 = rustpython_parser::parser::parse_expression("(1, 2, 3, 4)[1]").unwrap(); let ast12 = rustpython_parser::parser::parse_expression("(1, True, 3, False)[1]").unwrap(); - let ast13 = rustpython_parser::parser::parse_expression("[1, True, 2]").unwrap(); + // let ast13 = rustpython_parser::parser::parse_expression("[1, True, 2]").unwrap(); let folded = inf.fold_expr(ast1).unwrap(); let folded_2 = inf.fold_expr(ast2).unwrap(); @@ -670,6 +681,22 @@ pub mod test { println!("{:?}", folded_10.custom); println!("{:?}", folded_11.custom); println!("{:?}", folded_12.custom); - let folded_13 = inf.fold_expr(ast13); + // let folded_13 = inf.fold_expr(ast13); + } + + #[test] + fn test_err_msg() { + let mut inf = new_ctx(); + for prog in vec![ + "[1, True, 2]", + "True if 1 else False", + "1 if True else False", + + ] { + let ast = rustpython_parser::parser::parse_expression(prog).unwrap(); + let _folded = inf.fold_expr(ast); + println!(""); + } + // println!("{:?}", folded); } } \ No newline at end of file