You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
3.2 KiB
93 lines
3.2 KiB
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'; |
|
|
|
afterEach(clearSpansDB); |
|
|
|
describe('insertSpan', () => { |
|
it('should merge spans with the same status', async () => { |
|
/* |
|
* Config for test |
|
* insert -- what we want to insert in our test database |
|
* expectedAfterInsertion -- expected database state after insertion |
|
*/ |
|
const insertSteps = [ |
|
{ |
|
insert: [ |
|
{ from: 1, to: 3, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 5, status: Detection.DetectionStatus.RUNNING } |
|
], |
|
expectedAfterInsertion: [ |
|
{ from: 1, to: 3, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 5, status: Detection.DetectionStatus.RUNNING } |
|
] |
|
}, |
|
{ |
|
insert: [ { from: 5, to: 9, status: Detection.DetectionStatus.RUNNING } ], |
|
expectedAfterInsertion: [ |
|
{ from: 1, to: 3, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 9, status: Detection.DetectionStatus.RUNNING } |
|
] |
|
}, |
|
{ |
|
insert: [ { from: 2, to: 5, status: Detection.DetectionStatus.READY } ], |
|
expectedAfterInsertion: [ |
|
{ from: 1, to: 5, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 9, status: Detection.DetectionStatus.RUNNING } |
|
] |
|
}, |
|
]; |
|
|
|
for(let step of insertSteps) { |
|
await insertSpans(step.insert); |
|
const spansInDB = await Detection.findMany(TEST_ANALYTIC_UNIT_ID, {}); |
|
const spansOptions = convertSpansToOptions(spansInDB); |
|
expect(spansOptions).toEqual(step.expectedAfterInsertion); |
|
} |
|
}); |
|
|
|
|
|
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 () => { |
|
await insertSpans([ |
|
{ from: 1, to: 3, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 5, status: Detection.DetectionStatus.RUNNING } |
|
]); |
|
|
|
const testCases = [ |
|
{ |
|
from: 1, to: 5, |
|
expected: [ |
|
{ from: 1, to: 3, status: Detection.DetectionStatus.READY }, |
|
{ from: 4, to: 5, status: Detection.DetectionStatus.RUNNING } |
|
] |
|
}, |
|
{ from: 4, to: 5, expected: [{ from: 4, to: 5, 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); |
|
} |
|
}); |
|
}); |
|
|
|
|