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.

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