@omg-dev/vite-plugin 0.4.24

package/src/feedback-inject.test.ts

@@ -0,0 +1,79 @@
+// Verifies the build-time injection of the shake-to-report feedback widget
+// (@omg-dev/sdk/feedback/auto) into the app entry. The brand badge bundles
+// feedback when it's on (the default), so the STANDALONE widget only injects
+// when the brand badge is disabled — these tests run with brandBadge:false.
+// Opt-out via vibes({ feedback: false }), build-only, app-entry-only.
+
+import { describe, expect, test } from "bun:test"
+import vibes from "./index.ts"
+
+const ROOT = "/app"
+const ENTRY = `${ROOT}/src/main.tsx`
+const FEEDBACK_IMPORT = `import "@omg-dev/sdk/feedback/auto";`
+
+// Minimal Rollup plugin-context stub: resolve() succeeds for everything,
+// warn() is captured. transform() is a plain async fn we bind `this` onto.
+function ctx() {
+  const warnings: string[] = []
+  return {
+    warnings,
+    resolve: async (src: string) => ({ id: src }),
+    warn: (msg: string) => warnings.push(msg),
+  }
+}
+
+async function runTransform(
+  plugin: ReturnType<typeof vibes>,
+  code: string,
+  id: string,
+  context: ReturnType<typeof ctx>,
+) {
+  const fn = plugin.transform as (this: unknown, code: string, id: string) => Promise<unknown>
+  return fn.call(context, code, id) as Promise<{ code: string } | null>
+}
+
+function buildPlugin(opts?: Parameters<typeof vibes>[0]) {
+  const plugin = vibes({ root: ROOT, brandBadge: false, ...opts })
+  // configResolved sets command + root that transform gates on.
+  const configResolved = plugin.configResolved as (cfg: unknown) => void
+  configResolved({ command: "build", root: ROOT })
+  return plugin
+}
+
+describe("feedback widget injection", () => {
+  test("default ON: appends the auto-mount import to the app entry", async () => {
+    const plugin = buildPlugin()
+    const out = await runTransform(plugin, "export default 1", ENTRY, ctx())
+    expect(out?.code).toContain(FEEDBACK_IMPORT)
+  })
+
+  test("opt-out via feedback:false omits the import", async () => {
+    const plugin = buildPlugin({ feedback: false, pwa: false })
+    const out = await runTransform(plugin, "export default 1", ENTRY, ctx())
+    // Nothing to inject → transform returns null.
+    expect(out).toBeNull()
+  })
+
+  test("injected exactly once across multiple entry transforms", async () => {
+    const plugin = buildPlugin({ pwa: false })
+    const first = await runTransform(plugin, "export default 1", ENTRY, ctx())
+    const second = await runTransform(plugin, "export default 2", ENTRY, ctx())
+    expect(first?.code).toContain(FEEDBACK_IMPORT)
+    expect(second).toBeNull() // already injected
+  })
+
+  test("non-entry modules are untouched", async () => {
+    const plugin = buildPlugin()
+    const out = await runTransform(plugin, "export const x = 1", `${ROOT}/src/util.ts`, ctx())
+    expect(out).toBeNull()
+  })
+
+  test("warns (and skips) when @omg-dev/sdk cannot be resolved", async () => {
+    const plugin = buildPlugin({ pwa: false })
+    const context = ctx()
+    context.resolve = async () => null // simulate missing dependency
+    const out = await runTransform(plugin, "export default 1", ENTRY, context)
+    expect(out).toBeNull()
+    expect(context.warnings.some((w) => w.includes("shake-to-report skipped"))).toBe(true)
+  })
+})