import { Reporter, ReportItemWithContext, Transport, Collector } from '..';
import { sleep } from '../../../../tests/shared';

export class TestReporter extends Reporter<Record<string, any>> {}

export class TestCollector extends Collector<Record<string, any>, Reporter<any>> {
  public reporter?: Reporter<Record<string, any>>;
}

export class TestTransport extends Transport<Record<string, any>, Reporter<any>> {
  /** 待消费上报项（测试用） */
  recordsToConsume: ReportItemWithContext<Record<string, any>>[] = [];

  /** 取得当前待消费上报项（测试用） */
  consume() {
    const records = [...this.recordsToConsume];
    this.recordsToConsume = [];
    return records;
  }

  send(records: ReportItemWithContext<Record<string, any>>[]) {
    this.recordsToConsume.push(...records);
  }
}

export class TestAwaitedTransport extends TestTransport {
  constructor(protected timeout = 1000) {
    super();
  }
  async send(records: ReportItemWithContext<Record<string, any>>[]) {
    await sleep(this.timeout);

    this.recordsToConsume.push(...records);
  }
}