Bohdan Triapitsyn
0e65a435ee
Fixes switching and unlocking password-protected remote instances Stores SSH forwarded host client tokens from saved UI passwords Avoids unnecessary auth churn when no runtime headers are configured
71 lines
2.4 KiB
JavaScript
71 lines
2.4 KiB
JavaScript
import { afterEach, describe, expect, test } from 'bun:test';
|
|
import fs from 'node:fs';
|
|
import fsp from 'node:fs/promises';
|
|
import http from 'node:http';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
|
|
import { ElectronSshManager } from './ssh-manager.mjs';
|
|
|
|
const servers = [];
|
|
const tempDirs = [];
|
|
|
|
const listen = async (server) => {
|
|
await new Promise((resolve) => server.listen(0, '127.0.0.1', resolve));
|
|
servers.push(server);
|
|
const address = server.address();
|
|
if (!address || typeof address === 'string') throw new Error('Expected TCP server address');
|
|
return `http://127.0.0.1:${address.port}`;
|
|
};
|
|
|
|
const readBody = async (req) => {
|
|
let body = '';
|
|
for await (const chunk of req) body += chunk.toString();
|
|
return body;
|
|
};
|
|
|
|
afterEach(async () => {
|
|
while (servers.length > 0) {
|
|
const server = servers.pop();
|
|
await new Promise((resolve) => server.close(() => resolve()));
|
|
}
|
|
while (tempDirs.length > 0) {
|
|
await fsp.rm(tempDirs.pop(), { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
describe('ElectronSshManager', () => {
|
|
test('stores a client token for forwarded OpenChamber hosts when UI password is configured', async () => {
|
|
let loginPayload = null;
|
|
const server = http.createServer(async (req, res) => {
|
|
if (req.method === 'POST' && req.url === '/auth/session') {
|
|
loginPayload = JSON.parse(await readBody(req));
|
|
res.writeHead(200, { 'Content-Type': 'application/json' });
|
|
res.end(JSON.stringify({ authenticated: true, clientToken: 'ssh-client-token' }));
|
|
return;
|
|
}
|
|
res.writeHead(404).end();
|
|
});
|
|
const localUrl = await listen(server);
|
|
const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'openchamber-ssh-manager-test-'));
|
|
tempDirs.push(tempDir);
|
|
const settingsFilePath = path.join(tempDir, 'settings.json');
|
|
const manager = new ElectronSshManager({
|
|
settingsFilePath,
|
|
appVersion: '0.0.0-test',
|
|
emit: () => undefined,
|
|
});
|
|
|
|
const token = await manager.issueClientToken(localUrl, 'ui-secret');
|
|
await manager.updateHostRuntime('ssh-1', 'SSH Host', localUrl, token);
|
|
|
|
const settings = JSON.parse(fs.readFileSync(settingsFilePath, 'utf8'));
|
|
expect(loginPayload).toMatchObject({
|
|
password: 'ui-secret',
|
|
trustDevice: true,
|
|
issueClientToken: true,
|
|
});
|
|
expect(settings.desktopHosts).toEqual([{ id: 'ssh-1', label: 'SSH Host', url: localUrl, apiUrl: localUrl, clientToken: 'ssh-client-token' }]);
|
|
});
|
|
});
|