energylink/public/app/features/annotations/specs/annotations_srv_specs.jest.ts

import { makeRegions, dedupAnnotations } from '../events_processing';

describe('Annotations', () => {
  describe('Annotations regions', () => {
    let testAnnotations: any[];

    beforeEach(() => {
      testAnnotations = [
        { id: 1, time: 1 },
        { id: 2, time: 2 },
        { id: 3, time: 3, regionId: 3 },
        { id: 4, time: 5, regionId: 3 },
        { id: 5, time: 4, regionId: 5 },
        { id: 6, time: 8, regionId: 5 },
      ];
    });

    it('should convert single region events to regions', () => {
      const range = { from: 0, to: 10 };
      const expectedAnnotations = [
        { id: 3, regionId: 3, isRegion: true, time: 3, timeEnd: 5 },
        { id: 5, regionId: 5, isRegion: true, time: 4, timeEnd: 8 },
        { id: 1, time: 1 },
        { id: 2, time: 2 },
      ];

      let regions = makeRegions(testAnnotations, { range: range });
      expect(regions).toEqual(expectedAnnotations);
    });

    it('should cut regions to current time range', () => {
      const range = { from: 0, to: 8 };
      testAnnotations = [{ id: 5, time: 4, regionId: 5 }];
      const expectedAnnotations = [
        { id: 5, regionId: 5, isRegion: true, time: 4, timeEnd: 7 },
      ];

      let regions = makeRegions(testAnnotations, { range: range });
      expect(regions).toEqual(expectedAnnotations);
    });
  });

  describe('Annotations deduplication', () => {
    it('should remove duplicated annotations', () => {
      const testAnnotations = [
        { id: 1, time: 1 },
        { id: 2, time: 2 },
        { id: 2, time: 2 },
        { id: 5, time: 5 },
        { id: 5, time: 5 },
      ];
      const expectedAnnotations = [
        { id: 1, time: 1 },
        { id: 2, time: 2 },
        { id: 5, time: 5 },
      ];

      let deduplicated = dedupAnnotations(testAnnotations);
      expect(deduplicated).toEqual(expectedAnnotations);
    });

    it('should leave non "panel-alert" event if present', () => {
      const testAnnotations = [
        { id: 1, time: 1 },
        { id: 2, time: 2 },
        { id: 2, time: 2, eventType: 'panel-alert' },
        { id: 5, time: 5 },
        { id: 5, time: 5 },
      ];
      const expectedAnnotations = [
        { id: 1, time: 1 },
        { id: 2, time: 2 },
        { id: 5, time: 5 },
      ];

      let deduplicated = dedupAnnotations(testAnnotations);
      expect(deduplicated).toEqual(expectedAnnotations);
    });
  });
});