prometheus

1
import { DataTableProps } from './DataTable';
2
import { isHeatmapData } from './GraphHeatmapHelpers';
3

4
describe('GraphHeatmapHelpers', () => {
5
  it('isHeatmapData should return false for scalar and string resultType', () => {
6
    let data = {
7
      resultType: 'scalar',
8
      result: [1703091180.125, '1703091180.125'],
9
    } as DataTableProps['data'];
10
    expect(isHeatmapData(data)).toBe(false);
11

12
    data = {
13
      resultType: 'string',
14
      result: [1704305680.332, '2504'],
15
    } as DataTableProps['data'];
16
    expect(isHeatmapData(data)).toBe(false);
17
  });
18

19
  it('isHeatmapData should return false for a vector and matrix if length < 2', () => {
20
    let data = {
21
      resultType: 'vector',
22
      result: [
23
        {
24
          metric: {
25
            __name__: 'my_gauge',
26
            job: 'target',
27
          },
28
          value: [1703091180.683, '6'],
29
        },
30
      ],
31
    } as DataTableProps['data'];
32
    expect(isHeatmapData(data)).toBe(false);
33

34
    data = {
35
      resultType: 'matrix',
36
      result: [
37
        {
38
          metric: {},
39
          values: [[1703091180.683, '6']],
40
        },
41
      ],
42
    } as DataTableProps['data'];
43
    expect(isHeatmapData(data)).toBe(false);
44
  });
45

46
  it('isHeatmapData should return true for valid heatmap data', () => {
47
    const data = {
48
      resultType: 'matrix',
49
      result: [
50
        {
51
          metric: {
52
            le: '100',
53
          },
54
          values: [[1703091180.683, '6']],
55
        },
56
        {
57
          metric: {
58
            le: '1000',
59
          },
60
          values: [[1703091190.683, '6.1']],
61
        },
62
      ],
63
    } as DataTableProps['data'];
64
    expect(isHeatmapData(data)).toBe(true);
65
  });
66
});
67