#![cfg(test)]
use std::cell::RefCell;
use std::rc::Rc;
use std::str::FromStr;

use super::{Parser, ParseResult, tokenize};
use crate::ast::*;
use super::Declaration::*;
use super::Signature;
use super::TypeIdentifier::*;
use super::TypeSingletonName;
use super::ExpressionKind::*;
use super::Variant::*;
use super::ForBody::*;

fn make_parser(input: &str) -> Parser {
  let source_map = crate::source_map::SourceMap::new();
  let source_map_handle = Rc::new(RefCell::new(source_map));
  let tokens: Vec<crate::tokenizing::Token> = tokenize(input);
  let mut parser = super::Parser::new(source_map_handle);
  parser.add_new_tokens(tokens);
  parser
}

fn parse(input: &str) -> ParseResult<AST> {
  let mut parser = make_parser(input);
  parser.parse()
}

macro_rules! parse_test {
  ($string:expr, $correct:expr) => {
    assert_eq!(parse($string).unwrap(), $correct)
  };
}
macro_rules! parse_test_wrap_ast {
  ($string:expr, $correct:expr) => { parse_test!($string, AST { id: ItemIdStore::new_id(), statements: vec![$correct] }) }
}
macro_rules! parse_error {
  ($string:expr) => { assert!(parse($string).is_err()) }
}
macro_rules! qname {
  (  $( $component:expr),* ) => {
    {
    let mut components = vec![];
    $(
      components.push(rc!($component));
    )*
    QualifiedName { components, id: ItemIdStore::new_id() }
    }
  };
}
macro_rules! val {
  ($var:expr) => { Value(QualifiedName { components: vec![Rc::new($var.to_string())], id: ItemIdStore::new_id() }) };
}
macro_rules! ty {
  ($name:expr) => { Singleton(tys!($name)) }
}
macro_rules! tys {
  ($name:expr) => { TypeSingletonName { name: Rc::new($name.to_string()), params: vec![] } };
}

macro_rules! decl {
  ($expr_type:expr) => {
    Statement { id: ItemIdStore::new_id(), kind: StatementKind::Declaration($expr_type) }
  };
}

macro_rules! import {
  ($import_spec:expr) => {
    Statement { id: ItemIdStore::new_id(), kind: StatementKind::Import($import_spec) }
  }
}

macro_rules! module {
  ($module_spec:expr) => {
    Statement { id: ItemIdStore::new_id(), kind: StatementKind::Module($module_spec) }
  }
}

macro_rules! ex {
  ($expr_type:expr) => { Expression::new(ItemIdStore::new_id(), $expr_type) };
  ($expr_type:expr, $type_anno:expr) => { Expression::with_anno(ItemIdStore::new_id(), $expr_type, $type_anno) };
  (s $expr_text:expr) => {
    {
      let mut parser = make_parser($expr_text);
      parser.expression().unwrap()
    }
  };
}

macro_rules! inv {
  ($expr_type:expr) => { InvocationArgument::Positional($expr_type) }
}

macro_rules! binexp {
  ($op:expr, $lhs:expr, $rhs:expr) => { BinExp(BinOp::from_sigil($op), bx!(Expression::new(ItemIdStore::new_id(), $lhs).into()), bx!(Expression::new(ItemIdStore::new_id(), $rhs).into())) }
}
macro_rules! prefexp {
  ($op:expr, $lhs:expr) => { PrefixExp(PrefixOp::from_str($op).unwrap(), bx!(Expression::new(ItemIdStore::new_id(), $lhs).into())) }
}
macro_rules! exst {
  ($expr_type:expr) => { Statement { id: ItemIdStore::new_id(), kind: StatementKind::Expression(Expression::new(ItemIdStore::new_id(), $expr_type).into())} };
  ($expr_type:expr, $type_anno:expr) => { Statement { id: ItemIdStore::new_id(), kind: StatementKind::Expression(Expression::with_anno(ItemIdStore::new_id(), $expr_type, $type_anno).into())} };
  ($op:expr, $lhs:expr, $rhs:expr) => { Statement { id: ItemIdStore::new_id(), ,kind: StatementKind::Expression(ex!(binexp!($op, $lhs, $rhs)))}
  };
  (s $statement_text:expr) => {
    {
      let mut parser = make_parser($statement_text);
      parser.statement().unwrap()
    }
  }
}

#[test]
fn parsing_number_literals_and_binexps() {
  parse_test_wrap_ast! { ".2", exst!(FloatLiteral(0.2)) };
  parse_test_wrap_ast! { "8.1", exst!(FloatLiteral(8.1)) };

  parse_test_wrap_ast! { "0b010", exst!(NatLiteral(2)) };
  parse_test_wrap_ast! { "0b0_1_0_", exst!(NatLiteral(2)) }

  parse_test_wrap_ast! {"0xff", exst!(NatLiteral(255)) };
  parse_test_wrap_ast! {"0xf_f_", exst!(NatLiteral(255)) };

  parse_test_wrap_ast! {"0xf_f_+1", exst!(binexp!("+", NatLiteral(255), NatLiteral(1))) };

  parse_test! {"3; 4; 4.3",
    AST {
      id: ItemIdStore::new_id(),
      statements: vec![exst!(NatLiteral(3)), exst!(NatLiteral(4)),
      exst!(FloatLiteral(4.3))]
    }
  };

  parse_test_wrap_ast!("1 + 2 * 3",
    exst!(binexp!("+", NatLiteral(1), binexp!("*", NatLiteral(2), NatLiteral(3))))
  );

  parse_test_wrap_ast!("1 * 2 + 3",
    exst!(binexp!("+", binexp!("*", NatLiteral(1), NatLiteral(2)), NatLiteral(3)))
    ) ;

  parse_test_wrap_ast!("1 && 2", exst!(binexp!("&&", NatLiteral(1), NatLiteral(2))));

  parse_test_wrap_ast!("1 + 2 * 3 + 4", exst!(
    binexp!("+",
      binexp!("+", NatLiteral(1), binexp!("*", NatLiteral(2), NatLiteral(3))),
      NatLiteral(4))));

  parse_test_wrap_ast!("(1 + 2) * 3",
    exst!(binexp!("*", binexp!("+", NatLiteral(1), NatLiteral(2)), NatLiteral(3))));

  parse_test_wrap_ast!(".1 + .2", exst!(binexp!("+", FloatLiteral(0.1), FloatLiteral(0.2))));
  parse_test_wrap_ast!("1 / 2", exst!(binexp!("/", NatLiteral(1), NatLiteral(2))));
}

#[test]
fn parsing_tuples() {
  parse_test_wrap_ast!("()", exst!(TupleLiteral(vec![])));
  parse_test_wrap_ast!("(\"hella\", 34)", exst!(
    TupleLiteral(
      vec![ex!(s r#""hella""#).into(), ex!(s "34").into()]
    )
  ));
  parse_test_wrap_ast!("((1+2), \"slough\")", exst!(TupleLiteral(vec![
    ex!(binexp!("+", NatLiteral(1), NatLiteral(2))).into(),
    ex!(StringLiteral(rc!(slough))).into(),
  ])))
}

#[test]
fn parsing_identifiers() {
  parse_test_wrap_ast!("a", exst!(val!("a")));
  parse_test_wrap_ast!("some_value", exst!(val!("some_value")));
  parse_test_wrap_ast!("a + b", exst!(binexp!("+", val!("a"), val!("b"))));
  //parse_test!("a[b]", AST(vec![Expression(
  //parse_test!("a[]", <- TODO THIS NEEDS TO FAIL
  //parse_test("a()[b]()[d]")
  //TODO fix this parsing stuff
  /*
  parse_test! { "perspicacity()[a]", AST(vec![
    exst!(Index {
      indexee: bx!(ex!(Call { f: bx!(ex!(val!("perspicacity"))), arguments: vec![] })),
      indexers: vec![ex!(val!("a"))]
    })
  ])
  }
  */
  parse_test_wrap_ast!("a[b,c]", exst!(Index { indexee: bx!(ex!(val!("a"))), indexers: vec![ex!(val!("b")), ex!(val!("c"))]} ));

  parse_test_wrap_ast!("None", exst!(val!("None")));
  parse_test_wrap_ast!("Pandas {  a: x + y }",
    exst!(NamedStruct { name: qname!(Pandas), fields: vec![(rc!(a), ex!(binexp!("+", val!("x"), val!("y"))))]})
  );
  parse_test_wrap_ast! { "Pandas { a: n, b: q, }",
    exst!(NamedStruct { name: qname!(Pandas), fields:
    vec![(rc!(a), ex!(val!("n"))), (rc!(b), ex!(val!("q")))]
    }
    )
  };
}


#[test]
fn qualified_identifiers() {
  parse_test_wrap_ast! {
    "let q_q = Yolo::Swaggins",
    decl!(Binding { name: rc!(q_q), constant: true, type_anno: None,
    expr: Expression::new(ItemIdStore::new_id(), Value(qname!(Yolo, Swaggins))),
    })
  }

  parse_test_wrap_ast! {
    "thing::item::call()",
    exst!(Call { f: bx![ex!(Value(qname!(thing, item, call)))], arguments: vec![] })
  }
}

#[test]
fn reserved_words() {
  parse_error!("module::item::call()");
}

#[test]
fn parsing_complicated_operators() {
  parse_test_wrap_ast!("a <- b", exst!(binexp!("<-", val!("a"), val!("b"))));
  parse_test_wrap_ast!("a || b", exst!(binexp!("||", val!("a"), val!("b"))));
  parse_test_wrap_ast!("a<>b", exst!(binexp!("<>", val!("a"), val!("b"))));
  parse_test_wrap_ast!("a.b.c.d", exst!(binexp!(".",
                                              binexp!(".",
                                                binexp!(".", val!("a"), val!("b")),
                                              val!("c")),
                                              val!("d"))));
  parse_test_wrap_ast!("-3", exst!(prefexp!("-", NatLiteral(3))));
  parse_test_wrap_ast!("-0.2", exst!(prefexp!("-", FloatLiteral(0.2))));
  parse_test_wrap_ast!("!3", exst!(prefexp!("!", NatLiteral(3))));
  parse_test_wrap_ast!("a <- -b", exst!(binexp!("<-", val!("a"), prefexp!("-", val!("b")))));
  parse_test_wrap_ast!("a <--b", exst!(binexp!("<--", val!("a"), val!("b"))));
}

#[test]
fn parsing_functions() {
  parse_test_wrap_ast!("fn oi()",  decl!(FuncSig(Signature { name: rc!(oi), operator: false, params: vec![], type_anno: None })));
  parse_test_wrap_ast!("oi()", exst!(Call { f: bx!(ex!(val!("oi"))), arguments: vec![] }));
  parse_test_wrap_ast!("oi(a, 2 + 2)", exst!(Call
  { f: bx!(ex!(val!("oi"))),
    arguments: vec![inv!(ex!(val!("a"))), inv!(ex!(binexp!("+", NatLiteral(2), NatLiteral(2)))).into()]
  }));
  parse_error!("a(b,,c)");

  parse_test_wrap_ast!("fn a(b, c: Int): Int", decl!(
      FuncSig(Signature { name: rc!(a), operator: false, params: vec![
        FormalParam { name: rc!(b), anno: None, default: None },
        FormalParam { name: rc!(c), anno: Some(ty!("Int")), default: None }
      ], type_anno: Some(ty!("Int")) })));


  parse_test_wrap_ast!("fn a(x) { x() }", decl!(
      FuncDecl(Signature { name: rc!(a), operator: false, params: vec![FormalParam { name: rc!(x), anno: None, default: None }], type_anno: None },
      vec![exst!(Call { f: bx!(ex!(val!("x"))), arguments: vec![] })])));
  parse_test_wrap_ast!("fn a(x) {\n x() }", decl!(
      FuncDecl(Signature { name: rc!(a), operator: false, params: vec![FormalParam { name: rc!(x), anno: None, default: None }], type_anno: None },
      vec![exst!(Call { f: bx!(ex!(val!("x"))), arguments: vec![] })])));

  let multiline = r#"
fn a(x) {
x()
}
"#;
parse_test_wrap_ast!(multiline, decl!(
    FuncDecl(Signature { name: rc!(a), operator: false, params: vec![FormalParam { name: rc!(x), default: None, anno: None }], type_anno: None },
    vec![exst!(Call { f: bx!(ex!(val!("x"))), arguments: vec![] })])));
  let multiline2 = r#"
fn a(x) {

x()

}
"#;
parse_test_wrap_ast!(multiline2, decl!(
    FuncDecl(Signature { name: rc!(a), operator: false, params: vec![FormalParam { name: rc!(x), default: None, anno: None }], type_anno: None },
    vec![exst!(s "x()")])));
}

#[test]
fn functions_with_default_args() {
  parse_test_wrap_ast! {
    "fn func(x: Int, y: Int = 4) { }",
    decl!(
      FuncDecl(Signature { name: rc!(func), operator: false, type_anno: None, params: vec![
        FormalParam { name: rc!(x), default: None, anno: Some(ty!("Int")) },
        FormalParam { name: rc!(y), default: Some(ex!(s "4")), anno: Some(ty!("Int")) }
      ]}, vec![])
    )
  };
}

#[test]
fn parsing_bools() {
  parse_test_wrap_ast!("false", exst!(BoolLiteral(false)));
  parse_test_wrap_ast!("true", exst!(BoolLiteral(true)));
}

#[test]
fn parsing_strings() {
  parse_test_wrap_ast!(r#""hello""#, exst!(StringLiteral(rc!(hello))));
}

#[test]
fn parsing_types() {
  parse_test_wrap_ast!("type Yolo = Yolo", decl!(TypeDecl { name: tys!("Yolo"), body: TypeBody(vec![UnitStruct(rc!(Yolo))]), mutable: false} ));
  parse_test_wrap_ast!("type mut Yolo = Yolo", decl!(TypeDecl { name: tys!("Yolo"), body: TypeBody(vec![UnitStruct(rc!(Yolo))]), mutable: true} ));
  parse_test_wrap_ast!("type alias Sex = Drugs", decl!(TypeAlias { alias: rc!(Sex), original: rc!(Drugs) }));
  parse_test_wrap_ast!("type Sanchez = Miguel | Alejandro(Int, Option<a>) | Esperanza { a: Int, b: String }",
  decl!(TypeDecl {
    name: tys!("Sanchez"),
    body: TypeBody(vec![
      UnitStruct(rc!(Miguel)),
      TupleStruct(rc!(Alejandro), vec![
        Singleton(TypeSingletonName { name: rc!(Int), params: vec![] }),
        Singleton(TypeSingletonName { name: rc!(Option), params: vec![Singleton(TypeSingletonName { name: rc!(a), params: vec![] })] }),
      ]),
      Record{
        name: rc!(Esperanza),
        members: vec![
          (rc!(a), Singleton(TypeSingletonName { name: rc!(Int), params: vec![] })),
          (rc!(b), Singleton(TypeSingletonName { name: rc!(String), params: vec![] })),
        ]
      }
    ]),
    mutable: false
  }));

  parse_test_wrap_ast! {
    "type Jorge<a> = Diego | Kike(a)",
    decl!(TypeDecl{
      name: TypeSingletonName { name: rc!(Jorge), params: vec![Singleton(TypeSingletonName { name: rc!(a), params: vec![] })] },
      body: TypeBody(vec![UnitStruct(rc!(Diego)), TupleStruct(rc!(Kike), vec![Singleton(TypeSingletonName { name: rc!(a), params: vec![] })])]),
      mutable: false
    }
    )
  };
}

#[test]
fn parsing_bindings() {
  parse_test_wrap_ast!("let mut a = 10", decl!(Binding { name: rc!(a), constant: false, type_anno: None, expr: ex!(NatLiteral(10)) } ));
  parse_test_wrap_ast!("let a = 2 + 2", decl!(Binding { name: rc!(a), constant: true, type_anno: None, expr: ex!(binexp!("+", NatLiteral(2), NatLiteral(2))) }));
  parse_test_wrap_ast!("let a: Nat = 2 + 2", decl!(
      Binding { name: rc!(a), constant: true, type_anno: Some(Singleton(TypeSingletonName { name: rc!(Nat), params: vec![] })),
      expr: ex!(binexp!("+", NatLiteral(2), NatLiteral(2))) }
  ));
}

#[test]
fn parsing_block_expressions() {
  parse_test_wrap_ast! {
    "if a() then { b(); c() }", exst!(
      IfExpression {
        discriminator: Some(bx! {
          ex!(Call { f: bx!(ex!(val!("a"))), arguments: vec![]})
        }),
        body: bx! {
          IfExpressionBody::SimpleConditional {
            then_case: vec![exst!(Call { f: bx!(ex!(val!("b"))), arguments: vec![]}), exst!(Call { f: bx!(ex!(val!("c"))), arguments: vec![] })],
            else_case: None,
          }
        }
      }
    )
  };

  parse_test_wrap_ast! {
    "if a() then { b(); c() } else { q }", exst!(
      IfExpression {
        discriminator: Some(bx! {
          ex!(Call { f: bx!(ex!(val!("a"))), arguments: vec![]})
        }),
        body: bx! {
          IfExpressionBody::SimpleConditional {
            then_case: vec![exst!(Call { f: bx!(ex!(val!("b"))), arguments: vec![]}), exst!(Call { f: bx!(ex!(val!("c"))), arguments: vec![] })],
            else_case: Some(vec![exst!(val!("q"))]),
          }
        }
      }
    )
  };

  /*
  parse_test!("if a() then { b(); c() }", AST(vec![exst!(
      IfExpression(bx!(ex!(Call { f: bx!(ex!(val!("a"))), arguments: vec![]})),
        vec![exst!(Call { f: bx!(ex!(val!("b"))), arguments: vec![]}), exst!(Call { f: bx!(ex!(val!("c"))), arguments: vec![] })],
        None)
      )]));
  parse_test!(r#"
  if true then {
    const a = 10
    b
  } else {
    c
  }"#,
  AST(vec![exst!(IfExpression(bx!(ex!(BoolLiteral(true))),
    vec![decl!(Binding { name: rc!(a), constant: true, expr: ex!(NatLiteral(10)) }),
          exst!(val!(rc!(b)))],
    Some(vec![exst!(val!(rc!(c)))])))])
  );

  parse_test!("if a { b } else { c }", AST(vec![exst!(
        IfExpression(bx!(ex!(val!("a"))),
          vec![exst!(val!("b"))],
          Some(vec![exst!(val!("c"))])))]));

  parse_test!("if (A {a: 1}) { b } else { c }", AST(vec![exst!(
        IfExpression(bx!(ex!(NamedStruct { name: rc!(A), fields: vec![(rc!(a), ex!(NatLiteral(1)))]})),
          vec![exst!(val!("b"))],
          Some(vec![exst!(val!("c"))])))]));

  parse_error!("if A {a: 1} { b } else { c }");
  */
}
#[test]
fn parsing_interfaces() {
  parse_test_wrap_ast!("interface Unglueable { fn unglue(a: Glue); fn mar(): Glue }",
  decl!(Interface {
    name: rc!(Unglueable),
    signatures: vec![
      Signature {
        name: rc!(unglue),
        operator: false,
        params: vec![
          FormalParam { name: rc!(a), anno: Some(Singleton(TypeSingletonName { name: rc!(Glue), params: vec![] })), default: None }
        ],
        type_anno: None
      },
      Signature { name: rc!(mar), operator: false, params: vec![], type_anno: Some(Singleton(TypeSingletonName { name: rc!(Glue), params: vec![] })) },
    ]
  })
  );
}

#[test]
fn parsing_impls() {
  parse_test_wrap_ast!("impl Heh { fn yolo(); fn swagg(); }",
  decl!(Impl {
    type_name: ty!("Heh"),
    interface_name: None,
    block: vec![
      FuncSig(Signature { name: rc!(yolo), operator: false, params: vec![], type_anno: None }),
      FuncSig(Signature { name: rc!(swagg), operator: false, params: vec![], type_anno: None })
    ] }));

  parse_test_wrap_ast!("impl Mondai for Lollerino { fn yolo(); fn swagg(); }",
  decl!(Impl {
    type_name: ty!("Lollerino"),
    interface_name: Some(TypeSingletonName { name: rc!(Mondai), params: vec![] }),
    block: vec![
      FuncSig(Signature { name: rc!(yolo), operator: false, params: vec![], type_anno: None}),
      FuncSig(Signature { name: rc!(swagg), operator: false, params: vec![], type_anno: None })
    ] }));

  parse_test_wrap_ast!("impl Hella<T> for (Alpha, Omega) { }",
  decl!(Impl {
    type_name: Tuple(vec![ty!("Alpha"), ty!("Omega")]),
    interface_name: Some(TypeSingletonName { name: rc!(Hella), params: vec![ty!("T")] }),
    block: vec![]
  })
  );

  parse_test_wrap_ast!("impl Option<WTFMate> { fn oi() }",
  decl!(Impl {
    type_name: Singleton(TypeSingletonName { name: rc!(Option), params: vec![ty!("WTFMate")]}),
    interface_name: None,
    block: vec![
      FuncSig(Signature { name: rc!(oi), operator: false, params: vec![], type_anno: None }),
    ]
  }));
}

#[test]
fn parsing_type_annotations() {
  parse_test_wrap_ast!("let a = b : Int",
    decl!(Binding { name: rc!(a), constant: true, type_anno: None, expr:
      ex!(val!("b"), ty!("Int")) }));

  parse_test_wrap_ast!("a : Int",
    exst!(val!("a"), ty!("Int"))
  );

  parse_test_wrap_ast!("a : Option<Int>",
    exst!(val!("a"), Singleton(TypeSingletonName { name: rc!(Option), params: vec![ty!("Int")] }))
  );

  parse_test_wrap_ast!("a : KoreanBBQSpecifier<Kimchi, Option<Bulgogi> >",
    exst!(val!("a"), Singleton(TypeSingletonName { name: rc!(KoreanBBQSpecifier), params: vec![
      ty!("Kimchi"), Singleton(TypeSingletonName { name: rc!(Option), params: vec![ty!("Bulgogi")] })
    ] }))
  );

  parse_test_wrap_ast!("a : (Int, Yolo<a>)",
    exst!(val!("a"), Tuple(
      vec![ty!("Int"), Singleton(TypeSingletonName {
        name: rc!(Yolo), params: vec![ty!("a")]
      })])));
}

#[test]
fn parsing_lambdas() {
  parse_test_wrap_ast! { r#"\(x) { x + 1}"#, exst!(
      Lambda { params: vec![FormalParam { name: rc!(x), anno: None, default: None } ], type_anno: None, body: vec![exst!(s "x + 1")] }
    )
  }

  parse_test_wrap_ast!(r#"\  (x: Int, y) { a;b;c;}"#,
    exst!(Lambda {
      params: vec![
        FormalParam { name: rc!(x), anno: Some(ty!("Int")), default: None },
        FormalParam { name: rc!(y), anno: None, default: None }
      ],
      type_anno: None,
      body: vec![exst!(s "a"), exst!(s "b"), exst!(s "c")]
    })
  );

  parse_test_wrap_ast! { r#"\(x){y}(1)"#,
    exst!(Call { f: bx!(ex!(
      Lambda {
        params: vec![
          FormalParam { name: rc!(x), anno: None, default: None }
        ],
        type_anno: None,
        body: vec![exst!(s "y")] }
      )),
      arguments: vec![inv!(ex!(NatLiteral(1))).into()] })
  };

  parse_test_wrap_ast! {
    r#"\(x: Int): String { "q" }"#,
    exst!(Lambda {
      params: vec![
        FormalParam { name: rc!(x), anno: Some(ty!("Int")), default: None },
      ],
      type_anno: Some(ty!("String")),
      body: vec![exst!(s r#""q""#)]
    })
  }
}

#[test]
fn single_param_lambda() {
  parse_test_wrap_ast! {
    r"\x { x + 10 }",
    exst!(Lambda {
      params: vec![FormalParam { name: rc!(x), anno: None, default: None }],
      type_anno: None,
      body: vec![exst!(s r"x + 10")]
    })
  }

  parse_test_wrap_ast! {
    r"\x: Nat { x + 10 }",
    exst!(Lambda {
      params: vec![FormalParam { name: rc!(x), anno: Some(ty!("Nat")), default: None }],
      type_anno: None,
      body: vec![exst!(s r"x + 10")]
    })
  }
}

#[test]
fn more_advanced_lambdas() {
  parse_test! {
    r#"fn wahoo() { let a = 10; \(x) { x + a } };
    wahoo()(3) "#,
    AST {
      id: ItemIdStore::new_id(),
      statements: vec![
        exst!(s r"fn wahoo() { let a = 10; \(x) { x + a } }"),
        exst! {
        Call {
          f: bx!(ex!(Call { f: bx!(ex!(val!("wahoo"))), arguments: vec![] })),
          arguments: vec![inv!(ex!(NatLiteral(3))).into()],
        }
      }
    ]
  }
  }
}

#[test]
  fn list_literals() {
    parse_test_wrap_ast! {
      "[1,2]",
        exst!(ListLiteral(vec![ex!(NatLiteral(1)), ex!(NatLiteral(2))]))
    };
  }

#[test]
fn while_expr() {
  parse_test_wrap_ast! {
    "while { }",
    exst!(WhileExpression { condition: None, body: vec![] })
  }

  parse_test_wrap_ast! {
    "while a == b { }",
    exst!(WhileExpression { condition: Some(bx![ex![binexp!("==", val!("a"), val!("b"))]]), body: vec![] })
  }
}

#[test]
fn for_expr() {
  parse_test_wrap_ast! {
    "for { a <- maybeValue } return 1",
    exst!(ForExpression {
      enumerators: vec![Enumerator { id: rc!(a), generator: ex!(val!("maybeValue")) }],
      body: bx!(MonadicReturn(ex!(s "1")))
    })
  }

  parse_test_wrap_ast! {
    "for n <- someRange { f(n); }",
    exst!(ForExpression { enumerators: vec![Enumerator { id: rc!(n), generator: ex!(val!("someRange"))}],
      body: bx!(ForBody::StatementBlock(vec![exst!(s "f(n)")]))
    })
  }
}

#[test]
fn patterns() {
  parse_test_wrap_ast! {
    "if x is Some(a) then { 4 } else { 9 }", exst!(
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::SimplePatternMatch {
          pattern: Pattern::TupleStruct(qname!(Some), vec![Pattern::VarOrName(qname!(a))]),
          then_case: vec![exst!(s "4")],
          else_case: Some(vec![exst!(s "9")]) })
        }
      )
  }

  parse_test_wrap_ast! {
    "if x is Some(a) then 4 else 9", exst!(
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::SimplePatternMatch {
            pattern: Pattern::TupleStruct(qname!(Some), vec![Pattern::VarOrName(qname!(a))]),
            then_case: vec![exst!(s "4")],
            else_case: Some(vec![exst!(s "9")]) }
          )
        }
    )
  }

  parse_test_wrap_ast! {
    "if x is Something { a, b: x } then { 4 } else { 9 }", exst!(
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::SimplePatternMatch {
          pattern: Pattern::Record(qname!(Something), vec![
                (rc!(a),Pattern::Literal(PatternLiteral::StringPattern(rc!(a)))),
                (rc!(b),Pattern::VarOrName(qname!(x)))
            ]),
          then_case: vec![exst!(s "4")],
          else_case: Some(vec![exst!(s "9")])
          }
        )
      }
    )
  }
}

#[test]
fn pattern_literals() {
  parse_test_wrap_ast! {
    "if x is -1 then 1 else 2",
    exst!(
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::SimplePatternMatch {
            pattern: Pattern::Literal(PatternLiteral::NumPattern { neg: true, num: NatLiteral(1) }),
            then_case: vec![exst!(NatLiteral(1))],
            else_case: Some(vec![exst!(NatLiteral(2))]),
        })
      }
    )
  }

  parse_test_wrap_ast! {
    "if x is 1 then 1 else 2",
    exst!(
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::SimplePatternMatch {
          pattern: Pattern::Literal(PatternLiteral::NumPattern { neg: false, num: NatLiteral(1) }),
          then_case: vec![exst!(s "1")],
          else_case: Some(vec![exst!(s "2")]),
        })
      }
    )
  }

  parse_test_wrap_ast! {
    "if x is true then 1 else 2",
    exst!(
    IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(
            IfExpressionBody::SimplePatternMatch {
              pattern: Pattern::Literal(PatternLiteral::BoolPattern(true)),
              then_case: vec![exst!(NatLiteral(1))],
              else_case: Some(vec![exst!(NatLiteral(2))]),
          })
      }
    )
  }

  parse_test_wrap_ast! {
    "if x is \"gnosticism\" then 1 else 2",
      exst!(
        IfExpression {
          discriminator: Some(bx!(ex!(s "x"))),
          body: bx!(IfExpressionBody::SimplePatternMatch {
              pattern: Pattern::Literal(PatternLiteral::StringPattern(rc!(gnosticism))),
              then_case: vec![exst!(s "1")],
              else_case: Some(vec![exst!(s "2")]),
          })
        }
      )
  }
}

#[test]
fn imports() {
  parse_test_wrap_ast! {
    "import harbinger::draughts::Norgleheim",
    import!(ImportSpecifier { 
      id: ItemIdStore::new_id(),
      path_components: vec![rc!(harbinger), rc!(draughts), rc!(Norgleheim)],
      imported_names: ImportedNames::LastOfPath
    })
  }
}

#[test]
fn imports_2() {
  parse_test_wrap_ast! {
    "import harbinger::draughts::{Norgleheim, Xraksenlaigar}",
    import!(ImportSpecifier { 
      id: ItemIdStore::new_id(),
      path_components: vec![rc!(harbinger), rc!(draughts)],
      imported_names: ImportedNames::List(vec![
        rc!(Norgleheim),
        rc!(Xraksenlaigar)
      ])
    })
  }
}

#[test]
fn imports_3() {
  parse_test_wrap_ast! {
    "import bespouri::{}",
    import!(ImportSpecifier { 
      id: ItemIdStore::new_id(),
      path_components: vec![rc!(bespouri)],
      imported_names: ImportedNames::List(vec![])
    })
  }
}


#[test]
fn imports_4() {
  parse_test_wrap_ast! {
    "import bespouri::*",
    import!(ImportSpecifier {
      id: ItemIdStore::new_id(),
      path_components: vec![rc!(bespouri)],
      imported_names: ImportedNames::All
    })
  }
}


#[test]
fn if_expr() {
  parse_test_wrap_ast! {
    "if x { is 1 then 5, else 20 }",
    exst! {
      IfExpression {
        discriminator: Some(bx!(ex!(s "x"))),
        body: bx!(IfExpressionBody::CondList(
          vec![
            ConditionArm {
              condition: Condition::Pattern(Pattern::Literal(PatternLiteral::NumPattern { neg: false, num: NatLiteral(1)})),
              guard: None,
              body: vec![exst!(s "5")],
            },
            ConditionArm {
              condition: Condition::Else,
              guard: None,
              body: vec![exst!(s "20")],
            },
          ]
        ))
      }
    }
  }
}

#[test]
fn modules() {
  parse_test_wrap_ast! {
r#"
  module ephraim {
    let a = 10
    fn nah() { 33 }
  }
"#,
module!(
  ModuleSpecifier { name: rc!(ephraim), contents: vec![
    decl!(Binding { name: rc!(a), constant: true, type_anno: None, expr: ex!(s "10") }),
    decl!(FuncDecl(Signature { name: rc!(nah), operator: false, params: vec![], type_anno: None }, vec![exst!(NatLiteral(33))])),
  ] }
)
  }
}