tests/tinytest/test-rule.R

In astgrepr: Parse and Manipulate R Code

source("helpers.R")
using("astgrepr")

expect_error(
  ast_rule(),
  "requires at least one non-NULL parameter"
)

expect_snapshot(
  "ast_rule_basic_pattern",
  ast_rule(pattern = "print($A)")
)

expect_snapshot(
  "ast_rule_basic_pattern_with_id",
  ast_rule(pattern = "print($A)", id = "print-pattern")
)

expect_snapshot(
  "ast_rule_basic_kind",
  ast_rule(kind = "field_definition")
)

expect_snapshot(
  "ast_rule_basic_regex",
  ast_rule(regex = "\\w+")
)

expect_snapshot(
  "ast_rule_with_pattern_rule",
  ast_rule(
    pattern = pattern_rule(
      selector = "field_definition",
      context = "class A { $FIELD = $INIT }"
    )
  )
)

expect_snapshot(
  "ast_rule_with_pattern_and_relational_rule",
  ast_rule(
    pattern = "any(duplicated($VAR))",
    inside = relational_rule(
      field = "if_statement"
    )
  )
)

expect_snapshot(
  "ast_rule_with_nested_any",
  ast_rule(
    pattern = "print($A)",
    inside = ast_rule(
      any = ast_rule(
        kind = c("for_statement", "while_statement")
      )
    )
  )
)