feat: extract common test logic to nursery

Author: HerringtonDarkholme

packages/toml/nursery.js

@@ -1,13 +1,15 @@
 const { setup } = require('@ast-grep/nursery')
 const toml = require('./index')
+const assert = require('node:assert')
 
 setup({
   name: 'toml',
   packageName: '@tree-sitter-grammars/tree-sitter-toml',
-  testConfig: {
-    languageRegistration: toml,
-    code: `
-    $NAME = "toml"
-    `
+  languageRegistration: toml,
+  testRunner: (parse) => {
+    const sg = parse('name = "Tom"')
+    const root = sg.root()
+    const pair = root.find('$NAME = "Tom"')
+    assert.equal(pair.kind(), 'pair')
   }
 })

scripts/nursery/index.js

@@ -8,30 +8,25 @@ async function copySrc(packageName) {
   await cp(src, 'src', { recursive: true })
 }
 
-function test(name, lang, code) {
-  registerDynamicLanguage({ [name]: lang })
-  const tree = parse(name, code)
-  const root = tree.root()
-  const node = root.find('$NAME = "toml"')
-  console.log(node.kind())
+function test(setupConfig) {
+  const { name, languageRegistration, testRunner } = setupConfig
+  registerDynamicLanguage({ [name]: languageRegistration })
+  testRunner((code) => parse(name, code))
 }
 
 /** Setup ast-grep/lang package's pre-release and
  * @param {Object} setupConfig
- * @param {string} setupConfig.packageName
  * @param {string} setupConfig.name
- * @param {Object} setupConfig.testConfig
- * @param {Object} setupConfig.testConfig.languageRegistration
- * @param {string} setupConfig.testConfig.code
+ * @param {string} setupConfig.packageName
+ * @param {Function} setupConfig.testRunner
  * @returns {void}
  */
 function setup(setupConfig){
   const arg = process.argv[2]
   if (arg === 'copy') {
     copySrc(setupConfig.packageName)
   } else if (arg === 'test') {
-    const testConfig = setupConfig.testConfig
-    test(setupConfig.name, testConfig.languageRegistration, testConfig.code)
+    test(setupConfig)
   }
 }