SnowLuma v2.4.2 source snapshot

packages/onebot/tests/http-server-websocket-config.test.ts

@@ -0,0 +1,103 @@
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import {
+    assertValidOneBotConfig,
+    loadOneBotConfig,
+    makeDefaultOneBotConfig,
+    prepareOneBotConfigForRestore,
+    saveOneBotConfig,
+} from '../src/config';
+
+describe.each(['enableWebSocket', 'enableSse'] as const)(
+    'HTTP server %s configuration',
+    (switchName) => {
+        let previousCwd: string;
+        let tempDir: string;
+
+        beforeEach(() => {
+            previousCwd = process.cwd();
+            tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'snowluma-http-ws-config-'));
+            process.chdir(tempDir);
+        });
+
+        afterEach(() => {
+            process.chdir(previousCwd);
+            fs.rmSync(tempDir, { recursive: true, force: true });
+        });
+
+        it('persists the opt-in switch through save and load', () => {
+            const config = makeDefaultOneBotConfig();
+            expect(config.networks.httpServers[0][switchName]).toBe(false);
+            config.networks.httpServers[0][switchName] = true;
+
+            saveOneBotConfig('10001', config);
+
+            const onDisk = JSON.parse(
+                fs.readFileSync(path.join(tempDir, 'config', 'onebot_10001.json'), 'utf8'),
+            ) as {
+                networks: {
+                    httpServers: Array<{ enableWebSocket?: boolean; enableSse?: boolean }>;
+                };
+            };
+            expect(onDisk.networks.httpServers[0][switchName]).toBe(true);
+            expect(loadOneBotConfig('10001').networks.httpServers[0][switchName]).toBe(true);
+        });
+
+        it('rejects non-boolean values at public save and restore boundaries', () => {
+            const config = makeDefaultOneBotConfig();
+            (config.networks.httpServers[0] as unknown as Record<string, unknown>)[switchName] =
+                'yes';
+            expect(() => assertValidOneBotConfig(config)).toThrow(
+                `${switchName} must be a boolean`,
+            );
+
+            expect(() =>
+                prepareOneBotConfigForRestore(
+                    {
+                        networks: {
+                            httpServers: [
+                                {
+                                    name: 'http',
+                                    host: '127.0.0.1',
+                                    port: 3000,
+                                    path: '/',
+                                    [switchName]: 'yes',
+                                },
+                            ],
+                        },
+                    },
+                    'per-uin',
+                ),
+            ).toThrow(`${switchName} must be a boolean`);
+        });
+
+        it('fails fast when a hand-written config contains a non-boolean switch', () => {
+            fs.mkdirSync(path.join(tempDir, 'config'), { recursive: true });
+            fs.writeFileSync(
+                path.join(tempDir, 'config', 'onebot_10002.json'),
+                JSON.stringify({
+                    mode: 'snapshot',
+                    networks: {
+                        httpServers: [
+                            {
+                                name: 'http',
+                                host: '127.0.0.1',
+                                port: 3000,
+                                path: '/',
+                                [switchName]: 'yes',
+                            },
+                        ],
+                        httpClients: [],
+                        wsServers: [],
+                        wsClients: [],
+                    },
+                }),
+                'utf8',
+            );
+
+            expect(() => loadOneBotConfig('10002')).toThrow(`${switchName} must be a boolean`);
+        });
+    },
+);