Skip to content

Commit

Permalink
Fix more types
Browse files Browse the repository at this point in the history
  • Loading branch information
JiaweiWu committed Nov 10, 2022
1 parent f0b1c66 commit 897ec3f
Showing 1 changed file with 13 additions and 12 deletions.
25 changes: 13 additions & 12 deletions x-pack/plugins/alerting/server/mocks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,18 @@ const createAbortableSearchServiceMock = () => {
};
};

const createRuleMonitoringServiceMock = () => {
const mock = {
setLastRunMetricsTotalSearchDurationMs: jest.fn(),
setLastRunMetricsTotalIndexingDurationMs: jest.fn(),
setLastRunMetricsTotalAlertsDetected: jest.fn(),
setLastRunMetricsTotalAlertsCreated: jest.fn(),
setLastRunMetricsGapDurationS: jest.fn(),
} as unknown as jest.Mocked<PublicRuleMonitoringService>;

return mock;
};

const createRuleExecutorServicesMock = <
InstanceState extends AlertInstanceState = AlertInstanceState,
InstanceContext extends AlertInstanceContext = AlertInstanceContext
Expand All @@ -118,6 +130,7 @@ const createRuleExecutorServicesMock = <
shouldStopExecution: () => true,
search: createAbortableSearchServiceMock(),
searchSourceClient: searchSourceCommonMock,
ruleMonitoringService: createRuleMonitoringServiceMock(),
};
};
export type RuleExecutorServicesMock = ReturnType<typeof createRuleExecutorServicesMock>;
Expand All @@ -129,16 +142,4 @@ export const alertsMock = {
createRuleExecutorServices: createRuleExecutorServicesMock,
};

const createRuleMonitoringServiceMock = () => {
const mock = {
setLastRunMetricsTotalSearchDurationMs: jest.fn(),
setLastRunMetricsTotalIndexingDurationMs: jest.fn(),
setLastRunMetricsTotalAlertsDetected: jest.fn(),
setLastRunMetricsTotalAlertsCreated: jest.fn(),
setLastRunMetricsGapDurationS: jest.fn(),
} as unknown as jest.Mocked<PublicRuleMonitoringService>;

return mock;
};

export const ruleMonitoringServiceMock = { create: createRuleMonitoringServiceMock };

0 comments on commit 897ec3f

Please sign in to comment.