test: RBS support

Cargo.toml

@@ -20,4 +20,5 @@ anyhow = "1.0.89"
 tree-sitter = "0.26"
 tree-sitter-ruby = "0.23.1"
 tree-sitter-embedded-template = "0.25.0"
+tree-sitter-rbs = "0.2.3"
 insta = { version = "1.46", features = ["yaml"] }

languages/rbs/highlights.scm

@@ -98,7 +98,7 @@
 (type_variable) @constant
 
 (namespace
-  (constant) @module)
+  (constant) @type)
 
 (builtin_type) @type.builtin
 

languages/rbs/indents.scm

@@ -5,7 +5,7 @@
   (parameters)
   (tuple_type)
   (record_type)
-] @indent.begin
+] @indent
 
 (_
   "["
@@ -18,5 +18,3 @@
 (_
   "("
   ")" @end) @indent
-
-(comment) @indent.ignore

tests/rbs.rs

@@ -0,0 +1,18 @@
+mod support;
+
+#[test]
+fn parses_and_highlights_rbs() {
+    let source = "module Billing\n  class Invoice\n    def total: (Integer quantity) -> Numeric\n  end\nend\n";
+    let query = std::fs::read_to_string("languages/rbs/highlights.scm").unwrap();
+    let captures = support::run_query(source, &query, "rbs");
+
+    assert!(captures
+        .iter()
+        .any(|capture| capture.name == "type" && capture.text == "Billing"));
+    assert!(captures
+        .iter()
+        .any(|capture| capture.name == "function.method" && capture.text == "total"));
+    assert!(captures
+        .iter()
+        .any(|capture| capture.name == "variable.parameter" && capture.text == "quantity"));
+}

tests/support/mod.rs

@@ -19,6 +19,7 @@ fn language_for_id(language_id: &str) -> tree_sitter::Language {
     match language_id {
         "ruby" => tree_sitter_ruby::LANGUAGE.into(),
         "erb" => tree_sitter_embedded_template::LANGUAGE.into(),
+        "rbs" => tree_sitter_rbs::LANGUAGE.into(),
         _ => panic!("Unsupported language id for query tests: {language_id}"),
     }
 }