diff --git a/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.test.tsx b/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.test.tsx new file mode 100644 index 0000000000000..11becb14625a9 --- /dev/null +++ b/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.test.tsx @@ -0,0 +1,81 @@ +/* + * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one + * or more contributor license agreements. Licensed under the Elastic License; + * you may not use this file except in compliance with the Elastic License. + */ + +import uuid from 'uuid'; +import { createMemoryHistory } from 'history'; + +const history = createMemoryHistory(); + +import { mockRule } from './__mocks__/mock'; +import { getActions } from './columns'; + +jest.mock('./actions', () => ({ + duplicateRulesAction: jest.fn(), + deleteRulesAction: jest.fn(), +})); + +import { duplicateRulesAction, deleteRulesAction } from './actions'; + +describe('AllRulesTable Columns', () => { + describe('getActions', () => { + const rule = mockRule(uuid.v4()); + let results: string[] = []; + const dispatch = jest.fn(); + const dispatchToaster = jest.fn(); + const reFetchRules = jest.fn(); + + beforeEach(() => { + results = []; + + reFetchRules.mockImplementation(() => { + results.push('reFetchRules'); + Promise.resolve(); + }); + }); + + test('duplicate rule onClick should call refetch after the rule is duplicated', async () => { + (duplicateRulesAction as jest.Mock).mockImplementation( + () => + new Promise(resolve => + setTimeout(() => { + results.push('duplicateRulesAction'); + resolve(); + }, 500) + ) + ); + + const duplicateRulesActionObject = getActions( + dispatch, + dispatchToaster, + history, + reFetchRules + )[1]; + await duplicateRulesActionObject.onClick(rule); + expect(results).toEqual(['duplicateRulesAction', 'reFetchRules']); + }); + + test('delete rule onClick should call refetch after the rule is deleted', async () => { + (deleteRulesAction as jest.Mock).mockImplementation( + () => + new Promise(resolve => + setTimeout(() => { + results.push('deleteRulesAction'); + resolve(); + }, 500) + ) + ); + + const deleteRulesActionObject = getActions( + dispatch, + dispatchToaster, + history, + reFetchRules + )[3]; + await deleteRulesActionObject.onClick(rule); + expect(results).toEqual(['deleteRulesAction', 'reFetchRules']); + }); + }); +}); diff --git a/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.tsx b/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.tsx index 03b8930ee836d..8887fa788d49e 100644 --- a/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.tsx +++ b/x-pack/legacy/plugins/siem/public/pages/detection_engine/rules/all/columns.tsx @@ -34,7 +34,7 @@ import { } from './actions'; import { Action } from './reducer'; -const getActions = ( +export const getActions = ( dispatch: React.Dispatch, dispatchToaster: Dispatch, history: H.History, @@ -53,9 +53,9 @@ const getActions = ( type: 'icon', icon: 'copy', name: i18n.DUPLICATE_RULE, - onClick: (rowItem: Rule) => { - duplicateRulesAction([rowItem], [rowItem.id], dispatch, dispatchToaster); - reFetchRules(true); + onClick: async (rowItem: Rule) => { + await duplicateRulesAction([rowItem], [rowItem.id], dispatch, dispatchToaster); + await reFetchRules(true); }, }, { @@ -71,9 +71,9 @@ const getActions = ( type: 'icon', icon: 'trash', name: i18n.DELETE_RULE, - onClick: (rowItem: Rule) => { - deleteRulesAction([rowItem.id], dispatch, dispatchToaster); - reFetchRules(true); + onClick: async (rowItem: Rule) => { + await deleteRulesAction([rowItem.id], dispatch, dispatchToaster); + await reFetchRules(true); }, }, ];