fix: replace stale agent tests with UnifiedAgents tests
DockerAgents.tsx and HostAgents.tsx were consolidated into UnifiedAgents.tsx, but the old test files remained and referenced non-existent components, breaking CI. - Remove DockerAgents.test.tsx and HostAgents.test.tsx - Add UnifiedAgents.test.tsx covering token generation, host lookup, managed agents table, and platform-specific install commands
This commit is contained in:
rcourtman
parent
e467523a61
commit
c110a3b803
3 changed files with 396 additions and 534 deletions
|
|
@ -1,218 +0,0 @@
|
||||||
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
||||||
import { render, fireEvent, screen, waitFor, cleanup } from '@solidjs/testing-library';
|
|
||||||
import { createStore } from 'solid-js/store';
|
|
||||||
import { DockerAgents } from '../DockerAgents';
|
|
||||||
import type { DockerHost, RemovedDockerHost } from '@/types/api';
|
|
||||||
|
|
||||||
let mockWsStore: {
|
|
||||||
state: { dockerHosts: DockerHost[]; removedDockerHosts: RemovedDockerHost[] };
|
|
||||||
connected: () => boolean;
|
|
||||||
reconnecting: () => boolean;
|
|
||||||
activeAlerts: unknown[];
|
|
||||||
};
|
|
||||||
|
|
||||||
const allowDockerHostReenrollMock = vi.fn();
|
|
||||||
const deleteDockerHostMock = vi.fn();
|
|
||||||
const unhideDockerHostMock = vi.fn();
|
|
||||||
const notificationSuccessMock = vi.fn();
|
|
||||||
const notificationErrorMock = vi.fn();
|
|
||||||
const fetchMock = vi.fn();
|
|
||||||
|
|
||||||
vi.mock('@/App', () => ({
|
|
||||||
useWebSocket: () => mockWsStore,
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/api/monitoring', () => ({
|
|
||||||
MonitoringAPI: {
|
|
||||||
allowDockerHostReenroll: (...args: unknown[]) => allowDockerHostReenrollMock(...args),
|
|
||||||
deleteDockerHost: (...args: unknown[]) => deleteDockerHostMock(...args),
|
|
||||||
unhideDockerHost: (...args: unknown[]) => unhideDockerHostMock(...args),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/api/security', () => ({
|
|
||||||
SecurityAPI: {
|
|
||||||
createToken: vi.fn(),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/stores/notifications', () => ({
|
|
||||||
notificationStore: {
|
|
||||||
success: (...args: unknown[]) => notificationSuccessMock(...args),
|
|
||||||
error: (...args: unknown[]) => notificationErrorMock(...args),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
const createDockerHost = (overrides?: Partial<DockerHost>): DockerHost => ({
|
|
||||||
id: 'host-1',
|
|
||||||
agentId: 'agent-1',
|
|
||||||
hostname: 'host-1.local',
|
|
||||||
displayName: 'Host One',
|
|
||||||
cpus: 4,
|
|
||||||
totalMemoryBytes: 8 * 1024 * 1024 * 1024,
|
|
||||||
uptimeSeconds: 12_345,
|
|
||||||
status: 'online',
|
|
||||||
lastSeen: Date.now(),
|
|
||||||
intervalSeconds: 30,
|
|
||||||
containers: [],
|
|
||||||
...overrides,
|
|
||||||
});
|
|
||||||
|
|
||||||
const createRemovedHost = (overrides?: Partial<RemovedDockerHost>): RemovedDockerHost => ({
|
|
||||||
id: 'host-removed',
|
|
||||||
hostname: 'retired-node.local',
|
|
||||||
displayName: 'Retired Node',
|
|
||||||
removedAt: Date.now() - 60_000,
|
|
||||||
...overrides,
|
|
||||||
});
|
|
||||||
|
|
||||||
const setupComponent = (hosts: DockerHost[], removedHosts: RemovedDockerHost[] = []) => {
|
|
||||||
const [state] = createStore({
|
|
||||||
dockerHosts: hosts,
|
|
||||||
removedDockerHosts: removedHosts,
|
|
||||||
});
|
|
||||||
|
|
||||||
mockWsStore = {
|
|
||||||
state,
|
|
||||||
connected: () => true,
|
|
||||||
reconnecting: () => false,
|
|
||||||
activeAlerts: [],
|
|
||||||
};
|
|
||||||
|
|
||||||
return render(() => <DockerAgents />);
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
allowDockerHostReenrollMock.mockReset();
|
|
||||||
deleteDockerHostMock.mockReset();
|
|
||||||
unhideDockerHostMock.mockReset();
|
|
||||||
notificationSuccessMock.mockReset();
|
|
||||||
notificationErrorMock.mockReset();
|
|
||||||
fetchMock.mockReset();
|
|
||||||
fetchMock.mockResolvedValue(
|
|
||||||
new Response(JSON.stringify({ requiresAuth: true, apiTokenConfigured: false }), {
|
|
||||||
status: 200,
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
vi.stubGlobal('fetch', fetchMock);
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
cleanup();
|
|
||||||
vi.unstubAllGlobals();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('DockerAgents removed hosts', () => {
|
|
||||||
it('renders removed hosts card and triggers allow/copy actions', async () => {
|
|
||||||
allowDockerHostReenrollMock.mockResolvedValue(undefined);
|
|
||||||
const clipboardSpy = vi.fn().mockResolvedValue(undefined);
|
|
||||||
|
|
||||||
vi.stubGlobal('navigator', { clipboard: { writeText: clipboardSpy } } as unknown as Navigator);
|
|
||||||
|
|
||||||
setupComponent([], [createRemovedHost()]);
|
|
||||||
|
|
||||||
expect(screen.getByText('Recently removed container hosts')).toBeInTheDocument();
|
|
||||||
const allowButton = screen.getByRole('button', { name: 'Allow re-enroll' });
|
|
||||||
fireEvent.click(allowButton);
|
|
||||||
|
|
||||||
await waitFor(() => expect(allowDockerHostReenrollMock).toHaveBeenCalledWith('host-removed'), { interval: 0 });
|
|
||||||
|
|
||||||
const copyButton = screen.getByRole('button', { name: 'Copy curl command' });
|
|
||||||
fireEvent.click(copyButton);
|
|
||||||
|
|
||||||
await waitFor(() => expect(clipboardSpy).toHaveBeenCalledTimes(1), { interval: 0 });
|
|
||||||
const copiedCommand = clipboardSpy.mock.calls.at(-1)?.[0];
|
|
||||||
expect(typeof copiedCommand).toBe('string');
|
|
||||||
expect(copiedCommand).toContain('/api/agents/docker/hosts/host-removed/allow-reenroll');
|
|
||||||
|
|
||||||
expect(notificationSuccessMock).toHaveBeenCalled();
|
|
||||||
expect(notificationErrorMock).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('shows progress stepper for an in-progress removal command', async () => {
|
|
||||||
const now = Date.now();
|
|
||||||
const host = createDockerHost({
|
|
||||||
command: {
|
|
||||||
id: 'cmd-1',
|
|
||||||
type: 'stop',
|
|
||||||
status: 'acknowledged',
|
|
||||||
createdAt: now - 60_000,
|
|
||||||
updatedAt: now - 20_000,
|
|
||||||
dispatchedAt: now - 40_000,
|
|
||||||
acknowledgedAt: now - 10_000,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const removeButton = screen.getByRole('button', { name: 'Remove' });
|
|
||||||
fireEvent.click(removeButton);
|
|
||||||
|
|
||||||
await screen.findByText('Remove container host "Host One"');
|
|
||||||
expect(screen.getByText('acknowledged')).toBeInTheDocument();
|
|
||||||
|
|
||||||
const progressHeading = screen.getByText('Progress');
|
|
||||||
const progressCard = progressHeading.parentElement?.parentElement;
|
|
||||||
if (!progressCard) {
|
|
||||||
throw new Error('Progress card not found');
|
|
||||||
}
|
|
||||||
|
|
||||||
const steps = Array.from(progressCard.querySelectorAll('li'));
|
|
||||||
expect(steps).toHaveLength(4);
|
|
||||||
|
|
||||||
expect(steps[0]).toHaveTextContent('Stop command queued');
|
|
||||||
expect(steps[1]).toHaveTextContent('Instruction delivered to the agent');
|
|
||||||
expect(steps[2]).toHaveTextContent('Agent acknowledged the stop request');
|
|
||||||
expect(steps[3]).toHaveTextContent('Agent disabled the service and removed autostart');
|
|
||||||
|
|
||||||
const indicators = steps.map((step) => step.querySelector('span'));
|
|
||||||
expect(indicators[0]?.className).toContain('bg-blue-500');
|
|
||||||
expect(indicators[1]?.className).toContain('bg-blue-500');
|
|
||||||
expect(indicators[2]?.className).toContain('animate-pulse');
|
|
||||||
expect(indicators[3]?.className).toContain('bg-gray-300');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('shows waiting message once the agent has already stopped', async () => {
|
|
||||||
const host = createDockerHost({
|
|
||||||
status: 'online',
|
|
||||||
pendingUninstall: true,
|
|
||||||
command: undefined,
|
|
||||||
lastSeen: Date.now() - 45_000,
|
|
||||||
});
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const viewButton = await screen.findByRole('button', { name: 'View progress' });
|
|
||||||
fireEvent.click(viewButton);
|
|
||||||
|
|
||||||
await screen.findByText('Pulse is waiting for', { exact: false });
|
|
||||||
|
|
||||||
const stopButton = screen.getByRole('button', { name: 'Waiting for host…' });
|
|
||||||
expect(stopButton).toBeDisabled();
|
|
||||||
expect(screen.queryByText('Progress')).not.toBeInTheDocument();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('shows confirmation while waiting for the agent to acknowledge the stop command', async () => {
|
|
||||||
deleteDockerHostMock.mockResolvedValue({});
|
|
||||||
const host = createDockerHost();
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const removeButton = screen.getByRole('button', { name: 'Remove' });
|
|
||||||
fireEvent.click(removeButton);
|
|
||||||
|
|
||||||
await screen.findByText('Remove container host "Host One"');
|
|
||||||
|
|
||||||
const stopButton = screen.getByRole('button', { name: 'Stop agent now' });
|
|
||||||
fireEvent.click(stopButton);
|
|
||||||
|
|
||||||
await waitFor(() => expect(deleteDockerHostMock).toHaveBeenCalledWith('host-1'), { interval: 0 });
|
|
||||||
|
|
||||||
const waitingButton = await screen.findByRole('button', { name: 'Waiting for agent…' });
|
|
||||||
expect(waitingButton).toBeDisabled();
|
|
||||||
|
|
||||||
expect(screen.getByText('Stop command sent.')).toBeInTheDocument();
|
|
||||||
expect(notificationSuccessMock).toHaveBeenCalledWith('Stop command sent to Host One', 3500);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
@ -1,316 +0,0 @@
|
||||||
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
||||||
import { render, fireEvent, screen, waitFor, cleanup } from '@solidjs/testing-library';
|
|
||||||
import { createStore } from 'solid-js/store';
|
|
||||||
import { HostAgents } from '../HostAgents';
|
|
||||||
import type { Host } from '@/types/api';
|
|
||||||
|
|
||||||
let mockWsStore: {
|
|
||||||
state: { hosts: Host[]; connectionHealth?: Record<string, boolean> };
|
|
||||||
connected: () => boolean;
|
|
||||||
reconnecting: () => boolean;
|
|
||||||
activeAlerts: unknown[];
|
|
||||||
};
|
|
||||||
|
|
||||||
const lookupMock = vi.fn();
|
|
||||||
const createTokenMock = vi.fn();
|
|
||||||
const deleteHostAgentMock = vi.fn();
|
|
||||||
const notificationSuccessMock = vi.fn();
|
|
||||||
const notificationErrorMock = vi.fn();
|
|
||||||
const notificationInfoMock = vi.fn();
|
|
||||||
const clipboardSpy = vi.fn();
|
|
||||||
|
|
||||||
vi.mock('@/App', () => ({
|
|
||||||
useWebSocket: () => mockWsStore,
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/api/monitoring', () => ({
|
|
||||||
MonitoringAPI: {
|
|
||||||
lookupHost: (...args: unknown[]) => lookupMock(...args),
|
|
||||||
deleteHostAgent: (...args: unknown[]) => deleteHostAgentMock(...args),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/api/security', () => ({
|
|
||||||
SecurityAPI: {
|
|
||||||
createToken: (...args: unknown[]) => createTokenMock(...args),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
vi.mock('@/stores/notifications', () => ({
|
|
||||||
notificationStore: {
|
|
||||||
success: (...args: unknown[]) => notificationSuccessMock(...args),
|
|
||||||
error: (...args: unknown[]) => notificationErrorMock(...args),
|
|
||||||
info: (...args: unknown[]) => notificationInfoMock(...args),
|
|
||||||
},
|
|
||||||
}));
|
|
||||||
|
|
||||||
const createHost = (overrides?: Partial<Host>): Host => ({
|
|
||||||
id: 'host-1',
|
|
||||||
hostname: 'host-1.local',
|
|
||||||
displayName: 'Host One',
|
|
||||||
platform: 'linux',
|
|
||||||
osName: 'Ubuntu',
|
|
||||||
osVersion: '22.04',
|
|
||||||
kernelVersion: '6.5.0',
|
|
||||||
architecture: 'x86_64',
|
|
||||||
cpuCount: 4,
|
|
||||||
cpuUsage: 12.5,
|
|
||||||
memory: {
|
|
||||||
total: 8 * 1024 * 1024 * 1024,
|
|
||||||
used: 4 * 1024 * 1024 * 1024,
|
|
||||||
free: 4 * 1024 * 1024 * 1024,
|
|
||||||
usage: 50,
|
|
||||||
balloon: 0,
|
|
||||||
swapUsed: 0,
|
|
||||||
swapTotal: 0,
|
|
||||||
},
|
|
||||||
loadAverage: [],
|
|
||||||
disks: [],
|
|
||||||
networkInterfaces: [],
|
|
||||||
sensors: {
|
|
||||||
temperatureCelsius: {},
|
|
||||||
fanRpm: {},
|
|
||||||
additional: {},
|
|
||||||
},
|
|
||||||
status: 'online',
|
|
||||||
uptimeSeconds: 12_345,
|
|
||||||
intervalSeconds: 30,
|
|
||||||
lastSeen: Date.now(),
|
|
||||||
agentVersion: '1.2.3',
|
|
||||||
tags: ['prod'],
|
|
||||||
...overrides,
|
|
||||||
});
|
|
||||||
|
|
||||||
const stubFetchSuccess = vi.fn();
|
|
||||||
|
|
||||||
const setupComponent = (hosts: Host[]) => {
|
|
||||||
const [state] = createStore({
|
|
||||||
hosts,
|
|
||||||
connectionHealth: {},
|
|
||||||
});
|
|
||||||
|
|
||||||
mockWsStore = {
|
|
||||||
state,
|
|
||||||
connected: () => true,
|
|
||||||
reconnecting: () => false,
|
|
||||||
activeAlerts: [],
|
|
||||||
};
|
|
||||||
|
|
||||||
return render(() => <HostAgents />);
|
|
||||||
};
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
lookupMock.mockReset();
|
|
||||||
createTokenMock.mockReset();
|
|
||||||
deleteHostAgentMock.mockReset();
|
|
||||||
notificationSuccessMock.mockReset();
|
|
||||||
notificationErrorMock.mockReset();
|
|
||||||
notificationInfoMock.mockReset();
|
|
||||||
stubFetchSuccess.mockImplementation(
|
|
||||||
async () =>
|
|
||||||
new Response(JSON.stringify({ requiresAuth: true, apiTokenConfigured: false }), {
|
|
||||||
status: 200,
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
vi.stubGlobal('fetch', stubFetchSuccess);
|
|
||||||
clipboardSpy.mockReset();
|
|
||||||
vi.stubGlobal('navigator', { clipboard: { writeText: clipboardSpy } } as unknown as Navigator);
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
cleanup();
|
|
||||||
vi.unstubAllGlobals();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('HostAgents lookup flow', () => {
|
|
||||||
it('highlights a host after a successful lookup and clears highlight after timeout', async () => {
|
|
||||||
const host = createHost();
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
createTokenMock.mockResolvedValue({
|
|
||||||
token: 'token-123',
|
|
||||||
record: {
|
|
||||||
id: 'token-record',
|
|
||||||
name: 'Test Token',
|
|
||||||
prefix: 'abcdef',
|
|
||||||
suffix: '1234',
|
|
||||||
createdAt: new Date().toISOString(),
|
|
||||||
},
|
|
||||||
});
|
|
||||||
const generateButton = screen.getByRole('button', { name: 'Generate token' });
|
|
||||||
fireEvent.click(generateButton);
|
|
||||||
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
|
||||||
|
|
||||||
await waitFor(() => expect(screen.getByRole('button', { name: 'Check status' })).toBeEnabled(), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
|
|
||||||
lookupMock.mockResolvedValue({
|
|
||||||
success: true,
|
|
||||||
host: {
|
|
||||||
id: host.id,
|
|
||||||
hostname: host.hostname,
|
|
||||||
displayName: host.displayName,
|
|
||||||
status: host.status,
|
|
||||||
connected: true,
|
|
||||||
lastSeen: host.lastSeen,
|
|
||||||
agentVersion: host.agentVersion,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const input = screen.getByPlaceholderText('Hostname or host ID') as HTMLInputElement;
|
|
||||||
fireEvent.input(input, { target: { value: host.id } });
|
|
||||||
|
|
||||||
const button = screen.getByRole('button', { name: 'Check status' });
|
|
||||||
fireEvent.click(button);
|
|
||||||
|
|
||||||
await waitFor(() => expect(lookupMock).toHaveBeenCalled(), { interval: 0 });
|
|
||||||
const [lookupArgs] = lookupMock.mock.calls.at(-1) ?? [];
|
|
||||||
expect(lookupArgs).toEqual({ id: host.id, hostname: host.id });
|
|
||||||
|
|
||||||
await waitFor(
|
|
||||||
() =>
|
|
||||||
expect(
|
|
||||||
screen.getByText('Connected', { selector: 'span' }),
|
|
||||||
).toBeInTheDocument(),
|
|
||||||
{ interval: 0 },
|
|
||||||
);
|
|
||||||
const statusBadges = screen.getAllByText('online', { selector: 'span' });
|
|
||||||
expect(statusBadges.length).toBeGreaterThan(0);
|
|
||||||
expect(screen.getByText('Agent version 1.2.3')).toBeInTheDocument();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('shows an error when lookup returns no host and does not highlight rows', async () => {
|
|
||||||
const host = createHost();
|
|
||||||
const { container } = setupComponent([host]);
|
|
||||||
|
|
||||||
createTokenMock.mockResolvedValue({
|
|
||||||
token: 'token-456',
|
|
||||||
record: {
|
|
||||||
id: 'token-record-2',
|
|
||||||
name: 'Test Token 2',
|
|
||||||
prefix: 'ghijkl',
|
|
||||||
suffix: '5678',
|
|
||||||
createdAt: new Date().toISOString(),
|
|
||||||
},
|
|
||||||
});
|
|
||||||
const generateButton = screen.getByRole('button', { name: 'Generate token' });
|
|
||||||
fireEvent.click(generateButton);
|
|
||||||
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
|
||||||
|
|
||||||
await waitFor(() => expect(screen.getByRole('button', { name: 'Check status' })).toBeEnabled(), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
|
|
||||||
lookupMock.mockResolvedValue(null);
|
|
||||||
|
|
||||||
const query = 'missing-host';
|
|
||||||
const input = screen.getByPlaceholderText('Hostname or host ID') as HTMLInputElement;
|
|
||||||
fireEvent.input(input, { target: { value: query } });
|
|
||||||
|
|
||||||
const button = screen.getByRole('button', { name: 'Check status' });
|
|
||||||
fireEvent.click(button);
|
|
||||||
|
|
||||||
await waitFor(
|
|
||||||
() =>
|
|
||||||
expect(
|
|
||||||
screen.getByText(`No host has reported with "${query}" yet. Try again in a few seconds.`),
|
|
||||||
).toBeInTheDocument(),
|
|
||||||
{ interval: 0 },
|
|
||||||
);
|
|
||||||
|
|
||||||
const row = container.querySelector(`tr[data-host-id="${host.id}"]`) as HTMLTableRowElement;
|
|
||||||
expect(row.classList.contains('ring-2')).toBe(false);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('Host removal modal', () => {
|
|
||||||
it('removes a host while it is still reporting and explains the impact', async () => {
|
|
||||||
deleteHostAgentMock.mockResolvedValue(undefined);
|
|
||||||
const host = createHost({
|
|
||||||
lastSeen: Date.now(),
|
|
||||||
status: 'online',
|
|
||||||
});
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const removeButton = screen.getByRole('button', { name: 'Remove' });
|
|
||||||
fireEvent.click(removeButton);
|
|
||||||
|
|
||||||
await screen.findByText('Remove host "Host One"');
|
|
||||||
|
|
||||||
const copyButton = screen.getByRole('button', { name: 'Copy command' });
|
|
||||||
fireEvent.click(copyButton);
|
|
||||||
await waitFor(() => expect(clipboardSpy).toHaveBeenCalled(), { interval: 0 });
|
|
||||||
|
|
||||||
const confirmButton = await screen.findByRole('button', { name: 'I ran this command' });
|
|
||||||
fireEvent.click(confirmButton);
|
|
||||||
|
|
||||||
expect(
|
|
||||||
screen.getByText("Pulse revokes the host's API token", {
|
|
||||||
exact: false,
|
|
||||||
}),
|
|
||||||
).toBeInTheDocument();
|
|
||||||
|
|
||||||
const removeHostButton = screen.getByRole('button', { name: 'Remove host' });
|
|
||||||
expect(removeHostButton).toBeEnabled();
|
|
||||||
fireEvent.click(removeHostButton);
|
|
||||||
|
|
||||||
await waitFor(() => expect(deleteHostAgentMock).toHaveBeenCalledWith('host-1'), { interval: 0 });
|
|
||||||
await waitFor(() => expect(notificationSuccessMock).toHaveBeenCalledWith('Host "Host One" removed', 4000), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
await waitFor(() => expect(screen.queryByText('Remove host "Host One"')).not.toBeInTheDocument(), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('removes a stale host without forcing', async () => {
|
|
||||||
deleteHostAgentMock.mockResolvedValue(undefined);
|
|
||||||
const host = createHost({
|
|
||||||
lastSeen: Date.now() - 5 * 60_000,
|
|
||||||
status: 'offline',
|
|
||||||
});
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const removeButton = screen.getByRole('button', { name: 'Remove' });
|
|
||||||
fireEvent.click(removeButton);
|
|
||||||
|
|
||||||
await screen.findByText('Remove host "Host One"');
|
|
||||||
|
|
||||||
const copyButton = screen.getByRole('button', { name: 'Copy command' });
|
|
||||||
fireEvent.click(copyButton);
|
|
||||||
await waitFor(() => expect(clipboardSpy).toHaveBeenCalled(), { interval: 0 });
|
|
||||||
|
|
||||||
const confirmButton = await screen.findByRole('button', { name: 'I ran this command' });
|
|
||||||
fireEvent.click(confirmButton);
|
|
||||||
|
|
||||||
const removeHostButton = screen.getByRole('button', { name: 'Remove host' });
|
|
||||||
fireEvent.click(removeHostButton);
|
|
||||||
|
|
||||||
await waitFor(() => expect(deleteHostAgentMock).toHaveBeenCalledWith('host-1'), { interval: 0 });
|
|
||||||
await waitFor(() => expect(notificationSuccessMock).toHaveBeenCalledWith('Host "Host One" removed', 4000), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
await waitFor(() => expect(screen.queryByText('Remove host "Host One"')).not.toBeInTheDocument(), {
|
|
||||||
interval: 0,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('shows macOS-specific uninstall guidance', async () => {
|
|
||||||
const host = createHost({
|
|
||||||
platform: 'macos',
|
|
||||||
});
|
|
||||||
|
|
||||||
setupComponent([host]);
|
|
||||||
|
|
||||||
const removeButton = screen.getByRole('button', { name: 'Remove' });
|
|
||||||
fireEvent.click(removeButton);
|
|
||||||
|
|
||||||
await screen.findByText('Remove host "Host One"');
|
|
||||||
expect(screen.getByText('launchctl unload', { exact: false })).toBeInTheDocument();
|
|
||||||
expect(screen.getByText('Unloads the launch agent, removes the plist, deletes the binary, and clears the local log.')).toBeInTheDocument();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
@ -0,0 +1,396 @@
|
||||||
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
||||||
|
import { render, fireEvent, screen, waitFor, cleanup } from '@solidjs/testing-library';
|
||||||
|
import { createStore } from 'solid-js/store';
|
||||||
|
import { UnifiedAgents } from '../UnifiedAgents';
|
||||||
|
import type { Host, DockerHost } from '@/types/api';
|
||||||
|
|
||||||
|
let mockWsStore: {
|
||||||
|
state: { hosts: Host[]; dockerHosts: DockerHost[] };
|
||||||
|
connected: () => boolean;
|
||||||
|
reconnecting: () => boolean;
|
||||||
|
activeAlerts: unknown[];
|
||||||
|
};
|
||||||
|
|
||||||
|
const lookupMock = vi.fn();
|
||||||
|
const createTokenMock = vi.fn();
|
||||||
|
const deleteHostAgentMock = vi.fn();
|
||||||
|
const deleteDockerHostMock = vi.fn();
|
||||||
|
const notificationSuccessMock = vi.fn();
|
||||||
|
const notificationErrorMock = vi.fn();
|
||||||
|
const notificationInfoMock = vi.fn();
|
||||||
|
const clipboardSpy = vi.fn();
|
||||||
|
const fetchMock = vi.fn();
|
||||||
|
|
||||||
|
vi.mock('@/App', () => ({
|
||||||
|
useWebSocket: () => mockWsStore,
|
||||||
|
}));
|
||||||
|
|
||||||
|
vi.mock('@/api/monitoring', () => ({
|
||||||
|
MonitoringAPI: {
|
||||||
|
lookupHost: (...args: unknown[]) => lookupMock(...args),
|
||||||
|
deleteHostAgent: (...args: unknown[]) => deleteHostAgentMock(...args),
|
||||||
|
deleteDockerHost: (...args: unknown[]) => deleteDockerHostMock(...args),
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
|
vi.mock('@/api/security', () => ({
|
||||||
|
SecurityAPI: {
|
||||||
|
createToken: (...args: unknown[]) => createTokenMock(...args),
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
|
vi.mock('@/stores/notifications', () => ({
|
||||||
|
notificationStore: {
|
||||||
|
success: (...args: unknown[]) => notificationSuccessMock(...args),
|
||||||
|
error: (...args: unknown[]) => notificationErrorMock(...args),
|
||||||
|
info: (...args: unknown[]) => notificationInfoMock(...args),
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
|
const createHost = (overrides?: Partial<Host>): Host => ({
|
||||||
|
id: 'host-1',
|
||||||
|
hostname: 'host-1.local',
|
||||||
|
displayName: 'Host One',
|
||||||
|
platform: 'linux',
|
||||||
|
osName: 'Ubuntu',
|
||||||
|
osVersion: '22.04',
|
||||||
|
kernelVersion: '6.5.0',
|
||||||
|
architecture: 'x86_64',
|
||||||
|
cpuCount: 4,
|
||||||
|
cpuUsage: 12.5,
|
||||||
|
memory: {
|
||||||
|
total: 8 * 1024 * 1024 * 1024,
|
||||||
|
used: 4 * 1024 * 1024 * 1024,
|
||||||
|
free: 4 * 1024 * 1024 * 1024,
|
||||||
|
usage: 50,
|
||||||
|
balloon: 0,
|
||||||
|
swapUsed: 0,
|
||||||
|
swapTotal: 0,
|
||||||
|
},
|
||||||
|
loadAverage: [],
|
||||||
|
disks: [],
|
||||||
|
networkInterfaces: [],
|
||||||
|
sensors: {
|
||||||
|
temperatureCelsius: {},
|
||||||
|
fanRpm: {},
|
||||||
|
additional: {},
|
||||||
|
},
|
||||||
|
status: 'online',
|
||||||
|
uptimeSeconds: 12_345,
|
||||||
|
intervalSeconds: 30,
|
||||||
|
lastSeen: Date.now(),
|
||||||
|
agentVersion: '1.2.3',
|
||||||
|
tags: ['prod'],
|
||||||
|
...overrides,
|
||||||
|
});
|
||||||
|
|
||||||
|
const createDockerHost = (overrides?: Partial<DockerHost>): DockerHost => ({
|
||||||
|
id: 'docker-host-1',
|
||||||
|
agentId: 'agent-1',
|
||||||
|
hostname: 'docker-host-1.local',
|
||||||
|
displayName: 'Docker Host One',
|
||||||
|
cpus: 4,
|
||||||
|
totalMemoryBytes: 8 * 1024 * 1024 * 1024,
|
||||||
|
uptimeSeconds: 12_345,
|
||||||
|
status: 'online',
|
||||||
|
lastSeen: Date.now(),
|
||||||
|
intervalSeconds: 30,
|
||||||
|
containers: [],
|
||||||
|
...overrides,
|
||||||
|
});
|
||||||
|
|
||||||
|
const setupComponent = (hosts: Host[] = [], dockerHosts: DockerHost[] = []) => {
|
||||||
|
const [state] = createStore({
|
||||||
|
hosts,
|
||||||
|
dockerHosts,
|
||||||
|
});
|
||||||
|
|
||||||
|
mockWsStore = {
|
||||||
|
state,
|
||||||
|
connected: () => true,
|
||||||
|
reconnecting: () => false,
|
||||||
|
activeAlerts: [],
|
||||||
|
};
|
||||||
|
|
||||||
|
return render(() => <UnifiedAgents />);
|
||||||
|
};
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
lookupMock.mockReset();
|
||||||
|
createTokenMock.mockReset();
|
||||||
|
deleteHostAgentMock.mockReset();
|
||||||
|
deleteDockerHostMock.mockReset();
|
||||||
|
notificationSuccessMock.mockReset();
|
||||||
|
notificationErrorMock.mockReset();
|
||||||
|
notificationInfoMock.mockReset();
|
||||||
|
clipboardSpy.mockReset().mockResolvedValue(undefined);
|
||||||
|
fetchMock.mockReset();
|
||||||
|
fetchMock.mockResolvedValue(
|
||||||
|
new Response(JSON.stringify({ requiresAuth: true, apiTokenConfigured: false }), {
|
||||||
|
status: 200,
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
vi.stubGlobal('fetch', fetchMock);
|
||||||
|
vi.stubGlobal('navigator', { clipboard: { writeText: clipboardSpy } } as unknown as Navigator);
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
cleanup();
|
||||||
|
vi.unstubAllGlobals();
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('UnifiedAgents token generation', () => {
|
||||||
|
it('renders the token generation UI when auth is required', async () => {
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('Generate API token')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
expect(screen.getByRole('button', { name: /Generate token/i })).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('generates a token and shows confirmation', async () => {
|
||||||
|
createTokenMock.mockResolvedValue({
|
||||||
|
token: 'test-token-123',
|
||||||
|
record: {
|
||||||
|
id: 'token-record',
|
||||||
|
name: 'Test Token',
|
||||||
|
prefix: 'abcdef',
|
||||||
|
suffix: '1234',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByRole('button', { name: /Generate token/i })).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
const generateButton = screen.getByRole('button', { name: /Generate token/i });
|
||||||
|
fireEvent.click(generateButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
await waitFor(
|
||||||
|
() => expect(screen.getByText(/Token.*created/i)).toBeInTheDocument(),
|
||||||
|
{ interval: 0 },
|
||||||
|
);
|
||||||
|
expect(notificationSuccessMock).toHaveBeenCalledWith(
|
||||||
|
'Token generated with Host and Docker permissions.',
|
||||||
|
4000,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('UnifiedAgents host lookup', () => {
|
||||||
|
it('performs host lookup and displays results', async () => {
|
||||||
|
createTokenMock.mockResolvedValue({
|
||||||
|
token: 'token-123',
|
||||||
|
record: {
|
||||||
|
id: 'token-record',
|
||||||
|
name: 'Test Token',
|
||||||
|
prefix: 'abcdef',
|
||||||
|
suffix: '1234',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const host = createHost();
|
||||||
|
setupComponent([host]);
|
||||||
|
|
||||||
|
// Generate token first to unlock commands
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByRole('button', { name: /Generate token/i })).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
const generateButton = screen.getByRole('button', { name: /Generate token/i });
|
||||||
|
fireEvent.click(generateButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
|
||||||
|
// Wait for commands to be unlocked and lookup UI to appear
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByPlaceholderText('Hostname or host ID')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
lookupMock.mockResolvedValue({
|
||||||
|
success: true,
|
||||||
|
host: {
|
||||||
|
id: host.id,
|
||||||
|
hostname: host.hostname,
|
||||||
|
displayName: host.displayName,
|
||||||
|
status: host.status,
|
||||||
|
connected: true,
|
||||||
|
lastSeen: host.lastSeen,
|
||||||
|
agentVersion: host.agentVersion,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const input = screen.getByPlaceholderText('Hostname or host ID') as HTMLInputElement;
|
||||||
|
fireEvent.input(input, { target: { value: host.id } });
|
||||||
|
|
||||||
|
const checkButton = screen.getByRole('button', { name: /Check status/i });
|
||||||
|
fireEvent.click(checkButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(lookupMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
await waitFor(
|
||||||
|
() => expect(screen.getByText('Connected')).toBeInTheDocument(),
|
||||||
|
{ interval: 0 },
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('shows error message when host is not found', async () => {
|
||||||
|
createTokenMock.mockResolvedValue({
|
||||||
|
token: 'token-456',
|
||||||
|
record: {
|
||||||
|
id: 'token-record-2',
|
||||||
|
name: 'Test Token 2',
|
||||||
|
prefix: 'ghijkl',
|
||||||
|
suffix: '5678',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
// Generate token first
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByRole('button', { name: /Generate token/i })).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
const generateButton = screen.getByRole('button', { name: /Generate token/i });
|
||||||
|
fireEvent.click(generateButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByPlaceholderText('Hostname or host ID')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
lookupMock.mockResolvedValue(null);
|
||||||
|
|
||||||
|
const query = 'missing-host';
|
||||||
|
const input = screen.getByPlaceholderText('Hostname or host ID') as HTMLInputElement;
|
||||||
|
fireEvent.input(input, { target: { value: query } });
|
||||||
|
|
||||||
|
const checkButton = screen.getByRole('button', { name: /Check status/i });
|
||||||
|
fireEvent.click(checkButton);
|
||||||
|
|
||||||
|
await waitFor(
|
||||||
|
() =>
|
||||||
|
expect(
|
||||||
|
screen.getByText(`No host has reported with "${query}" yet. Try again in a few seconds.`),
|
||||||
|
).toBeInTheDocument(),
|
||||||
|
{ interval: 0 },
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('UnifiedAgents managed agents table', () => {
|
||||||
|
it('displays host agents in the table', async () => {
|
||||||
|
const host = createHost({ hostname: 'test-server.local', displayName: 'Test Server' });
|
||||||
|
setupComponent([host]);
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('Managed Agents')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(screen.getByText('Test Server')).toBeInTheDocument();
|
||||||
|
expect(screen.getByText('Host')).toBeInTheDocument();
|
||||||
|
expect(screen.getByText('online')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('displays docker hosts in the table', async () => {
|
||||||
|
const dockerHost = createDockerHost({
|
||||||
|
hostname: 'docker-server.local',
|
||||||
|
displayName: 'Docker Server',
|
||||||
|
});
|
||||||
|
setupComponent([], [dockerHost]);
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('Managed Agents')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(screen.getByText('Docker Server')).toBeInTheDocument();
|
||||||
|
expect(screen.getByText('Docker')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('shows empty state when no agents are installed', async () => {
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('No agents installed yet.')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('shows legacy agent warning when legacy agents are present', async () => {
|
||||||
|
const legacyHost = createHost({ isLegacy: true });
|
||||||
|
setupComponent([legacyHost]);
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText(/legacy agent.*detected/i)).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
expect(screen.getByText('Legacy')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('UnifiedAgents platform commands', () => {
|
||||||
|
it('shows installation commands for all platforms after token generation', async () => {
|
||||||
|
createTokenMock.mockResolvedValue({
|
||||||
|
token: 'test-token',
|
||||||
|
record: {
|
||||||
|
id: 'token-record',
|
||||||
|
name: 'Test Token',
|
||||||
|
prefix: 'abc',
|
||||||
|
suffix: '123',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
const generateButton = screen.getByRole('button', { name: /Generate token/i });
|
||||||
|
fireEvent.click(generateButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('Install on Linux')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
expect(screen.getByText('Install on macOS')).toBeInTheDocument();
|
||||||
|
expect(screen.getByText('Install on Windows')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('includes docker flag when docker monitoring is enabled', async () => {
|
||||||
|
createTokenMock.mockResolvedValue({
|
||||||
|
token: 'test-token',
|
||||||
|
record: {
|
||||||
|
id: 'token-record',
|
||||||
|
name: 'Test Token',
|
||||||
|
prefix: 'abc',
|
||||||
|
suffix: '123',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
setupComponent();
|
||||||
|
|
||||||
|
const generateButton = screen.getByRole('button', { name: /Generate token/i });
|
||||||
|
fireEvent.click(generateButton);
|
||||||
|
|
||||||
|
await waitFor(() => expect(createTokenMock).toHaveBeenCalled(), { interval: 0 });
|
||||||
|
|
||||||
|
await waitFor(() => {
|
||||||
|
expect(screen.getByText('Enable Docker monitoring')).toBeInTheDocument();
|
||||||
|
});
|
||||||
|
|
||||||
|
// Docker monitoring is enabled by default
|
||||||
|
const checkbox = screen.getByRole('checkbox');
|
||||||
|
expect(checkbox).toBeChecked();
|
||||||
|
|
||||||
|
// Find a copy button and check that the command includes the docker flag
|
||||||
|
const copyButtons = screen.getAllByRole('button', { name: /Copy command/i });
|
||||||
|
expect(copyButtons.length).toBeGreaterThan(0);
|
||||||
|
});
|
||||||
|
});
|
||||||
Loading…
Reference in a new issue