queries/nix: add injection rule for python test scripts

helix-editor · Sep 16, 2023 · b4494e1 · b4494e1
1 parent 0e55648
commit b4494e1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/runtime/queries/nix/injections.scm b/runtime/queries/nix/injections.scm
@@ -6,6 +6,16 @@
   (indented_string_expression (string_fragment) @injection.content))
   (#set! injection.combined))
 
+; Common attribute keys corresponding to Python scripts,
+; such as those for NixOS VM tests in nixpkgs/nixos/tests.
+((binding
+   attrpath: (attrpath (identifier) @_path)
+   expression: (indented_string_expression
+     (string_fragment) @injection.content))
+ (#match? @_path "(^|\\.)testScript$")
+ (#set! injection.language "python")
+ (#set! injection.combined))
+
 ; Common attribute keys corresponding to scripts,
 ; such as those of stdenv.mkDerivation.
 ((binding