import { dimensionVariable, fieldsVariable, labelsVariable, setupMockedDataSource } from './mocks/CloudWatchDataSource'; import { setupMockedResourcesAPI } from './mocks/ResourcesAPI'; import { VariableQuery, VariableQueryType } from './types'; import { CloudWatchVariableSupport } from './variables'; const defaultQuery: VariableQuery = { queryType: VariableQueryType.Regions, namespace: 'foo', region: 'bar', metricName: '', dimensionKey: '', instanceID: '', attributeName: '', resourceType: '', refId: '', }; const mock = setupMockedDataSource({ variables: [labelsVariable, dimensionVariable, fieldsVariable] }); mock.datasource.resources.getRegions = jest.fn().mockResolvedValue([{ label: 'a', value: 'a' }]); mock.datasource.resources.getNamespaces = jest.fn().mockResolvedValue([{ label: 'b', value: 'b' }]); mock.datasource.resources.getAccounts = jest.fn().mockResolvedValue([]); const getMetrics = jest.fn().mockResolvedValue([{ label: 'c', value: 'c' }]); const getDimensionKeys = jest.fn().mockResolvedValue([{ label: 'd', value: 'd' }]); const getDimensionValues = jest.fn().mockResolvedValue([{ label: 'e', value: 'e' }]); const getEbsVolumeIds = jest.fn().mockResolvedValue([{ label: 'f', value: 'f' }]); const getEc2InstanceAttribute = jest.fn().mockResolvedValue([{ label: 'g', value: 'g' }]); const getResourceARNs = jest.fn().mockResolvedValue([{ label: 'h', value: 'h' }]); const getLogGroups = jest .fn() .mockResolvedValue([{ value: { arn: 'a', name: 'a' } }, { value: { arn: 'b', name: 'b' } }]); const variables = new CloudWatchVariableSupport(mock.datasource.resources); describe('variables', () => { it('should run regions', async () => { const result = await variables.execute({ ...defaultQuery }); expect(result).toEqual([{ text: 'a', value: 'a', expandable: true }]); }); it('should run namespaces', async () => { const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.Namespaces }); expect(result).toEqual([{ text: 'b', value: 'b', expandable: true }]); }); it('should run metrics', async () => { mock.datasource.resources.getMetrics = getMetrics; const query = { ...defaultQuery, queryType: VariableQueryType.Metrics, accountId: '123' }; const result = await variables.execute(query); expect(getMetrics).toHaveBeenCalledWith({ region: query.region, namespace: 'foo', accountId: query.accountId, }); expect(result).toEqual([{ text: 'c', value: 'c', expandable: true }]); }); it('should run dimension keys', async () => { mock.datasource.resources.getDimensionKeys = getDimensionKeys; const query = { ...defaultQuery, queryType: VariableQueryType.DimensionKeys, accountId: '123' }; const result = await variables.execute(query); expect(getDimensionKeys).toHaveBeenCalledWith({ region: query.region, namespace: query.namespace, accountId: query.accountId, }); expect(result).toEqual([{ text: 'd', value: 'd', expandable: true }]); }); describe('accounts', () => { it('should run accounts', async () => { const { api } = setupMockedResourcesAPI(); const getAccountMock = jest.fn().mockResolvedValue([]); api.getAccounts = getAccountMock; const variables = new CloudWatchVariableSupport(api); await variables.execute({ ...defaultQuery, queryType: VariableQueryType.Accounts }); expect(getAccountMock).toHaveBeenCalledWith({ region: defaultQuery.region }); }); it('should map accounts to metric find value and insert "all" option', async () => { const { api } = setupMockedResourcesAPI(); api.getAccounts = jest.fn().mockResolvedValue([{ id: '123', label: 'Account1' }]); const variables = new CloudWatchVariableSupport(api); const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.Accounts }); expect(result).toEqual([ { text: 'All', value: 'all', expandable: true }, { text: 'Account1', value: '123', expandable: true }, ]); }); }); describe('dimension values', () => { const query = { ...defaultQuery, queryType: VariableQueryType.DimensionValues, metricName: 'abc', dimensionKey: 'efg', dimensionFilters: { a: 'b' }, accountId: '123', }; beforeEach(() => { mock.datasource.resources.getDimensionValues = getDimensionValues; getDimensionValues.mockClear(); }); it('should not run if dimension key not set', async () => { const result = await variables.execute({ ...query, dimensionKey: '' }); expect(getDimensionValues).not.toBeCalled(); expect(result).toEqual([]); }); it('should not run if metric name not set', async () => { const result = await variables.execute({ ...query, metricName: '' }); expect(getDimensionValues).not.toBeCalled(); expect(result).toEqual([]); }); it('should run if values are set', async () => { const result = await variables.execute(query); expect(getDimensionValues).toHaveBeenCalledWith({ region: query.region, namespace: query.namespace, metricName: query.metricName, dimensionKey: query.dimensionKey, dimensionFilters: query.dimensionFilters, accountId: query.accountId, }); expect(result).toEqual([{ text: 'e', value: 'e', expandable: true }]); }); }); describe('EBS volume ids', () => { beforeEach(() => { mock.datasource.resources.getEbsVolumeIds = getEbsVolumeIds; getEbsVolumeIds.mockClear(); }); it('should not run if instance id not set', async () => { const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.EBSVolumeIDs }); expect(getEbsVolumeIds).not.toBeCalled(); expect(result).toEqual([]); }); it('should run if instance id set', async () => { const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.EBSVolumeIDs, instanceID: 'foo', }); expect(getEbsVolumeIds).toHaveBeenCalledWith(defaultQuery.region, 'foo'); expect(result).toEqual([{ text: 'f', value: 'f', expandable: true }]); }); }); describe('EC2 instance attributes', () => { const query = { ...defaultQuery, queryType: VariableQueryType.EC2InstanceAttributes, attributeName: 'abc', ec2Filters: { a: ['b'] }, }; beforeEach(() => { mock.datasource.resources.getEc2InstanceAttribute = getEc2InstanceAttribute; getEc2InstanceAttribute.mockClear(); }); it('should not run if instance id not set', async () => { const result = await variables.execute({ ...query, attributeName: '' }); expect(getEc2InstanceAttribute).not.toBeCalled(); expect(result).toEqual([]); }); it('should run if instance id set', async () => { const result = await variables.execute(query); expect(getEc2InstanceAttribute).toHaveBeenCalledWith(query.region, query.attributeName, { a: ['b'] }); expect(result).toEqual([{ text: 'g', value: 'g', expandable: true }]); }); }); describe('resource arns', () => { const query = { ...defaultQuery, queryType: VariableQueryType.ResourceArns, resourceType: 'abc', tags: { a: ['b'] }, }; beforeEach(() => { mock.datasource.resources.getResourceARNs = getResourceARNs; getResourceARNs.mockClear(); }); it('should not run if instance id not set', async () => { const result = await variables.execute({ ...query, resourceType: '' }); expect(getResourceARNs).not.toBeCalled(); expect(result).toEqual([]); }); it('should run if instance id set', async () => { const result = await variables.execute(query); expect(getResourceARNs).toHaveBeenCalledWith(query.region, query.resourceType, { a: ['b'] }); expect(result).toEqual([{ text: 'h', value: 'h', expandable: true }]); }); }); it('should run statistics', async () => { const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.Statistics }); expect(result).toEqual([ { text: 'Average', value: 'Average', expandable: true }, { text: 'Maximum', value: 'Maximum', expandable: true }, { text: 'Minimum', value: 'Minimum', expandable: true }, { text: 'Sum', value: 'Sum', expandable: true }, { text: 'SampleCount', value: 'SampleCount', expandable: true }, { text: 'IQM', value: 'IQM', expandable: true }, ]); }); describe('log groups', () => { beforeEach(() => { mock.datasource.resources.getLogGroups = getLogGroups; getLogGroups.mockClear(); }); it('should call describe log groups', async () => { const result = await variables.execute({ ...defaultQuery, queryType: VariableQueryType.LogGroups }); expect(result).toEqual([ { text: 'a', value: 'a', expandable: true }, { text: 'b', value: 'b', expandable: true }, ]); }); it('should replace variables', async () => { const query = { ...defaultQuery, queryType: VariableQueryType.LogGroups, logGroupPrefix: '$fields', accountId: '123', }; await variables.execute(query); expect(getLogGroups).toHaveBeenCalledWith({ region: query.region, logGroupNamePrefix: 'templatedField', listAllLogGroups: true, accountId: query.accountId, }); }); }); });