test: add test case

Author: aojunhao123

tests/focus.test.tsx

@@ -0,0 +1,66 @@
+import { fireEvent, render } from '@testing-library/react';
+import InputNumber, { InputNumberRef } from 'rc-input-number';
+import { spyElementPrototypes } from 'rc-util/lib/test/domHook';
+import React from 'react';
+
+const getInputRef = () => {
+  const ref = React.createRef<InputNumberRef>();
+  render(<InputNumber ref={ref} defaultValue={12345} />);
+  return ref;
+};
+
+describe('InputNumber.Focus', () => {
+  let inputSpy: ReturnType<typeof spyElementPrototypes>;
+  let focus: ReturnType<typeof jest.fn>;
+  let setSelectionRange: ReturnType<typeof jest.fn>;
+
+  beforeEach(() => {
+    focus = jest.fn();
+    setSelectionRange = jest.fn();
+    inputSpy = spyElementPrototypes(HTMLInputElement, {
+      focus,
+      setSelectionRange,
+    });
+  });
+
+  afterEach(() => {
+    inputSpy.mockRestore();
+  });
+
+  it('start', () => {
+    const input = getInputRef();
+    input.current?.focus({ cursor: 'start' });
+
+    expect(focus).toHaveBeenCalled();
+    expect(setSelectionRange).toHaveBeenCalledWith(expect.anything(), 0, 0);
+  });
+
+  it('end', () => {
+    const input = getInputRef();
+    input.current?.focus({ cursor: 'end' });
+
+    expect(focus).toHaveBeenCalled();
+    expect(setSelectionRange).toHaveBeenCalledWith(expect.anything(), 5, 5);
+  });
+
+  it('all', () => {
+    const input = getInputRef();
+    input.current?.focus({ cursor: 'all' });
+
+    expect(focus).toHaveBeenCalled();
+    expect(setSelectionRange).toHaveBeenCalledWith(expect.anything(), 0, 5);
+  });
+
+  it('disabled should reset focus', () => {
+    const { container, rerender } = render(<InputNumber prefixCls="rc-input-number" />);
+    const input = container.querySelector('input')!;
+
+    fireEvent.focus(input);
+    expect(container.querySelector('.rc-input-number-focused')).toBeTruthy();
+
+    rerender(<InputNumber prefixCls="rc-input-number" disabled />);
+    fireEvent.blur(input);
+
+    expect(container.querySelector('.rc-input-number-focused')).toBeFalsy();
+  });
+});

tests/input.test.tsx

@@ -224,6 +224,16 @@ describe('InputNumber.Input', () => {
     expect(onChange).not.toHaveBeenCalled();
   });
 
+  it('input should work', () => {
+    const ref = React.createRef<InputNumberRef>();
+    const { container } = render(<InputNumber ref={ref} />);
+    const inputEl = container.querySelector('input')!;
+    const rootEl = container.querySelector('.rc-input-number')!;
+
+    expect(ref.current?.input).toBe(inputEl);
+    expect(ref.current?.nativeElement).toBe(rootEl);
+  });
+
   describe('nativeElement', () => {
     it('basic', () => {
       const ref = React.createRef<InputNumberRef>();