; Rust mod test
((attribute_item
  (attribute
    (identifier) @_attribute
    arguments: ((token_tree
      (identifier) @_test)
      (#eq? @_test "test")))
  (#eq? @_attribute "cfg"))
  .
  (mod_item
    name: (_) @run)
  (#set! tag rust-mod-test))

; Rust test
(((attribute_item
  (attribute
    [
      (identifier) @_attribute
      (scoped_identifier
        (identifier) @_attribute)
    ])
  (#match? @_attribute "test")) @_start
  .
  (attribute_item)*
  .
  [
    (line_comment)
    (block_comment)
  ]*
  .
  (function_item
    name: (_) @run @_test_name
    body: _) @_end)
  (#set! tag rust-test))

; Rust doc test
(((line_comment)*
  (line_comment
    doc: (_) @_comment_content) @_start @run
  (#match? @_comment_content "```")
  .
  (line_comment)*
  .
  (line_comment
    doc: (_) @_end_comment_content) @_end_code_block
  (#match? @_end_comment_content "```")
  .
  (line_comment)*
  (attribute_item)*
  .
  [
    (function_item
      name: (_) @_doc_test_name
      body: _)
    (function_signature_item
      name: (_) @_doc_test_name)
    (struct_item
      name: (_) @_doc_test_name)
    (enum_item
      name: (_) @_doc_test_name
      body: _)
    ((attribute_item)?
      (macro_definition
        name: (_) @_doc_test_name))
    (mod_item
      name: (_) @_doc_test_name)
  ] @_end)
  (#set! tag rust-doc-test))

; Rust main function
(((function_item
  name: (_) @run
  body: _) @_rust_main_function_end
  (#eq? @run "main"))
  (#set! tag rust-main))