more tests

avivkeller · avivkeller · commit acf3e08e7eb4 · 2025-06-05T16:48:18.000-04:00
diff --git a/src/generators/web/__tests__/index.test.mjs b/src/generators/web/__tests__/index.test.mjs
@@ -0,0 +1,45 @@
+import assert from 'node:assert/strict';
+import { describe, it, mock } from 'node:test';
+
+mock.module('node:fs/promises', {
+  namedExports: {
+    readFile: mock.fn(() => Promise.resolve('<html>{{title}}</html>')),
+    writeFile: mock.fn(() => Promise.resolve()),
+  },
+});
+
+const { executeServerCode, processEntry } = await import('../index.mjs');
+
+describe('executeServerCode', () => {
+  it('should execute and return the `code` variable', async () => {
+    const result = await executeServerCode('code = "test"', mock.fn());
+    assert.equal(typeof result, 'string');
+  });
+});
+
+describe('processEntry', () => {
+  it('should return HTML and CSS', async () => {
+    const entry = {
+      type: 'Literal',
+      data: {
+        heading: { data: { name: 'Test' } },
+        api: 'test',
+      },
+    };
+
+    const uuid = JSON.stringify(crypto.randomUUID());
+
+    const result = await processEntry(
+      entry,
+      '{{{javascript}}}',
+      {
+        buildServerProgram: mock.fn(() => ''),
+        buildClientProgram: mock.fn(() => uuid),
+      },
+      mock.fn(),
+      null
+    );
+
+    assert.ok(result.html.startsWith(uuid));
+  });
+});
diff --git a/src/generators/web/build/__tests__/plugins.test.mjs b/src/generators/web/build/__tests__/plugins.test.mjs
@@ -5,12 +5,12 @@ import { describe, it, mock } from 'node:test';
 const existsSync = mock.fn();
 mock.module('node:fs', { namedExports: { ...fs, existsSync } });
 
-const { uiComponentsResolverPlugin } = await import('../plugins.mjs');
+const { default: getPlugins } = await import('../plugins.mjs');
 
 describe('uiComponentsResolverPlugin', async () => {
   let onResolveCallback;
 
-  uiComponentsResolverPlugin.setup({
+  getPlugins(true)[0].setup({
     onResolve: (_, callback) => {
       onResolveCallback = callback;
     },
diff --git a/src/generators/web/build/plugins.mjs b/src/generators/web/build/plugins.mjs
@@ -13,7 +13,7 @@ import postcssCalc from 'postcss-calc';
  * strategy where it first tries to find an index.tsx file, then falls back to a
  * .tsx file with the same name.
  */
-export const uiComponentsResolverPlugin = {
+const uiComponentsResolverPlugin = {
   name: 'ui-components-resolver',
 
   /**
diff --git a/src/generators/web/index.mjs b/src/generators/web/index.mjs
@@ -17,7 +17,7 @@ import createASTBuilder from './build/generate.mjs';
  * @param {ReturnType<createRequire>} require - Node.js require function for dependencies
  * @returns {Promise<string>} The rendered HTML output
  */
-async function executeServerCode(serverCode, require) {
+export async function executeServerCode(serverCode, require) {
   const { js: bundledServer } = await bundleCode(serverCode, { server: true });
 
   const executedFunction = new Function(
@@ -37,7 +37,7 @@ async function executeServerCode(serverCode, require) {
  * @param {string} output - Output directory path
  * @returns {Promise<{html: string, css?: string}>}
  */
-async function processEntry(
+export async function processEntry(
   entry,
   template,
   { buildServerProgram, buildClientProgram },
