From 4eacd1aa9e5d3796c5346a22d76b7b1828adbf6b Mon Sep 17 00:00:00 2001
From: ychenfo <yc@m-labs.hk>
Date: Fri, 10 Sep 2021 21:26:39 +0800
Subject: [PATCH] nac3core: top level err test

---
 nac3core/src/toplevel/test.rs | 121 ++++++++++++++++++++++++++--------
 1 file changed, 93 insertions(+), 28 deletions(-)

diff --git a/nac3core/src/toplevel/test.rs b/nac3core/src/toplevel/test.rs
index f8c6cbde0..ed9b61edd 100644
--- a/nac3core/src/toplevel/test.rs
+++ b/nac3core/src/toplevel/test.rs
@@ -414,7 +414,47 @@ fn test_simple_function_analyze(source: Vec<&str>, tys: Vec<&str>, names: Vec<&s
     ];
     "list tuple generic"
 )]
+#[test_case(
+    vec![indoc! {"
+        class A:
+            pass
+    "}],
+    vec!["class def must have __init__ method defined"];
+    "err no __init__"
+)]
+#[test_case(
+    vec![indoc! {"
+        class A:
+            def __init__():
+                pass
+    "}],
+    vec!["__init__ function must have a `self` parameter"];
+    "err no self_1"
+)]
+#[test_case(
+    vec![
+        indoc! {"
+            class A(B, Generic[T], C):
+                def __init__(self):
+                    pass
+        "},
+        indoc! {"
+            class B:
+                def __init__(self):
+                    pass
+        "},
+        indoc! {"
+            class C:
+                def __init__(self):
+                    pass
+        "}
+
+    ],
+    vec!["a class def can only have at most one base class declaration and one generic declaration"];
+    "err multiple inheritance"
+)]
 fn test_simple_class_analyze(source: Vec<&str>, res: Vec<&str>) {
+    let print = false;
     let mut composer = TopLevelComposer::new();
 
     let tvar_t = composer.unifier.get_fresh_var();
@@ -422,8 +462,10 @@ fn test_simple_class_analyze(source: Vec<&str>, res: Vec<&str>) {
         .unifier
         .get_fresh_var_with_range(&[composer.primitives_ty.bool, composer.primitives_ty.int32]);
 
-    println!("t: {}, {:?}", tvar_t.1, tvar_t.0);
-    println!("v: {}, {:?}\n", tvar_v.1, tvar_v.0);
+    if print {
+        println!("t: {}, {:?}", tvar_t.1, tvar_t.0);
+        println!("v: {}, {:?}\n", tvar_v.1, tvar_v.0);
+    }
 
     let internal_resolver = Arc::new(ResolverInternal {
         id_to_def: Default::default(),
@@ -440,35 +482,58 @@ fn test_simple_class_analyze(source: Vec<&str>, res: Vec<&str>) {
         let ast = parse_program(s).unwrap();
         let ast = ast[0].clone();
 
-        let (id, def_id) = composer.register_top_level(ast, Some(resolver.clone())).unwrap();
+        let (id, def_id) = {
+            match composer.register_top_level(ast, Some(resolver.clone())) {
+                Ok(x) => x,
+                Err(msg) => {
+                    if print {
+                        println!("{}", msg);
+                    } else {
+                        assert_eq!(res[0], msg);
+                    }
+                    return
+                }
+            }
+        };
         internal_resolver.add_id_def(id, def_id);
     }
 
-    composer.start_analysis().unwrap();
-
-    // skip 5 to skip primitives
-    for (i, (def, _)) in composer.definition_ast_list.iter().skip(5).enumerate() {
-        let def = &*def.read();
-        // println!(
-        //     "{}: {}\n",
-        //     i + 5,
-        //     def.to_string(
-        //         composer.unifier.borrow_mut(),
-        //         &mut |id| format!("class{}", id),
-        //         &mut |id| format!("tvar{}", id)
-        //     )
-        // );
-        assert_eq!(
-            format!(
-                "{}: {}",
-                i + 5,
-                def.to_string(
-                    composer.unifier.borrow_mut(),
-                    &mut |id| format!("class{}", id.to_string()),
-                    &mut |id| format!("tvar{}", id.to_string()),
+    if let Err(msg) = composer.start_analysis() {
+        if print {
+            println!("{}", msg);
+        } else {
+            assert_eq!(res[0], msg);
+        }
+    } else {
+        // skip 5 to skip primitives
+        for (i, (def, _)) in composer.definition_ast_list.iter().skip(5).enumerate() {
+            let def = &*def.read();
+            
+            if print {
+                println!(
+                    "{}: {}\n",
+                    i + 5,
+                    def.to_string(
+                        composer.unifier.borrow_mut(),
+                        &mut |id| format!("class{}", id),
+                        &mut |id| format!("tvar{}", id)
+                    )
+                );
+            } else {
+                assert_eq!(
+                    format!(
+                        "{}: {}",
+                        i + 5,
+                        def.to_string(
+                            composer.unifier.borrow_mut(),
+                            &mut |id| format!("class{}", id.to_string()),
+                            &mut |id| format!("tvar{}", id.to_string()),
+                        )
+                    ),
+                    res[i]
                 )
-            ),
-            res[i]
-        )
+            }
+        }
     }
+
 }