diff --git a/awx/ui_next/src/screens/Template/TemplateList/TemplateList.test.jsx b/awx/ui_next/src/screens/Template/TemplateList/TemplateList.test.jsx
index b04ef9c35b..377e87583e 100644
--- a/awx/ui_next/src/screens/Template/TemplateList/TemplateList.test.jsx
+++ b/awx/ui_next/src/screens/Template/TemplateList/TemplateList.test.jsx
@@ -75,6 +75,7 @@ const mockTemplates = [
];
describe('', () => {
+ let debug;
beforeEach(() => {
UnifiedJobTemplatesAPI.read.mockResolvedValue({
data: {
@@ -88,10 +89,13 @@ describe('', () => {
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 () => {
diff --git a/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.js b/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.js
index 052d2a9197..42863dfdaf 100644
--- a/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.js
+++ b/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.js
@@ -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);
diff --git a/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.test.jsx b/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.test.jsx
new file mode 100644
index 0000000000..61bc6e042e
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/TemplateList/useWsTemplates.test.jsx
@@ -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 don’t 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
;
+}
+function Test({ templates }) {
+ const syncedTemplates = useWsTemplates(templates);
+ return ;
+}
+
+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();
+
+ 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();
+ });
+
+ 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();
+ });
+
+ 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();
+ });
+
+ 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();
+ });
+});