add tests

Author: anoblet

packages/yaml/nursery.js

@@ -1,12 +1,51 @@
 const { setup } = require('@ast-grep/nursery')
 const languageRegistration = require('./index')
+const assert = require('node:assert')
 
 setup({
   dirname: __dirname,
   name: 'yaml',
   treeSitterPackage: '@tree-sitter-grammars/tree-sitter-yaml',
   languageRegistration,
   testRunner: (parse) => {
-    // add test here
+    // Test simple value matching
+    const sg1 = parse('123')
+    const root1 = sg1.root()
+    const node1 = root1.find('123')
+    assert.equal(node1.kind(), 'integer_scalar')
+    
+    // Test pattern matching with key-value pairs
+    const sg2 = parse('foo: 123')
+    const root2 = sg2.root()
+    const node2 = root2.find('foo: $BAR')
+    assert.equal(node2.kind(), 'block_mapping_pair')
+    assert.equal(node2.find('$BAR').text(), '123')
+    
+    // Test non-matching pattern
+    const node3 = root2.find('bar: $BAR')
+    assert.equal(node3, null)
+    
+    // Test nested structures
+    const sg3 = parse(`
+foo:
+  - item1
+  - item2
+`)
+    const root3 = sg3.root()
+    const node4 = root3.find('foo: $$$')
+    assert.equal(node4.kind(), 'block_mapping_pair')
+    
+    // Test replacement functionality
+    const source = `
+key: value
+list:
+  - item1
+  - item2
+`
+    const sg4 = parse(source)
+    const root4 = sg4.root()
+    const node5 = root4.find('$KEY: value')
+    const replaced = node5.replace('value: $KEY')
+    assert.equal(replaced.includes('value: key'), true)
   }
 })