From 954f971ddf059ae3871a75139b65c40b4ceaa959 Mon Sep 17 00:00:00 2001
From: Stephen Chung <Stephen.Chung@intexact.com>
Date: Sat, 13 Jun 2020 21:57:46 +0800
Subject: [PATCH] New optimizer test.

---
 tests/optimizer.rs | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/tests/optimizer.rs b/tests/optimizer.rs
index 49b15c25..828b0da4 100644
--- a/tests/optimizer.rs
+++ b/tests/optimizer.rs
@@ -3,7 +3,7 @@
 use rhai::{Engine, EvalAltResult, OptimizationLevel, INT};
 
 #[test]
-fn test_optimizer() -> Result<(), Box<EvalAltResult>> {
+fn test_optimizer_run() -> Result<(), Box<EvalAltResult>> {
     fn run_test(engine: &mut Engine) -> Result<(), Box<EvalAltResult>> {
         assert_eq!(engine.eval::<INT>(r"if true { 42 } else { 123 }")?, 42);
         assert_eq!(
@@ -30,3 +30,27 @@ fn test_optimizer() -> Result<(), Box<EvalAltResult>> {
 
     Ok(())
 }
+
+#[test]
+fn test_optimizer_parse() -> Result<(), Box<EvalAltResult>> {
+    let mut engine = Engine::new();
+    engine.set_optimization_level(OptimizationLevel::Simple);
+
+    let ast = engine.compile("{ const DECISION = false; if DECISION { 42 } }")?;
+
+    assert_eq!(
+        format!("{:?}", ast),
+        "AST([], <module vars={}, functions=0>)"
+    );
+
+    engine.set_optimization_level(OptimizationLevel::Full);
+
+    let ast = engine.compile("if 1 == 2 { 42 }")?;
+
+    assert_eq!(
+        format!("{:?}", ast),
+        "AST([], <module vars={}, functions=0>)"
+    );
+
+    Ok(())
+}