import { describe, it, expect, beforeEach, test } from 'vitest';

import { TTestSuite } from '../testUtils/TVestMock';
import * as vest from '../vest';

let suite: TTestSuite;

beforeEach(() => {
  suite = genSuite();
});

describe('only', () => {
  it('Should only count included fields', () => {
    const res = suite.run({
      only: ['field_1', 'field_2'],
    });

    expect(res.tests.field_1.testCount).toBe(1);
    expect(res.tests.field_2.testCount).toBe(1);
    expect(res.tests.field_3.testCount).toBe(0);
    expect(res.tests.field_4.testCount).toBe(0);
    expect(res.tests.field_5.testCount).toBe(0);
  });
  it('Should only count included field', () => {
    const res = suite.run({
      only: 'field_1',
    });

    expect(res.tests.field_1.testCount).toBe(1);
    expect(res.tests.field_2.testCount).toBe(0);
    expect(res.tests.field_3.testCount).toBe(0);
    expect(res.tests.field_4.testCount).toBe(0);
    expect(res.tests.field_5.testCount).toBe(0);
  });
});
describe('skip', () => {
  it('Should count all but excluded fields', () => {
    const res = suite.run({
      skip: ['field_1', 'field_2'],
    });

    expect(res.tests.field_1.testCount).toBe(0);
    expect(res.tests.field_2.testCount).toBe(0);
    expect(res.tests.field_3.testCount).toBe(1);
    expect(res.tests.field_4.testCount).toBe(1);
    expect(res.tests.field_5.testCount).toBe(1);
  });

  it('Should count all but excluded field', () => {
    const res = suite.run({
      skip: 'field_1',
    });

    expect(res.tests.field_1.testCount).toBe(0);
    expect(res.tests.field_2.testCount).toBe(1);
    expect(res.tests.field_3.testCount).toBe(1);
    expect(res.tests.field_4.testCount).toBe(1);
    expect(res.tests.field_5.testCount).toBe(1);
  });
});

describe('Combined', () => {
  test('First declaration wins', () => {
    const res = suite.run({
      skip: ['field_1'],
      only: ['field_1', 'field_2', 'field_3'],
      skip_last: 'field_3',
    });

    expect(res.tests.field_1.testCount).toBe(0);
    expect(res.tests.field_2.testCount).toBe(1);
    expect(res.tests.field_3.testCount).toBe(1);
  });
});

function genSuite() {
  return vest.create((exclusion: Record<string, string | string[]> = {}) => {
    vest.skip(exclusion?.skip);
    vest.only(exclusion?.only);
    vest.skip(exclusion?.skip_last);

    vest.test('field_1', 'msg', () => {});
    vest.test('field_2', 'msg', () => {});
    vest.test('field_3', 'msg', () => {});
    vest.test('field_4', 'msg', () => {});
    vest.test('field_5', 'msg', () => {});
  });
}