hastic-server/server/spec/utils_for_tests/detection_spans.ts

import { TEST_ANALYTIC_UNIT_ID } from './analytic_units';

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

import * as _ from 'lodash';

export type DetectionSpanOptions = { from: number, to: number, status: Detection.DetectionStatus };

export function buildSpans(options: DetectionSpanOptions[]): Detection.DetectionSpan[] {
  return options.map(option => {
    return new Detection.DetectionSpan(TEST_ANALYTIC_UNIT_ID, option.from, option.to, option.status);
  });
}

export async function insertSpans(options: DetectionSpanOptions[]): Promise<void> {
  const spansToInsert = buildSpans(options);
  const insertPromises = spansToInsert.map(async span => Detection.insertSpan(span));
  await Promise.all(insertPromises);
}

export function convertSpansToOptions(spans: Detection.DetectionSpan[]): DetectionSpanOptions[] {
  const spansOptions = spans.map(span => ({ from: span.from, to: span.to, status: span.status }));
  return _.sortBy(spansOptions, spanOptions => spanOptions.from);
}

export async function clearSpansDB(): Promise<void> {
  await Detection.clearSpans(TEST_ANALYTIC_UNIT_ID);
}