hastic-server/server/spec/models/detection_model.jest.ts

import { TEST_ANALYTIC_UNIT_ID } from '../utils_for_tests/analytic_units';
import { insertSpans, clearSpansDB, convertSpansToOptions } from '../utils_for_tests/detection_spans';

import * as Detection from '../../src/models/detection_model';

import * as _ from 'lodash';

const INITIAL_SPANS_CONFIGS = [
  { from: 1, to: 3, status: Detection.DetectionStatus.READY },
  { from: 4, to: 5, status: Detection.DetectionStatus.RUNNING }
];

beforeEach(async () => {
  await insertSpans(INITIAL_SPANS_CONFIGS);
});

afterEach(clearSpansDB);

describe('insertSpan', () => {
  it('should merge spans with the same status', async () => {
    await insertSpans([
      { from: 3, to: 5, status: Detection.DetectionStatus.READY }
    ]);

    const expectedSpans = [
      { from: 1, to: 5, status: Detection.DetectionStatus.READY }
    ];
    const spansInDB = await Detection.findMany(TEST_ANALYTIC_UNIT_ID, { });
    const spansOptions = convertSpansToOptions(spansInDB);
    expect(spansOptions).toEqual(expectedSpans);
  });


  it('should merge spans if existing span is inside the one being inserted', async () => {
    await insertSpans([
      { from: 1, to: 6, status: Detection.DetectionStatus.RUNNING }
    ]);

    const expectedSpans = [
      { from: 1, to: 6, status: Detection.DetectionStatus.RUNNING }
    ];
    const spansInDB = await Detection.findMany(TEST_ANALYTIC_UNIT_ID, {});
    const spansOptions = convertSpansToOptions(spansInDB);
    expect(spansOptions).toEqual(expectedSpans);
  });
});

describe('getIntersectedSpans', () => {
  it('should find all intersections with the inserted span', async () => {
    const testCases = [
      {
        from: 1, to: 5,
        expected: [
          { from: 1, to: 3, status: Detection.DetectionStatus.READY },
          { from: 3, to: 4, status: Detection.DetectionStatus.RUNNING }
        ]
      },
      { from: 4, to: 5, expected: [{ from: 3, to: 4, status: Detection.DetectionStatus.RUNNING }] },
      { from: 6, to: 7, expected: [] }
    ]

    for(let testCase of testCases) {
      const intersectedSpans = await Detection.getIntersectedSpans(TEST_ANALYTIC_UNIT_ID, testCase.from, testCase.to);
      const intersectedSpansOptions = convertSpansToOptions(intersectedSpans);
      expect(intersectedSpansOptions).toEqual(testCase.expected);
    }
  });
});