import type {
  beforeEach as jestBeforeEach,
  afterEach as jestAfterEach,
} from '@jest/globals';
import { TestContext } from 'gwt-runner';

type Callback<T> = (this: T) => any;

export default (
  beforeEach: typeof jestBeforeEach,
  afterEach: typeof jestAfterEach,
) => (
  <T>(
    before: Callback<T>,
    after?: Callback<T>,
  ) => {
    beforeEach(async () => {
      TestContext.createContext();

      await (before.bind(TestContext.context as T) as any)();
    });

    afterEach(async () => {
      if (after) {
        await (after.bind(TestContext.context as T) as any)();
      }

      TestContext.releaseContext();
    });
  }
);