plotnik-compiler 0.3.2

use crate::Query;
use indoc::indoc;

#[test]
fn printer_with_spans() {
    let input = "Q = (call)";
    let q = Query::expect(input);

    let res = q.printer().with_spans(true).dump();

    insta::assert_snapshot!(res, @r"
    Root [0..10]
      Def [0..10] Q
        NamedNode [4..10] call
    ");
}

#[test]
fn printer_with_arities() {
    let input = "Q = (call)";
    let q = Query::expect(input);

    let res = q.printer().with_arities(true).dump();

    insta::assert_snapshot!(res, @r"
    Root¹
      Def¹ Q
        NamedNode¹ call
    ");
}

#[test]
fn printer_cst_with_trivia() {
    let input = "Q = {(a) (b)}";
    let q = Query::expect(input);

    let res = q.printer().raw(true).with_trivia(true).dump();

    insta::assert_snapshot!(res, @r#"
    Root
      Def
        Id "Q"
        Whitespace " "
        Equals "="
        Whitespace " "
        Seq
          BraceOpen "{"
          Tree
            ParenOpen "("
            Id "a"
            ParenClose ")"
          Whitespace " "
          Tree
            ParenOpen "("
            Id "b"
            ParenClose ")"
          BraceClose "}"
    "#);
}

#[test]
fn printer_alt_branches() {
    let input = indoc! {r#"
        Q = [A: (a) B: (b)]
    "#};
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        Alt
          Branch A:
            NamedNode a
          Branch B:
            NamedNode b
    ");
}

#[test]
fn printer_capture_with_type() {
    let input = "Q = (call) @x :: T";
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        CapturedExpr @x :: T
          NamedNode call
    ");
}

#[test]
fn printer_quantifiers() {
    let input = "Q = { (a)* (b)+ (c)? }";
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        Seq
          QuantifiedExpr *
            NamedNode a
          QuantifiedExpr +
            NamedNode b
          QuantifiedExpr ?
            NamedNode c
    ");
}

#[test]
fn printer_field() {
    let input = "Q = (call name: (id))";
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        NamedNode call
          FieldExpr name:
            NamedNode id
    ");
}

#[test]
fn printer_negated_field() {
    let input = "Q = (call -name)";
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        NamedNode call
          NegatedField -name
    ");
}

#[test]
fn printer_wildcard_and_anchor() {
    let input = "Q = (call _ . (arg))";
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Q
        NamedNode call
          AnonymousNode (any)
          .
          NamedNode arg
    ");
}

#[test]
fn printer_string_literal() {
    let input = r#"Q = (call "foo")"#;
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r#"
    Root
      Def Q
        NamedNode call
          AnonymousNode "foo"
    "#);
}

#[test]
fn printer_ref() {
    let input = indoc! {r#"
        Expr = (call)
        Q = (func (Expr))
    "#};
    let q = Query::expect(input);

    let res = q.printer().dump();

    insta::assert_snapshot!(res, @r"
    Root
      Def Expr
        NamedNode call
      Def Q
        NamedNode func
          Ref Expr
    ");
}

#[test]
fn printer_symbols_with_arities() {
    let input = indoc! {r#"
        A = (a)
        B = {(b) (c)}
        Q = (entry (A) (B))
    "#};
    let q = Query::expect(input);

    let res = q.printer().only_symbols(true).with_arities(true).dump();

    insta::assert_snapshot!(res, @r"
    A¹
    B⁺
    Q¹
      A¹
      B⁺
    ");
}

#[test]
fn printer_symbols_with_refs() {
    let input = indoc! {r#"
        A = (a)
        B = (b (A))
        Q = (entry (B))
    "#};
    let q = Query::expect(input);

    let res = q.printer().only_symbols(true).dump();

    insta::assert_snapshot!(res, @r"
    A
    B
      A
    Q
      B
        A
    ");
}

#[test]
fn printer_symbols_cycle() {
    let input = indoc! {r#"
        A = [(a) (B)]
        B = [(b) (A)]
        Q = (entry (A))
    "#};
    let q = Query::expect(input);

    let res = q.printer().only_symbols(true).dump();

    insta::assert_snapshot!(res, @r"
    A
      B
        A (cycle)
    B
      A
        B (cycle)
    Q
      A
        B
          A (cycle)
    ");
}

#[test]
fn printer_symbols_undefined_ref() {
    let input = "Q = (call (Undefined))";
    let q = Query::expect(input);

    let res = q.printer().only_symbols(true).dump();

    insta::assert_snapshot!(res, @r"
    Q
      Undefined?
    ");
}

#[test]
fn printer_symbols_broken_ref() {
    let input = "A = (foo (Undefined))";
    let q = Query::expect(input);

    let res = q.printer().only_symbols(true).dump();

    insta::assert_snapshot!(res, @r"
    A
      Undefined?
    ");
}

#[test]
fn printer_spans_comprehensive() {
    let input = indoc! {r#"
        Foo = (call name: (id) -bar)
        Q = [(a) (b)]
    "#};
    let q = Query::expect(input);

    let res = q.printer().with_spans(true).dump();

    insta::assert_snapshot!(res, @r"
    Root [0..43]
      Def [0..28] Foo
        NamedNode [6..28] call
          FieldExpr [12..22] name:
            NamedNode [18..22] id
          NegatedField [23..27] -bar
      Def [29..42] Q
        Alt [33..42]
          Branch [34..37]
            NamedNode [34..37] a
          Branch [38..41]
            NamedNode [38..41] b
    ");
}

#[test]
fn printer_spans_seq() {
    let input = "Q = {(a) (b)}";
    let q = Query::expect(input);

    let res = q.printer().with_spans(true).dump();

    insta::assert_snapshot!(res, @r"
    Root [0..13]
      Def [0..13] Q
        Seq [4..13]
          NamedNode [5..8] a
          NamedNode [9..12] b
    ");
}

#[test]
fn printer_spans_quantifiers() {
    let input = "Q = { (a)* (b)+ }";
    let q = Query::expect(input);

    let res = q.printer().with_spans(true).dump();

    insta::assert_snapshot!(res, @r"
    Root [0..17]
      Def [0..17] Q
        Seq [4..17]
          QuantifiedExpr [6..10] *
            NamedNode [6..9] a
          QuantifiedExpr [11..15] +
            NamedNode [11..14] b
    ");
}

#[test]
fn printer_spans_alt_branches() {
    let input = "Q = [A: (a) B: (b)]";
    let q = Query::expect(input);

    let res = q.printer().with_spans(true).dump();

    insta::assert_snapshot!(res, @r"
    Root [0..19]
      Def [0..19] Q
        Alt [4..19]
          Branch [5..11] A:
            NamedNode [8..11] a
          Branch [12..18] B:
            NamedNode [15..18] b
    ");
}