123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187
interface Hooks {  beforeAll: Array<() => void | Promise<void>>;  beforeEach: Array<() => void | Promise<void>>;  afterEach: Array<() => void | Promise<void>>;  afterAll: Array<() => void | Promise<void>>;
  waitingTests: number;  completedTests: number;
  onlyTests: number;  completedOnlyTests: number;}
interface StackItem extends Hooks {  name: string;}
interface GlobalContext extends Hooks {  stack: StackItem[];}
const globalContext: GlobalContext = {  stack: [],
  beforeAll: [],  beforeEach: [],  afterEach: [],  afterAll: [],
  waitingTests: 0,  completedTests: 0,
  onlyTests: 0,  completedOnlyTests: 0,};
function badArgs(): never {  throw new Error("Invalid test definition");}
async function callAll(fns: Array<() => void | Promise<void>>): Promise<void> {  await Promise.all(fns.map((fn) => fn()));}
/** * Registers a test for `deno test` while including grouping and hooks. */export function test(t: Deno.TestDefinition): void;/** * Registers a test for `deno test` while including grouping and hooks. */export function test(name: string, fn: () => void | Promise<void>): void;export function test(  t: Deno.TestDefinition | string,  testFn?: () => void | Promise<void>,): void {  // Extract args  const { name: testName, fn, ...opts } = typeof t === "object"    ? t    : (typeof testFn !== "undefined" ? { name: t, fn: testFn } : badArgs());
  // Set up waiting count.  if (!opts.ignore) {    globalContext.waitingTests++;    globalContext.stack.map((name) => name.waitingTests++);  }
  if (opts.only) {    globalContext.onlyTests++;    globalContext.stack.map((name) => name.onlyTests++);  }
  // Generate name.  let name = globalContext.stack.map(({ name: n }) => n);  name.push(testName);
  // Build hook stack.  const hooks: Hooks[] = [globalContext, ...globalContext.stack];  const revHooks: Hooks[] = [...hooks].reverse();
  Deno.test({    name: name.join(" > "),    async fn() {      // Before.      for (const { beforeAll, beforeEach, completedTests } of hooks) {        if (completedTests === 0) {          await callAll(beforeAll);        }
        await callAll(beforeEach);      }
      // Test.      await fn();      for (const hook of hooks) {        hook.completedTests++;
        if (opts.only) {          hook.completedOnlyTests++;        }      }
      // After.      for (        const {          afterAll,          afterEach,          waitingTests,          completedTests,          onlyTests,          completedOnlyTests,        } of revHooks      ) {        await callAll(afterEach);
        if (          waitingTests === completedTests ||          (onlyTests > 0 && onlyTests === completedOnlyTests)        ) {          await callAll(afterAll);        }      }    },    ...opts,  });}
/** * Creates an environment in which all tests and hooks are grouped together. */export function group(name: string, fn: () => void): void {  globalContext.stack.push({    name,
    beforeAll: [],    beforeEach: [],    afterEach: [],    afterAll: [],
    waitingTests: 0,    completedTests: 0,
    onlyTests: 0,    completedOnlyTests: 0,  });
  fn();
  globalContext.stack.pop();}
function getTopHooks(): Hooks {  if (globalContext.stack.length > 0) {    return globalContext.stack[globalContext.stack.length - 1];  } else {    return globalContext;  }}
/** * Adds a function to be called before all tests are run. */export function beforeAll(fn: () => void | Promise<void>): void {  getTopHooks().beforeAll.push(fn);}
/** * Adds a function to be called before each test runs. */export function beforeEach(fn: () => void | Promise<void>): void {  getTopHooks().beforeEach.push(fn);}
/** * Adds a function to be called after each test runs. */export function afterEach(fn: () => void | Promise<void>): void {  getTopHooks().afterEach.push(fn);}
/** * Adds a function to be called after all tests have run. */export function afterAll(fn: () => void | Promise<void>): void {  getTopHooks().afterAll.push(fn);}