@codeyam/codeyam-cli 0.1.0-staging.d3e886e → 0.1.0-staging.d4f25c3

package/codeyam-cli/src/webserver/__tests__/editorProxy.test.js

@@ -0,0 +1,628 @@
+import http from 'http';
+import zlib from 'zlib';
+import { injectHealthScript, PREVIEW_HEALTH_SCRIPT, getPreviewHealthReport, resetPreviewHealth, } from "../editorProxy.js";
+describe('editorProxy', () => {
+    describe('normalizeTargetUrl', () => {
+        it('should leave localhost as-is (no longer normalizes to 127.0.0.1)', () => {
+            const { normalizeTargetUrl } = require('../editorProxy');
+            expect(normalizeTargetUrl('http://localhost:3112')).toBe('http://localhost:3112');
+        });
+        it('should leave 127.0.0.1 unchanged', () => {
+            const { normalizeTargetUrl } = require('../editorProxy');
+            expect(normalizeTargetUrl('http://127.0.0.1:3112')).toBe('http://127.0.0.1:3112');
+        });
+        it('should leave other hostnames unchanged', () => {
+            const { normalizeTargetUrl } = require('../editorProxy');
+            expect(normalizeTargetUrl('http://myhost:3112')).toBe('http://myhost:3112');
+        });
+        it('should strip trailing slash', () => {
+            const { normalizeTargetUrl } = require('../editorProxy');
+            expect(normalizeTargetUrl('http://localhost:3112/')).toBe('http://localhost:3112');
+        });
+    });
+    describe('resolveLoopbackAddress', () => {
+        let server;
+        afterEach((done) => {
+            if (server?.listening) {
+                server.close(done);
+            }
+            else {
+                done();
+            }
+        });
+        it('should resolve to 127.0.0.1 when server binds to IPv4', async () => {
+            const { resolveLoopbackAddress } = require('../editorProxy');
+            server = http.createServer((_req, res) => res.end('ok'));
+            await new Promise((resolve) => server.listen(0, '127.0.0.1', () => resolve()));
+            const port = server.address().port;
+            const host = await resolveLoopbackAddress(port);
+            expect(host).toBe('127.0.0.1');
+        });
+        it('should resolve to ::1 when server binds to IPv6 only', async () => {
+            const { resolveLoopbackAddress } = require('../editorProxy');
+            server = http.createServer((_req, res) => res.end('ok'));
+            await new Promise((resolve) => server.listen(0, '::1', () => resolve()));
+            const port = server.address().port;
+            const host = await resolveLoopbackAddress(port);
+            expect(host).toBe('::1');
+        });
+        it('should return null when no server is listening', async () => {
+            const { resolveLoopbackAddress } = require('../editorProxy');
+            const host = await resolveLoopbackAddress(19876);
+            expect(host).toBeNull();
+        });
+    });
+    describe('proxy forwarding uses 127.0.0.1', () => {
+        let targetServer;
+        let targetPort;
+        afterEach((done) => {
+            if (targetServer?.listening) {
+                targetServer.close(done);
+            }
+            else {
+                done();
+            }
+        });
+        it('should forward requests to target and return correct response', async () => {
+            const { startEditorProxy, stopEditorProxy } = require('../editorProxy');
+            // Start a target server that returns 200
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('OK from target');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            // Start proxy pointing to the target using localhost URL
+            const result = await startEditorProxy({
+                port: 0, // let OS pick
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            // Make a request through the proxy
+            const response = await fetch(`http://127.0.0.1:${result.port}/`);
+            expect(response.status).toBe(200);
+            const body = await response.text();
+            expect(body).toBe('OK from target');
+            await stopEditorProxy();
+        });
+        it('should log when target returns non-2xx status', async () => {
+            const { startEditorProxy, stopEditorProxy } = require('../editorProxy');
+            // Start a target that returns 404
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(404, { 'Content-Type': 'text/plain' });
+                res.end('Not Found');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const warnSpy = jest.spyOn(console, 'warn').mockImplementation();
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            const response = await fetch(`http://127.0.0.1:${result.port}/test-path`);
+            expect(response.status).toBe(404);
+            // Verify a warning was logged about the non-2xx response
+            expect(warnSpy.mock.calls.some((call) => String(call[0]).includes('[editorProxy]') &&
+                String(call[0]).includes('404') &&
+                String(call[0]).includes('/test-path'))).toBe(true);
+            warnSpy.mockRestore();
+            await stopEditorProxy();
+        });
+        it('should verify forwarding works after start via self-test', async () => {
+            const { startEditorProxy, stopEditorProxy, verifyProxyForwarding, } = require('../editorProxy');
+            // Start a target that returns 200
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('healthy');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            const verified = await verifyProxyForwarding();
+            expect(verified).toBe(true);
+            await stopEditorProxy();
+        });
+        it('should report verification failure when target is down', async () => {
+            const { startEditorProxy, stopEditorProxy, verifyProxyForwarding, } = require('../editorProxy');
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: 'http://127.0.0.1:19876',
+            });
+            const verified = await verifyProxyForwarding();
+            expect(verified).toBe(false);
+            await stopEditorProxy();
+        });
+        it('should return false for verification when proxy is not running', async () => {
+            const { verifyProxyForwarding, stopEditorProxy, } = require('../editorProxy');
+            await stopEditorProxy();
+            const verified = await verifyProxyForwarding();
+            expect(verified).toBe(false);
+        });
+        it('should return 502 when target is unreachable', async () => {
+            const { startEditorProxy, stopEditorProxy } = require('../editorProxy');
+            // Start proxy pointing to a port with nothing listening
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: 'http://localhost:19876',
+            });
+            expect(result).not.toBeNull();
+            const response = await fetch(`http://127.0.0.1:${result.port}/`);
+            expect(response.status).toBe(502);
+            await stopEditorProxy();
+        });
+        it('should forward requests to an IPv6-only target server', async () => {
+            const { startEditorProxy, stopEditorProxy } = require('../editorProxy');
+            // Start a target server bound to IPv6 only (::1) — this is what Vite 6 does on macOS
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('OK from IPv6 target');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '::1', () => resolve()));
+            targetPort = targetServer.address().port;
+            // Start proxy with a localhost URL (simulating Vite's "Local: http://localhost:PORT")
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            // The proxy should successfully forward to the IPv6-only target
+            const response = await fetch(`http://127.0.0.1:${result.port}/`);
+            expect(response.status).toBe(200);
+            const body = await response.text();
+            expect(body).toBe('OK from IPv6 target');
+            await stopEditorProxy();
+        });
+    });
+    describe('injectHealthScript', () => {
+        it('should inject before </head> when present', () => {
+            const html = '<html><head><title>Test</title></head><body>Hello</body></html>';
+            const result = injectHealthScript(html);
+            expect(result).toContain(PREVIEW_HEALTH_SCRIPT + '</head>');
+            expect(result).toContain('<title>Test</title>');
+        });
+        it('should inject before </body> when no </head>', () => {
+            const html = '<html><body>Hello</body></html>';
+            const result = injectHealthScript(html);
+            expect(result).toContain(PREVIEW_HEALTH_SCRIPT + '</body>');
+        });
+        it('should append when no </head> or </body>', () => {
+            const html = '<div>Hello</div>';
+            const result = injectHealthScript(html);
+            expect(result).toBe('<div>Hello</div>' + PREVIEW_HEALTH_SCRIPT);
+        });
+        it('should include the data-codeyam-health attribute', () => {
+            const html = '<html><head></head><body></body></html>';
+            const result = injectHealthScript(html);
+            expect(result).toContain('data-codeyam-health');
+        });
+    });
+    describe('preview health endpoint', () => {
+        let targetServer;
+        let targetPort;
+        afterEach(async () => {
+            const { stopEditorProxy } = require('../editorProxy');
+            await stopEditorProxy();
+            if (targetServer?.listening) {
+                await new Promise((resolve) => targetServer.close(() => resolve()));
+            }
+            resetPreviewHealth();
+        });
+        it('should intercept POST /__codeyam__/preview-health and store errors', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            // Start a target that returns 200
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('ok');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            // Post error data to the health endpoint
+            const response = await fetch(`http://127.0.0.1:${result.port}/__codeyam__/preview-health`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    errors: [
+                        {
+                            type: 'error',
+                            message: 'Cannot read properties of undefined',
+                            timestamp: 1710000000,
+                        },
+                    ],
+                    url: 'http://localhost:3112/',
+                }),
+            });
+            expect(response.status).toBe(204);
+            // Check the stored health data
+            const health = getPreviewHealthReport();
+            expect(health).not.toBeNull();
+            expect(health.errors).toHaveLength(1);
+            expect(health.errors[0].message).toBe('Cannot read properties of undefined');
+            expect(health.url).toBe('http://localhost:3112/');
+        });
+        it('should store loaded/hasContent from load report', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('ok');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            // Post load report
+            await fetch(`http://127.0.0.1:${result.port}/__codeyam__/preview-health`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    loaded: true,
+                    hasContent: true,
+                    url: 'http://localhost:3112/',
+                    errorCount: 0,
+                }),
+            });
+            const health = getPreviewHealthReport();
+            expect(health).not.toBeNull();
+            expect(health.loaded).toBe(true);
+            expect(health.hasContent).toBe(true);
+        });
+        it('should inject health script into HTML responses', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            // Target that returns HTML
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/html' });
+                res.end('<html><head><title>App</title></head><body><div>Hello</div></body></html>');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            const response = await fetch(`http://127.0.0.1:${result.port}/`);
+            const body = await response.text();
+            expect(body).toContain('data-codeyam-health');
+            expect(body).toContain('/__codeyam__/preview-health');
+            expect(body).toContain('<title>App</title>');
+            expect(body).toContain('<div>Hello</div>');
+        });
+        it('should NOT inject health script into non-HTML responses', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end('{"data": "test"}');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            const response = await fetch(`http://127.0.0.1:${result.port}/api/data`);
+            const body = await response.text();
+            expect(body).toBe('{"data": "test"}');
+            expect(body).not.toContain('data-codeyam-health');
+        });
+        it('should accumulate errors across multiple health reports', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                res.end('ok');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            // First error report
+            await fetch(`http://127.0.0.1:${result.port}/__codeyam__/preview-health`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    errors: [{ type: 'error', message: 'Error 1', timestamp: 1 }],
+                    url: 'http://localhost:3112/',
+                }),
+            });
+            // Second error report
+            await fetch(`http://127.0.0.1:${result.port}/__codeyam__/preview-health`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    errors: [
+                        { type: 'console.error', message: 'Error 2', timestamp: 2 },
+                    ],
+                    url: 'http://localhost:3112/',
+                }),
+            });
+            const health = getPreviewHealthReport();
+            expect(health).not.toBeNull();
+            expect(health.errors).toHaveLength(2);
+            expect(health.errors[0].message).toBe('Error 1');
+            expect(health.errors[1].message).toBe('Error 2');
+        });
+    });
+    describe('buildLocalStorageScript', () => {
+        it('should return empty string when no localStorage config', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            expect(buildLocalStorageScript(null, 'scenario-123')).toBe('');
+            expect(buildLocalStorageScript(undefined, 'scenario-123')).toBe('');
+        });
+        it('should emit cleanup script when localStorage config is empty object', () => {
+            // BUG FIX: When switching from a scenario with localStorage data to one
+            // with empty localStorage (e.g. "First Launch - Welcome"), we must still
+            // clean up the previous scenario's keys. Otherwise the app reads stale data.
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const script = buildLocalStorageScript({}, 'scenario-123');
+            expect(script).not.toBe('');
+            // Should clean up previous keys
+            expect(script).toContain('localStorage.removeItem');
+            expect(script).toContain('__codeyam_ls_keys__');
+            expect(script).toContain('__codeyam_ls_sid__');
+            // Should NOT set any new keys (empty config)
+            expect(script).not.toContain('localStorage.setItem(\"margo');
+        });
+        it('should generate a script that sets localStorage entries', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const config = {
+                margo_articles: [{ id: 'a1', title: 'Test Article' }],
+                margo_collections: [{ id: 'c1', name: 'Test Collection' }],
+            };
+            const script = buildLocalStorageScript(config, 'scenario-abc');
+            // Should be wrapped in a script tag
+            expect(script).toMatch(/^<script data-codeyam-ls>/);
+            expect(script).toMatch(/<\/script>$/);
+            // Should gate on scenario ID to avoid re-seeding on reload
+            expect(script).toContain('scenario-abc');
+            expect(script).toContain('__codeyam_ls_sid__');
+            // Should set the localStorage entries
+            expect(script).toContain('margo_articles');
+            expect(script).toContain('margo_collections');
+            expect(script).toContain('localStorage.setItem');
+            // Should track which keys were set for cleanup
+            expect(script).toContain('__codeyam_ls_keys__');
+        });
+        it('should emit gated localStorage.clear() when prototypeId is provided and config is null', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const script = buildLocalStorageScript(null, 'scenario-123', 'proto-abc');
+            // Should not be empty — prototypeId triggers a clear script
+            expect(script).not.toBe('');
+            expect(script).toContain('localStorage.clear()');
+            // Should be gated by the __codeyam_proto__ marker
+            expect(script).toContain('__codeyam_proto__');
+            expect(script).toContain('proto-abc');
+        });
+        it('should not emit clear script when prototypeId is absent and config is null', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            // No prototypeId — same as before
+            expect(buildLocalStorageScript(null, 'scenario-123')).toBe('');
+            expect(buildLocalStorageScript(null, 'scenario-123', null)).toBe('');
+            expect(buildLocalStorageScript(undefined, 'scenario-123', null)).toBe('');
+        });
+        it('should clean up keys from previous scenario before setting new ones', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const config = { my_key: 'my_value' };
+            const script = buildLocalStorageScript(config, 'scenario-xyz');
+            // Should read previous keys and remove them
+            expect(script).toContain('__codeyam_ls_keys__');
+            expect(script).toContain('localStorage.removeItem');
+        });
+        it('should JSON.stringify non-string values', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const config = {
+                string_key: 'plain string',
+                object_key: { nested: true },
+                array_key: [1, 2, 3],
+            };
+            const script = buildLocalStorageScript(config, 'scenario-1');
+            // Object/array values should be JSON-stringified and embedded as JS string literals.
+            // JSON.stringify is applied twice: once to serialize the value, once to make it a safe JS string.
+            // So {"nested":true} becomes "{\"nested\":true}" in the script source.
+            expect(script).toContain('object_key');
+            expect(script).toContain('nested');
+            expect(script).toContain('array_key');
+            expect(script).toContain('[1,2,3]');
+        });
+        it('should include localStorage mutation watcher that sends postMessage on changes', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const config = { my_key: 'my_value' };
+            const script = buildLocalStorageScript(config, 'scenario-1');
+            // Should hook setItem, removeItem, and clear
+            expect(script).toContain('__codeyam_orig_setItem__');
+            expect(script).toContain('__codeyam_orig_removeItem__');
+            expect(script).toContain('__codeyam_orig_clear__');
+            // Should send postMessage to parent
+            expect(script).toContain('codeyam-localstorage-changed');
+            expect(script).toContain('postMessage');
+            // Should filter out internal codeyam keys
+            expect(script).toContain('__codeyam_');
+        });
+        it('should include localStorage mutation watcher even for empty localStorage config', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const script = buildLocalStorageScript({}, 'scenario-1');
+            // Empty config still needs the watcher since the app may write to localStorage
+            expect(script).toContain('codeyam-localstorage-changed');
+        });
+        it('should include a listener for codeyam-get-localstorage requests', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const config = { my_key: 'my_value' };
+            const script = buildLocalStorageScript(config, 'scenario-1');
+            // Should listen for get-localstorage requests from parent
+            expect(script).toContain('codeyam-get-localstorage');
+            expect(script).toContain('codeyam-localstorage-state');
+        });
+        it('should not include mutation watcher when no localStorage config and no prototypeId', () => {
+            const { buildLocalStorageScript } = require('../editorProxy');
+            // No config at all — returns empty string, no watcher needed
+            expect(buildLocalStorageScript(null, 'scenario-1')).toBe('');
+        });
+        it('should re-seed localStorage when scenario data changes even if ID is the same', () => {
+            // BUG: When a scenario is re-registered with updated data (e.g., adding
+            // collections to articles), the scenario ID stays the same. The guard
+            // checked only the ID, so the browser skipped re-seeding and showed stale
+            // data. The screenshot (captured by Playwright with a fresh browser)
+            // showed the new data, but the live preview didn't.
+            const { buildLocalStorageScript } = require('../editorProxy');
+            const originalData = {
+                articles: '[{"id":"a1","title":"Test"}]',
+            };
+            const updatedData = {
+                articles: '[{"id":"a1","title":"Test","collectionIds":["c1"]}]',
+                collections: '[{"id":"c1","name":"Dev"}]',
+            };
+            const script1 = buildLocalStorageScript(originalData, 'scenario-abc');
+            const script2 = buildLocalStorageScript(updatedData, 'scenario-abc');
+            // Both scripts use the same scenario ID
+            expect(script1).toContain('scenario-abc');
+            expect(script2).toContain('scenario-abc');
+            // The guard values must be DIFFERENT so the browser re-seeds
+            // Extract the guard comparison value from each script
+            const guardPattern = /__codeyam_ls_sid__.*?===\s*([^\)]+)\)/;
+            const guard1 = script1.match(guardPattern)?.[1];
+            const guard2 = script2.match(guardPattern)?.[1];
+            expect(guard1).toBeDefined();
+            expect(guard2).toBeDefined();
+            expect(guard1).not.toEqual(guard2);
+        });
+    });
+    describe('localStorage injection in HTML responses', () => {
+        it('should inject localStorage script before health script in HTML', () => {
+            const { buildLocalStorageScript, injectHealthScript, } = require('../editorProxy');
+            const lsScript = buildLocalStorageScript({ items: [1, 2] }, 'scenario-1');
+            // localStorage script should be injected before the health script
+            // (needs to run before app loads)
+            const html = '<html><head><title>Test</title></head><body>Hello</body></html>';
+            const result = injectHealthScript(html, lsScript);
+            expect(result).toContain('data-codeyam-ls');
+            expect(result).toContain('data-codeyam-health');
+            // localStorage script should appear BEFORE health script in the output
+            const lsIdx = result.indexOf('data-codeyam-ls');
+            const healthIdx = result.indexOf('data-codeyam-health');
+            expect(lsIdx).toBeLessThan(healthIdx);
+        });
+        it('should not inject localStorage script when none provided', () => {
+            const html = '<html><head><title>Test</title></head><body>Hello</body></html>';
+            const result = injectHealthScript(html);
+            expect(result).toContain('data-codeyam-health');
+            expect(result).not.toContain('data-codeyam-ls');
+        });
+    });
+    describe('compressed response handling', () => {
+        let targetServer;
+        let targetPort;
+        afterEach(async () => {
+            const { stopEditorProxy } = require('../editorProxy');
+            await stopEditorProxy();
+            if (targetServer?.listening) {
+                await new Promise((resolve) => targetServer.close(() => resolve()));
+            }
+            resetPreviewHealth();
+        });
+        it('should strip accept-encoding so target returns uncompressed HTML for injection', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            const originalHtml = '<html><head><title>Compressed</title></head><body><div>Gzipped</div></body></html>';
+            // Target server that compresses only when client sends Accept-Encoding
+            // (simulates real dev servers like Next.js/Vite)
+            targetServer = http.createServer((req, res) => {
+                const acceptEncoding = req.headers['accept-encoding'] || '';
+                if (acceptEncoding.includes('gzip')) {
+                    const compressed = zlib.gzipSync(Buffer.from(originalHtml));
+                    res.writeHead(200, {
+                        'Content-Type': 'text/html',
+                        'Content-Encoding': 'gzip',
+                        'Content-Length': compressed.length,
+                    });
+                    res.end(compressed);
+                }
+                else {
+                    res.writeHead(200, { 'Content-Type': 'text/html' });
+                    res.end(originalHtml);
+                }
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            // Request through proxy WITH Accept-Encoding (simulating a real browser)
+            const response = await fetch(`http://127.0.0.1:${result.port}/`, {
+                headers: { 'Accept-Encoding': 'gzip, deflate, br' },
+            });
+            // The proxy should return a valid, readable response
+            expect(response.status).toBe(200);
+            const body = await response.text();
+            // Should contain the original HTML content (not corrupted binary)
+            expect(body).toContain('<title>Compressed</title>');
+            expect(body).toContain('<div>Gzipped</div>');
+            // Should have injected the health script
+            expect(body).toContain('data-codeyam-health');
+            // Should NOT have Content-Encoding since the proxy serves uncompressed
+            expect(response.headers.get('content-encoding')).toBeNull();
+        });
+    });
+    describe('Cache-Control on HTML responses', () => {
+        let targetServer;
+        let targetPort;
+        afterEach(async () => {
+            const { stopEditorProxy } = require('../editorProxy');
+            await stopEditorProxy();
+            if (targetServer?.listening) {
+                await new Promise((resolve) => targetServer.close(() => resolve()));
+            }
+            resetPreviewHealth();
+        });
+        it('should add Cache-Control: no-store to proxied HTML responses', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            // Target server that serves plain HTML without cache headers
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'text/html' });
+                res.end('<html><head><title>Test</title></head><body>Hello</body></html>');
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            const response = await fetch(`http://127.0.0.1:${result.port}/`);
+            expect(response.status).toBe(200);
+            // Proxy must set no-store so the browser doesn't cache pages between
+            // scenario switches (application scenarios share the same proxy URL
+            // but differ in seed data).
+            expect(response.headers.get('cache-control')).toBe('no-store, must-revalidate');
+        });
+        it('should NOT add Cache-Control: no-store to non-HTML responses', async () => {
+            const { startEditorProxy } = require('../editorProxy');
+            // Target server that serves JSON without cache headers
+            targetServer = http.createServer((_req, res) => {
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({ ok: true }));
+            });
+            await new Promise((resolve) => targetServer.listen(0, '127.0.0.1', () => resolve()));
+            targetPort = targetServer.address().port;
+            const result = await startEditorProxy({
+                port: 0,
+                targetUrl: `http://localhost:${targetPort}`,
+            });
+            expect(result).not.toBeNull();
+            const response = await fetch(`http://127.0.0.1:${result.port}/api/data`);
+            expect(response.status).toBe(200);
+            // Non-HTML responses pass through the target's original headers;
+            // the proxy only forces no-store on HTML.
+            expect(response.headers.get('cache-control')).toBeNull();
+        });
+    });
+});
+//# sourceMappingURL=editorProxy.test.js.map