1
0
mirror of https://github.com/ansible/awx.git synced 2024-10-30 22:21:13 +03:00

add useWsInventorySources tests

This commit is contained in:
Keith Grant 2020-07-14 14:29:24 -07:00
parent 4fca57dfd2
commit f1e3b4f4f0
2 changed files with 128 additions and 0 deletions

View File

@ -55,8 +55,11 @@ const sources = {
describe('<InventorySourceList />', () => {
let wrapper;
let history;
let debug;
beforeEach(async () => {
debug = global.console.debug; // eslint-disable-line prefer-destructuring
global.console.debug = () => {};
InventoriesAPI.readSources.mockResolvedValue(sources);
InventorySourcesAPI.readOptions.mockResolvedValue({
data: {
@ -98,6 +101,7 @@ describe('<InventorySourceList />', () => {
afterEach(() => {
wrapper.unmount();
jest.clearAllMocks();
global.console.debug = debug;
});
test('should mount properly', async () => {

View File

@ -0,0 +1,124 @@
import React from 'react';
import { act } from 'react-dom/test-utils';
import WS from 'jest-websocket-mock';
import { mountWithContexts } from '../../../../testUtils/enzymeHelpers';
import useWsInventorySources from './useWsInventorySources';
/*
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({ sources }) {
const syncedSources = useWsInventorySources(sources);
return <TestInner sources={syncedSources} />;
}
describe('useWsInventorySources 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 sources list', () => {
const sources = [{ id: 1 }];
wrapper = mountWithContexts(<Test sources={sources} />);
expect(wrapper.find('TestInner').prop('sources')).toEqual(sources);
WS.clean();
});
test('should establish websocket connection', async () => {
global.document.cookie = 'csrftoken=abc123';
const mockServer = new WS('wss://localhost/websocket/');
const sources = [{ id: 1 }];
await act(async () => {
wrapper = await mountWithContexts(<Test sources={sources} />);
});
await mockServer.connected;
await expect(mockServer).toReceiveMessage(
JSON.stringify({
xrftoken: 'abc123',
groups: {
jobs: ['status_changed'],
control: ['limit_reached_1'],
},
})
);
WS.clean();
});
test('should update last job status', async () => {
global.document.cookie = 'csrftoken=abc123';
const mockServer = new WS('wss://localhost/websocket/');
const sources = [
{
id: 3,
status: 'running',
summary_fields: {
last_job: {
id: 5,
status: 'running',
},
},
},
];
await act(async () => {
wrapper = await mountWithContexts(<Test sources={sources} />);
});
await mockServer.connected;
await expect(mockServer).toReceiveMessage(
JSON.stringify({
xrftoken: 'abc123',
groups: {
jobs: ['status_changed'],
control: ['limit_reached_1'],
},
})
);
act(() => {
mockServer.send(
JSON.stringify({
unified_job_id: 5,
inventory_source_id: 3,
type: 'job',
status: 'successful',
finished: 'the_time',
})
);
});
wrapper.update();
const source = wrapper.find('TestInner').prop('sources')[0];
expect(source).toEqual({
id: 3,
status: 'successful',
last_updated: 'the_time',
summary_fields: {
last_job: {
id: 5,
status: 'successful',
finished: 'the_time',
},
},
});
WS.clean();
});
});