mirror of https://github.com/grafana/grafana
Alerting: Use new endpoints in the Modify Export (#75796)
Co-authored-by: Konrad Lalik <konrad.lalik@grafana.com>pull/76357/head^2
Sonia Aguilar
2 years ago
committed by
GitHub
parent
94c15e4926
commit
fbbf9b1a8f
@ -0,0 +1,125 @@ |
||||
import { RulerRuleDTO, RulerRuleGroupDTO } from 'app/types/unified-alerting-dto'; |
||||
|
||||
import { mockRulerGrafanaRule } from '../../../mocks'; |
||||
import { RuleFormValues } from '../../../types/rule-form'; |
||||
import { Annotation } from '../../../utils/constants'; |
||||
import { getDefaultFormValues } from '../../../utils/rule-form'; |
||||
|
||||
import { getPayloadToExport } from './ModifyExportRuleForm'; |
||||
|
||||
const rule1 = mockRulerGrafanaRule( |
||||
{ |
||||
for: '1m', |
||||
labels: { severity: 'critical', region: 'region1' }, |
||||
annotations: { [Annotation.summary]: 'This grafana rule1' }, |
||||
}, |
||||
{ uid: 'uid-rule-1', title: 'Rule1', data: [] } |
||||
); |
||||
|
||||
const rule2 = mockRulerGrafanaRule( |
||||
{ |
||||
for: '1m', |
||||
labels: { severity: 'notcritical', region: 'region2' }, |
||||
annotations: { [Annotation.summary]: 'This grafana rule2' }, |
||||
}, |
||||
{ uid: 'uid-rule-2', title: 'Rule2', data: [] } |
||||
); |
||||
|
||||
const rule3 = mockRulerGrafanaRule( |
||||
{ |
||||
for: '1m', |
||||
labels: { severity: 'notcritical3', region: 'region3' }, |
||||
annotations: { [Annotation.summary]: 'This grafana rule2' }, |
||||
}, |
||||
{ uid: 'uid-rule-3', title: 'Rule3', data: [] } |
||||
); |
||||
|
||||
// Prepare the form values for rule2 updated
|
||||
const defaultValues = getDefaultFormValues(); |
||||
const formValuesForRule2Updated: RuleFormValues = { |
||||
...defaultValues, |
||||
queries: [ |
||||
{ |
||||
refId: 'A', |
||||
relativeTimeRange: { from: 900, to: 1000 }, |
||||
datasourceUid: 'dsuid', |
||||
model: { |
||||
refId: 'A', |
||||
hide: true, |
||||
}, |
||||
queryType: 'query', |
||||
}, |
||||
], |
||||
condition: 'A', |
||||
forTime: 2455, |
||||
name: 'Rule2 updated', |
||||
labels: [{ key: 'newLabel', value: 'newLabel' }], |
||||
annotations: [{ key: 'summary', value: 'This grafana rule2 updated' }], |
||||
}; |
||||
|
||||
const expectedModifiedRule2 = (uid: string) => ({ |
||||
annotations: { |
||||
summary: 'This grafana rule2 updated', |
||||
}, |
||||
for: '5m', |
||||
grafana_alert: { |
||||
condition: 'A', |
||||
data: [ |
||||
{ |
||||
datasourceUid: 'dsuid', |
||||
model: { |
||||
refId: 'A', |
||||
hide: true, |
||||
}, |
||||
queryType: 'query', |
||||
refId: 'A', |
||||
relativeTimeRange: { |
||||
from: 900, |
||||
to: 1000, |
||||
}, |
||||
}, |
||||
], |
||||
exec_err_state: 'Error', |
||||
is_paused: false, |
||||
no_data_state: 'NoData', |
||||
title: 'Rule2 updated', |
||||
uid: uid, |
||||
}, |
||||
labels: { |
||||
newLabel: 'newLabel', |
||||
}, |
||||
}); |
||||
|
||||
describe('getPayloadFromDto', () => { |
||||
const groupDto: RulerRuleGroupDTO<RulerRuleDTO> = { |
||||
name: 'Test Group', |
||||
rules: [rule1, rule2, rule3], |
||||
}; |
||||
|
||||
it('should return a ModifyExportPayload with the updated rule added to a group with this rule belongs, in the same position', () => { |
||||
const result = getPayloadToExport('uid-rule-2', formValuesForRule2Updated, groupDto); |
||||
expect(result).toEqual({ |
||||
name: 'Test Group', |
||||
rules: [rule1, expectedModifiedRule2('uid-rule-2'), rule3], |
||||
}); |
||||
}); |
||||
it('should return a ModifyExportPayload with the updated rule added to a non empty rule where this rule does not belong, in the last position', () => { |
||||
const result = getPayloadToExport('uid-rule-5', formValuesForRule2Updated, groupDto); |
||||
expect(result).toEqual({ |
||||
name: 'Test Group', |
||||
rules: [rule1, rule2, rule3, expectedModifiedRule2('uid-rule-5')], |
||||
}); |
||||
}); |
||||
|
||||
it('should return a ModifyExportPayload with the updated rule added to an empty group', () => { |
||||
const emptyGroupDto: RulerRuleGroupDTO<RulerRuleDTO> = { |
||||
name: 'Empty Group', |
||||
rules: [], |
||||
}; |
||||
const result = getPayloadToExport('uid-rule-2', formValuesForRule2Updated, emptyGroupDto); |
||||
expect(result).toEqual({ |
||||
name: 'Empty Group', |
||||
rules: [expectedModifiedRule2('uid-rule-2')], |
||||
}); |
||||
}); |
||||
}); |
||||
Loading…
Reference in new issue