mirror of
https://github.com/UNITRONIX/BetterDesk.git
synced 2026-09-11 13:49:03 +00:00
UNITRONIX
43e0b6f40a
Expose global connection strategy in the web panel with systemd/Docker persistence, extend server health diagnostics, enforce org network policy in the signal handler, and document when relay fallback is expected vs misconfiguration.
62 lines
2.3 KiB
JavaScript
62 lines
2.3 KiB
JavaScript
'use strict';
|
|
|
|
const assert = require('assert');
|
|
const svc = require('../services/serverConnectionConfigService');
|
|
|
|
const sampleSystemd = `[Unit]
|
|
Description=BetterDesk Server
|
|
|
|
[Service]
|
|
Type=simple
|
|
ExecStart=/opt/betterdesk/betterdesk-server -mode all
|
|
Environment=P2P_FIRST=Y
|
|
Environment=ALWAYS_USE_RELAY=N
|
|
Restart=always
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
`;
|
|
|
|
const sampleCompose = `services:
|
|
server:
|
|
image: betterdesk-server:local
|
|
environment:
|
|
- ENCRYPTED_ONLY=1
|
|
- DB_URL=/opt/rustdesk/db_v2.sqlite3
|
|
- P2P_FIRST=Y
|
|
console:
|
|
environment:
|
|
- NODE_ENV=production
|
|
`;
|
|
|
|
assert.strictEqual(svc.modeFromEnvVars({ P2P_FIRST: 'Y', ALWAYS_USE_RELAY: 'N' }), 'p2p_first');
|
|
assert.strictEqual(svc.modeFromEnvVars({ P2P_FIRST: 'N', ALWAYS_USE_RELAY: 'Y' }), 'relay_only');
|
|
assert.strictEqual(svc.modeFromEnvVars({ ALWAYS_USE_RELAY: 'Y' }), 'relay_only');
|
|
|
|
const relayVars = svc.envVarsFromSettings({ mode: 'relay_only', p2p_fallback_ms: 3000, same_nat_relay: false });
|
|
assert.strictEqual(relayVars.P2P_FIRST, 'N');
|
|
assert.strictEqual(relayVars.ALWAYS_USE_RELAY, 'Y');
|
|
assert.strictEqual(relayVars.P2P_FALLBACK_MS, '3000');
|
|
assert.strictEqual(relayVars.SAME_NAT_RELAY, 'N');
|
|
|
|
const systemdEnv = svc.parseSystemdEnvironment(sampleSystemd);
|
|
assert.strictEqual(systemdEnv.P2P_FIRST, 'Y');
|
|
assert.strictEqual(systemdEnv.ALWAYS_USE_RELAY, 'N');
|
|
|
|
const patchedSystemd = svc.patchSystemdEnvironment(sampleSystemd, relayVars);
|
|
assert(patchedSystemd.includes('Environment=P2P_FIRST=N'), 'patched P2P_FIRST');
|
|
assert(patchedSystemd.includes('Environment=ALWAYS_USE_RELAY=Y'), 'patched ALWAYS_USE_RELAY');
|
|
assert(!patchedSystemd.match(/Environment=P2P_FIRST=Y/m), 'old P2P_FIRST removed');
|
|
|
|
const composeEnv = svc.parseDockerComposeEnvironment(sampleCompose);
|
|
assert.strictEqual(composeEnv.P2P_FIRST, 'Y');
|
|
assert.strictEqual(composeEnv.ALWAYS_USE_RELAY, undefined);
|
|
|
|
const patchedCompose = svc.patchDockerComposeEnvironment(sampleCompose, relayVars);
|
|
assert(patchedCompose.includes('- P2P_FIRST=N'), 'compose P2P_FIRST patched');
|
|
assert(patchedCompose.includes('- ALWAYS_USE_RELAY=Y'), 'compose ALWAYS_USE_RELAY added');
|
|
assert(patchedCompose.includes('- ENCRYPTED_ONLY=1'), 'other env preserved');
|
|
assert(!patchedCompose.includes('console:\n environment:\n - P2P_FIRST'), 'console env untouched');
|
|
|
|
console.log('serverConnectionConfig tests passed');
|