119 lines
3.4 KiB
TypeScript
119 lines
3.4 KiB
TypeScript
import React from 'react';
|
|
import { render, screen, waitFor } from '@testing-library/react';
|
|
import { HashRouter } from 'react-router-dom';
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { setLocale } from '../src/i18n';
|
|
|
|
const mocks = vi.hoisted(() => ({
|
|
invoke: vi.fn(),
|
|
}));
|
|
|
|
vi.mock('../src/pages/Login', async () => {
|
|
const ReactModule = await import('react');
|
|
return {
|
|
default: function LoginPageMock() {
|
|
return ReactModule.createElement('div', null, 'login-page');
|
|
},
|
|
};
|
|
});
|
|
|
|
vi.mock('../src/pages/Setting', async () => {
|
|
const ReactModule = await import('react');
|
|
|
|
return {
|
|
default: function SettingPageMock() {
|
|
ReactModule.useEffect(() => {
|
|
void import('../src/lib/host-api').then(({ hostApiFetch }) => {
|
|
void hostApiFetch('/api/gateway/status').catch(() => {});
|
|
});
|
|
}, []);
|
|
|
|
return ReactModule.createElement('div', null, 'protected-setting-page');
|
|
},
|
|
};
|
|
});
|
|
|
|
import { AppRouter } from '../src/router';
|
|
|
|
describe('AppRouter host api auth regression', () => {
|
|
function renderProtectedRoute() {
|
|
render(
|
|
<HashRouter>
|
|
<AppRouter />
|
|
</HashRouter>,
|
|
);
|
|
}
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
setLocale('en');
|
|
window.sessionStorage.clear();
|
|
window.localStorage.clear();
|
|
document.cookie = 'token=; expires=Thu, 01 Jan 1970 00:00:00 GMT; path=/';
|
|
window.location.hash = '#/setting';
|
|
window.sessionStorage.setItem('token', JSON.stringify('access-token'));
|
|
|
|
(window as typeof window & { api?: unknown }).api = {
|
|
invoke: mocks.invoke,
|
|
platform: 'darwin',
|
|
};
|
|
|
|
mocks.invoke.mockImplementation(async (channel: string) => {
|
|
if (channel === 'hostapi:fetch') {
|
|
return {
|
|
success: false,
|
|
ok: false,
|
|
status: 401,
|
|
code: 'HOST_API_UNAUTHORIZED',
|
|
error: 'Host API authentication failed',
|
|
};
|
|
}
|
|
|
|
throw new Error(`Unexpected IPC channel: ${channel}`);
|
|
});
|
|
});
|
|
|
|
it('keeps the user on a protected route when local host api auth fails', async () => {
|
|
renderProtectedRoute();
|
|
|
|
expect(await screen.findByText('protected-setting-page')).toBeTruthy();
|
|
|
|
await waitFor(() => {
|
|
expect(mocks.invoke).toHaveBeenCalledWith('hostapi:fetch', expect.objectContaining({
|
|
path: '/api/gateway/status',
|
|
}));
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(screen.queryByText('login-page')).toBeNull();
|
|
expect(screen.getByText('protected-setting-page')).toBeTruthy();
|
|
expect(window.sessionStorage.getItem('token')).toBe(JSON.stringify('access-token'));
|
|
});
|
|
});
|
|
|
|
it('keeps the user on a protected route when upstream api returns unauthorized', async () => {
|
|
mocks.invoke.mockResolvedValueOnce({
|
|
success: false,
|
|
ok: false,
|
|
status: 401,
|
|
error: 'Unauthorized',
|
|
});
|
|
|
|
renderProtectedRoute();
|
|
|
|
expect(await screen.findByText('protected-setting-page')).toBeTruthy();
|
|
|
|
await waitFor(() => {
|
|
expect(mocks.invoke).toHaveBeenCalledWith('hostapi:fetch', expect.objectContaining({
|
|
path: '/api/gateway/status',
|
|
}));
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(screen.queryByText('login-page')).toBeNull();
|
|
expect(screen.getByText('protected-setting-page')).toBeTruthy();
|
|
expect(window.sessionStorage.getItem('token')).toBe(JSON.stringify('access-token'));
|
|
});
|
|
});
|
|
});
|