rive-react/test/useStateMachine.test.tsx

import { mocked } from 'jest-mock';
import { renderHook } from '@testing-library/react-hooks';

import useStateMachineInput from '../src/hooks/useStateMachineInput';
import {Rive, StateMachineInput} from '@rive-app/webgl';

jest.mock('@rive-app/webgl', () => ({
  Rive: jest.fn().mockImplementation(() => ({
    on: jest.fn(),
    stop: jest.fn(),
  })),
  Layout: jest.fn(),
  Fit: {
    Cover: 'cover',
  },
  Alignment: {
    Center: 'center',
  },
  EventType: {
    Load: 'load',
  },
  StateMachineInputType: {
    Number: 1,
    Boolean: 2,
    Trigger: 3,
  },
}));

describe('useStateMachineInput', () => {
  it('returns null if there is null rive object passed', () => {
    const { result } = renderHook(() => useStateMachineInput(null));
    expect(result.current).toBeNull();
  });

  it('returns null if there is no state machine name', () => {
    const riveMock = {};
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, '', 'testInput'));
    expect(result.current).toBeNull();
  });

  it('returns null if there is no state machine input name', () => {
    const riveMock = {};
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, 'smName', ''));
    expect(result.current).toBeNull();
  });

  it('returns null if there are no inputs for the state machine', () => {
    const riveMock = {
      // eslint-disable-next-line @typescript-eslint/no-unused-vars
      stateMachineInputs: (_: string) => [] as StateMachineInput[],
    };
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, 'smName', ''));
    expect(result.current).toBeNull();
  });

  it('returns null if the input has no association to the inputs of the state machine', () => {
    const smInput = {
      name: 'boolInput',
    } as StateMachineInput;
    const riveMock = {
      // eslint-disable-next-line @typescript-eslint/no-unused-vars
      stateMachineInputs: (_: string) => [smInput] as StateMachineInput[],
    };
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, 'smName', 'numInput'));
    expect(result.current).toBeNull();
  });

  it('returns a selected input if the input requested is part of the state machine', () => {
    const smInput = {
      name: 'boolInput',
    } as StateMachineInput;
    const riveMock = {
      // eslint-disable-next-line @typescript-eslint/no-unused-vars
      stateMachineInputs: (_: string) => [smInput] as StateMachineInput[],
    };
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, 'smName', 'boolInput'));
    expect(result.current).toBe(smInput);
  });

  it('returns a selected input with an initial value if the input requested is part of the state machine', () => {
    const smInput = {
      name: 'boolInput',
      value: false,
    } as StateMachineInput;
    const riveMock = {
      // eslint-disable-next-line @typescript-eslint/no-unused-vars
      stateMachineInputs: (_: string) => [smInput] as StateMachineInput[],
    };
    mocked(Rive).mockImplementation(() => riveMock as Rive);

    const { result } = renderHook(() => useStateMachineInput(riveMock as Rive, 'smName', 'boolInput', true));
    expect(result.current).toStrictEqual({
      ...smInput,
      value: true,
    });
  });
});