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); } }); });