test: update tests

Author: pi0

test/parse.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect } from "vitest";
-import { parseRawArgs, findAutoMdBlocks } from "../src/_parse";
+import { parseRawArgs, findBlocks } from "../src/_parse";
 
 describe("parseRawArgs", () => {
   const tests = [
@@ -19,7 +19,7 @@ describe("parseRawArgs", () => {
   }
 });
 
-describe("findAutoMdBlocks", () => {
+describe("findBlocks", () => {
   const fixture = `
 <!-- automd:pm-x args=. -->
 (a)
@@ -43,7 +43,7 @@ describe("findAutoMdBlocks", () => {
   });
 
   it("should find all blocks", () => {
-    const blocks = findAutoMdBlocks(fixture);
+    const blocks = findBlocks(fixture);
     expect(blocks[0]).toMatchObject(mkBlock("pm-x", "args=.", "(a)"));
     expect(blocks[1]).toMatchObject(
       mkBlock("pm-install", "dev no-auto", "(b)"),

test/transform.test.ts

@@ -0,0 +1,45 @@
+import { describe, it, expect } from "vitest";
+import { transform } from "../src";
+
+describe("transform", () => {
+  it("basic transform works", async () => {
+    const input = `
+<!-- automd:test foo=bar -->
+<!-- /automd -->
+    `.trim();
+
+    const result = await transform(input, {
+      generators: {
+        test: {
+          name: "test",
+          generate({ args }) {
+            return { contents: JSON.stringify({ args }) };
+          },
+        },
+      },
+    });
+
+    expect(result.hasChanged).toBe(true);
+    expect(result.contents).toMatchInlineSnapshot(`
+      "<!-- automd:test foo=bar -->
+
+      {"args":{"foo":"bar"}}
+
+      <!-- /automd -->"
+    `);
+
+    expect(result.updates).toHaveLength(1);
+    expect(result.updates[0].block).toMatchInlineSnapshot(`
+      {
+        "contents": "
+      ",
+        "generator": "test",
+        "loc": {
+          "end": 29,
+          "start": 28,
+        },
+        "rawArgs": "foo=bar",
+      }
+    `);
+  });
+});