add useWsTemplates tests

This commit is contained in:
Keith Grant
2020-07-13 10:39:37 -07:00
parent 981c9527b2
commit b76783791a
3 changed files with 198 additions and 8 deletions

View File

@@ -75,6 +75,7 @@ const mockTemplates = [
];
describe('<TemplateList />', () => {
let debug;
beforeEach(() => {
UnifiedJobTemplatesAPI.read.mockResolvedValue({
data: {
@@ -88,10 +89,13 @@ describe('<TemplateList />', () => {
actions: [],
},
});
debug = global.console.debug; // eslint-disable-line prefer-destructuring
global.console.debug = () => {};
});
afterEach(() => {
jest.clearAllMocks();
global.console.debug = debug;
});
test('initially renders successfully', async () => {

View File

@@ -10,13 +10,6 @@ export default function useWsTemplates(initialTemplates) {
setTemplates(initialTemplates);
}, [initialTemplates]);
// x = {
// unified_job_id: 548,
// status: 'pending',
// type: 'job',
// group_name: 'jobs',
// unified_job_template_id: 26,
// };
useEffect(
function parseWsMessage() {
if (!lastMessage?.unified_job_id) {
@@ -89,7 +82,7 @@ function updateTemplate(template, message) {
const job = {
id: message.unified_job_id,
status: message.status,
finished: message.finished,
finished: message.finished || null,
type: message.type,
};
const index = recentJobs.findIndex(j => j.id === job.id);

View File

@@ -0,0 +1,193 @@
import React from 'react';
import { act } from 'react-dom/test-utils';
import WS from 'jest-websocket-mock';
import { mountWithContexts } from '../../../../testUtils/enzymeHelpers';
import useWsTemplates from './useWsTemplates';
/*
Jest mock timers dont play well with jest-websocket-mock,
so we'll stub out throttling to resolve immediately
*/
jest.mock('../../../util/useThrottle', () => ({
__esModule: true,
default: jest.fn(val => val),
}));
function TestInner() {
return <div />;
}
function Test({ templates }) {
const syncedTemplates = useWsTemplates(templates);
return <TestInner templates={syncedTemplates} />;
}
describe('useWsTemplates hook', () => {
let debug;
let wrapper;
beforeEach(() => {
debug = global.console.debug; // eslint-disable-line prefer-destructuring
global.console.debug = () => {};
});
afterEach(() => {
global.console.debug = debug;
});
test('should return templates list', () => {
const templates = [{ id: 1 }];
wrapper = mountWithContexts(<Test templates={templates} />);
expect(wrapper.find('TestInner').prop('templates')).toEqual(templates);
WS.clean();
});
test('should establish websocket connection', async () => {
global.document.cookie = 'csrftoken=abc123';
const mockServer = new WS('wss://localhost/websocket/');
const templates = [{ id: 1 }];
await act(async () => {
wrapper = await mountWithContexts(<Test templates={templates} />);
});
await mockServer.connected;
await expect(mockServer).toReceiveMessage(
JSON.stringify({
xrftoken: 'abc123',
groups: {
jobs: ['status_changed'],
control: ['limit_reached_1'],
},
})
);
WS.clean();
});
test('should update recent job status', async () => {
global.document.cookie = 'csrftoken=abc123';
const mockServer = new WS('wss://localhost/websocket/');
const templates = [
{
id: 1,
summary_fields: {
recent_jobs: [
{
id: 10,
type: 'job',
status: 'running',
},
{
id: 11,
type: 'job',
status: 'successful',
},
],
},
},
];
await act(async () => {
wrapper = await mountWithContexts(<Test templates={templates} />);
});
await mockServer.connected;
await expect(mockServer).toReceiveMessage(
JSON.stringify({
xrftoken: 'abc123',
groups: {
jobs: ['status_changed'],
control: ['limit_reached_1'],
},
})
);
expect(
wrapper.find('TestInner').prop('templates')[0].summary_fields
.recent_jobs[0].status
).toEqual('running');
act(() => {
mockServer.send(
JSON.stringify({
unified_job_template_id: 1,
unified_job_id: 10,
type: 'job',
status: 'successful',
})
);
});
wrapper.update();
expect(
wrapper.find('TestInner').prop('templates')[0].summary_fields
.recent_jobs[0].status
).toEqual('successful');
WS.clean();
});
test('should add new job status', async () => {
global.document.cookie = 'csrftoken=abc123';
const mockServer = new WS('wss://localhost/websocket/');
const templates = [
{
id: 1,
summary_fields: {
recent_jobs: [
{
id: 10,
type: 'job',
status: 'running',
},
{
id: 11,
type: 'job',
status: 'successful',
},
],
},
},
];
await act(async () => {
wrapper = await mountWithContexts(<Test templates={templates} />);
});
await mockServer.connected;
await expect(mockServer).toReceiveMessage(
JSON.stringify({
xrftoken: 'abc123',
groups: {
jobs: ['status_changed'],
control: ['limit_reached_1'],
},
})
);
expect(
wrapper.find('TestInner').prop('templates')[0].summary_fields
.recent_jobs[0].status
).toEqual('running');
act(() => {
mockServer.send(
JSON.stringify({
unified_job_template_id: 1,
unified_job_id: 13,
type: 'job',
status: 'running',
})
);
});
wrapper.update();
expect(
wrapper.find('TestInner').prop('templates')[0].summary_fields.recent_jobs
).toHaveLength(3);
expect(
wrapper.find('TestInner').prop('templates')[0].summary_fields
.recent_jobs[0]
).toEqual({
id: 13,
status: 'running',
finished: null,
type: 'job',
});
WS.clean();
});
});