123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189
import { interpret } from '../src/interpreter';import { createMachine } from '../src';import { waitFor } from '../src/waitFor';
describe('waitFor', () => {  it('should wait for a condition to be true and return the emitted value', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: { NEXT: 'b' }        },        b: {}      }    });
    const service = interpret(machine).start();
    setTimeout(() => service.send('NEXT'), 10);
    const state = await waitFor(service, (s) => s.matches('b'));
    expect(state.value).toEqual('b');  });
  it('should throw an error after a timeout', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: { NEXT: 'b' }        },        b: {          on: { NEXT: 'c' }        },        c: {}      }    });
    const service = interpret(machine).start();
    try {      await waitFor(service, (state) => state.matches('c'), { timeout: 10 });    } catch (e) {      expect(e).toBeInstanceOf(Error);    }  });
  it('should not reject immediately when passing Infinity as timeout', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: { NEXT: 'b' }        },        b: {          on: { NEXT: 'c' }        },        c: {}      }    });    const service = interpret(machine).start();    const result = await Promise.race([      waitFor(service, (state) => state.matches('c'), {        timeout: Infinity      }),      new Promise((res) => setTimeout(res, 10)).then(() => 'timeout')    ]);
    expect(result).toBe('timeout');    service.stop();  });
  it('should throw an error when reaching a final state that does not match the predicate', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: { NEXT: 'b' }        },        b: {          type: 'final'        }      }    });
    const service = interpret(machine).start();
    setTimeout(() => {      service.send('NEXT');    }, 10);
    await expect(      waitFor(service, (state) => state.matches('never'))    ).rejects.toMatchInlineSnapshot(      `[Error: Actor terminated without satisfying predicate]`    );  });
  it('should resolve correctly when the predicate immediately matches the current state', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {}      }    });
    const service = interpret(machine).start();
    await expect(      waitFor(service, (state) => state.matches('a'))    ).resolves.toHaveProperty('value', 'a');  });
  it('should internally unsubscribe when the predicate immediately matches the current state', async () => {    let count = 0;    const machine = createMachine({      initial: 'a',      states: {        a: {          on: {            NEXT: 'b'          }        },        b: {}      }    });
    const service = interpret(machine).start();
    await waitFor(service, (state) => {      count++;      return state.matches('a');    });
    service.send({ type: 'NEXT' });
    expect(count).toBe(1);  });
  it('should immediately resolve for an actor in its final state that matches the predicate', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: {            NEXT: 'b'          }        },        b: {          type: 'final'        }      }    });
    const service = interpret(machine).start();    service.send({ type: 'NEXT' });
    await expect(      waitFor(service, (state) => state.matches('b'))    ).resolves.toHaveProperty('value', 'b');  });
  it('should immediately reject for an actor in its final state that does not match the predicate', async () => {    const machine = createMachine({      initial: 'a',      states: {        a: {          on: {            NEXT: 'b'          }        },        b: {          type: 'final'        }      }    });
    const service = interpret(machine).start();    service.send({ type: 'NEXT' });
    await expect(      waitFor(service, (state) => state.matches('a'))    ).rejects.toMatchInlineSnapshot(      `[Error: Actor terminated without satisfying predicate]`    );  });});