Skip to content

Commit 9ca181f

Browse files
committed
Use more specific types and rename newTransformValidate
1 parent 9b8ffc6 commit 9ca181f

File tree

6 files changed

+23
-16
lines changed

6 files changed

+23
-16
lines changed

x-pack/plugins/security_solution/common/detection_engine/schemas/response/rules_schema.mocks.ts

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,12 @@
66
*/
77

88
import { DEFAULT_INDICATOR_SOURCE_PATH } from '../../../constants';
9-
import type { FullResponseSchema } from '../request';
9+
import type {
10+
EqlResponseSchema,
11+
MachineLearningResponseSchema,
12+
QueryResponseSchema,
13+
ThreatMatchResponseSchema,
14+
} from '../request';
1015
import { getListArrayMock } from '../types/lists.mock';
1116

1217
export const ANCHOR_DATE = '2020-02-20T03:57:54.037Z';
@@ -58,7 +63,7 @@ const getResponseBaseParams = (anchorDate: string = ANCHOR_DATE) => ({
5863
namespace: undefined,
5964
});
6065

61-
export const getRulesSchemaMock = (anchorDate: string = ANCHOR_DATE): FullResponseSchema => ({
66+
export const getRulesSchemaMock = (anchorDate: string = ANCHOR_DATE): QueryResponseSchema => ({
6267
...getResponseBaseParams(anchorDate),
6368
query: 'user.name: root or user.name: admin',
6469
type: 'query',
@@ -69,7 +74,9 @@ export const getRulesSchemaMock = (anchorDate: string = ANCHOR_DATE): FullRespon
6974
saved_id: undefined,
7075
});
7176

72-
export const getRulesMlSchemaMock = (anchorDate: string = ANCHOR_DATE): FullResponseSchema => {
77+
export const getRulesMlSchemaMock = (
78+
anchorDate: string = ANCHOR_DATE
79+
): MachineLearningResponseSchema => {
7380
return {
7481
...getResponseBaseParams(anchorDate),
7582
type: 'machine_learning',
@@ -80,7 +87,7 @@ export const getRulesMlSchemaMock = (anchorDate: string = ANCHOR_DATE): FullResp
8087

8188
export const getThreatMatchingSchemaMock = (
8289
anchorDate: string = ANCHOR_DATE
83-
): FullResponseSchema => {
90+
): ThreatMatchResponseSchema => {
8491
return {
8592
...getResponseBaseParams(anchorDate),
8693
type: 'threat_match',
@@ -124,7 +131,7 @@ export const getThreatMatchingSchemaMock = (
124131
*/
125132
export const getThreatMatchingSchemaPartialMock = (
126133
enabled = false
127-
): Partial<FullResponseSchema> => {
134+
): Partial<ThreatMatchResponseSchema> => {
128135
return {
129136
author: [],
130137
created_by: 'elastic',
@@ -193,7 +200,7 @@ export const getThreatMatchingSchemaPartialMock = (
193200
};
194201
};
195202

196-
export const getRulesEqlSchemaMock = (anchorDate: string = ANCHOR_DATE): FullResponseSchema => {
203+
export const getRulesEqlSchemaMock = (anchorDate: string = ANCHOR_DATE): EqlResponseSchema => {
197204
return {
198205
...getResponseBaseParams(anchorDate),
199206
language: 'eql',

x-pack/plugins/security_solution/server/lib/detection_engine/routes/rules/create_rules_route.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ import { readRules } from '../../rules/read_rules';
1616
import { buildSiemResponse } from '../utils';
1717

1818
import { createRulesSchema } from '../../../../../common/detection_engine/schemas/request';
19-
import { newTransformValidate } from './validate';
19+
import { transformValidate } from './validate';
2020
import { createRuleValidateTypeDependents } from '../../../../../common/detection_engine/schemas/request/create_rules_type_dependents';
2121
import { createRules } from '../../rules/create_rules';
2222

@@ -85,7 +85,7 @@ export const createRulesRoute = (
8585

8686
const ruleExecutionSummary = await ruleExecutionLog.getExecutionSummary(createdRule.id);
8787

88-
const [validated, errors] = newTransformValidate(createdRule, ruleExecutionSummary);
88+
const [validated, errors] = transformValidate(createdRule, ruleExecutionSummary);
8989
if (errors != null) {
9090
return siemResponse.error({ statusCode: 500, body: errors });
9191
} else {

x-pack/plugins/security_solution/server/lib/detection_engine/routes/rules/patch_rules_route.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ import { patchRules } from '../../rules/patch_rules';
1818
import { buildSiemResponse } from '../utils';
1919

2020
import { getIdError } from './utils';
21-
import { newTransformValidate } from './validate';
21+
import { transformValidate } from './validate';
2222
import { readRules } from '../../rules/read_rules';
2323
import { legacyMigrate } from '../../rules/utils';
2424

@@ -83,7 +83,7 @@ export const patchRulesRoute = (router: SecuritySolutionPluginRouter, ml: SetupP
8383
if (rule != null && rule.enabled != null && rule.name != null) {
8484
const ruleExecutionSummary = await ruleExecutionLog.getExecutionSummary(rule.id);
8585

86-
const [validated, errors] = newTransformValidate(rule, ruleExecutionSummary);
86+
const [validated, errors] = transformValidate(rule, ruleExecutionSummary);
8787
if (errors != null) {
8888
return siemResponse.error({ statusCode: 500, body: errors });
8989
} else {

x-pack/plugins/security_solution/server/lib/detection_engine/routes/rules/update_rules_route.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ import { throwAuthzError } from '../../../machine_learning/validation';
1616
import { buildSiemResponse } from '../utils';
1717

1818
import { getIdError } from './utils';
19-
import { newTransformValidate } from './validate';
19+
import { transformValidate } from './validate';
2020
import { updateRules } from '../../rules/update_rules';
2121
import { buildRouteValidation } from '../../../../utils/build_validation/route_validation';
2222
import { legacyMigrate } from '../../rules/utils';
@@ -73,7 +73,7 @@ export const updateRulesRoute = (router: SecuritySolutionPluginRouter, ml: Setup
7373
if (rule != null) {
7474
const ruleExecutionLog = ctx.securitySolution.getRuleExecutionLog();
7575
const ruleExecutionSummary = await ruleExecutionLog.getExecutionSummary(rule.id);
76-
const [validated, errors] = newTransformValidate(rule, ruleExecutionSummary);
76+
const [validated, errors] = transformValidate(rule, ruleExecutionSummary);
7777
if (errors != null) {
7878
return siemResponse.error({ statusCode: 500, body: errors });
7979
} else {

x-pack/plugins/security_solution/server/lib/detection_engine/routes/rules/validate.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
* 2.0.
66
*/
77

8-
import { newTransformValidate, transformValidateBulkError } from './validate';
8+
import { transformValidate, transformValidateBulkError } from './validate';
99
import type { BulkError } from '../utils';
1010
import { getRuleMock } from '../__mocks__/request_responses';
1111
import { ruleExecutionSummaryMock } from '../../../../../common/detection_engine/rule_monitoring/mocks';
@@ -82,7 +82,7 @@ describe('validate', () => {
8282
describe('transformValidate', () => {
8383
test('it should do a validation correctly of a partial alert', () => {
8484
const ruleAlert = getRuleMock(getQueryRuleParams());
85-
const [validated, errors] = newTransformValidate(ruleAlert, null);
85+
const [validated, errors] = transformValidate(ruleAlert, null);
8686
expect(validated).toEqual(ruleOutput());
8787
expect(errors).toEqual(null);
8888
});
@@ -91,7 +91,7 @@ describe('validate', () => {
9191
const ruleAlert = getRuleMock(getQueryRuleParams());
9292
// @ts-expect-error
9393
delete ruleAlert.name;
94-
const [validated, errors] = newTransformValidate(ruleAlert, null);
94+
const [validated, errors] = transformValidate(ruleAlert, null);
9595
expect(validated).toEqual(null);
9696
expect(errors).toEqual('Invalid value "undefined" supplied to "name"');
9797
});

x-pack/plugins/security_solution/server/lib/detection_engine/routes/rules/validate.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ import type { RuleParams } from '../../schemas/rule_schemas';
2020
import type { LegacyRulesActionsSavedObject } from '../../rule_actions/legacy_get_rule_actions_saved_object';
2121
import { internalRuleToAPIResponse } from '../../schemas/rule_converters';
2222

23-
export const newTransformValidate = (
23+
export const transformValidate = (
2424
rule: PartialRule<RuleParams>,
2525
ruleExecutionSummary: RuleExecutionSummary | null,
2626
legacyRuleActions?: LegacyRulesActionsSavedObject | null

0 commit comments

Comments
 (0)