update controlled input to reflect changes in controlled value

.changeset/violet-hornets-change.md

@@ -0,0 +1,5 @@
+---
+"@aws-amplify/ui-react": patch
+---
+
+update StepperField and CheckboxField to update the visual display when a controlled value is changed

packages/react/src/primitives/Checkbox/Checkbox.tsx

@@ -35,10 +35,15 @@ const CheckboxPrimitive: Primitive<CheckboxProps, 'input'> = (
   // controlled way should always override uncontrolled way
   const initialChecked = checked !== undefined ? checked : defaultChecked;
 
-  const { dataChecked, dataFocus, onBlur, onChange, onFocus } = useCheckbox(
-    initialChecked,
-    onChangeProp
-  );
+  const { dataChecked, dataFocus, onBlur, onChange, onFocus, setDataChecked } =
+    useCheckbox(initialChecked, onChangeProp);
+
+  React.useEffect(() => {
+    const isControlled = checked !== undefined;
+    if (isControlled && checked !== dataChecked) {
+      setDataChecked(checked);
+    }
+  }, [checked, dataChecked, setDataChecked]);
 
   const buttonTestId = useTestId(testId, ComponentClassNames.CheckboxButton);
   const iconTestId = useTestId(testId, ComponentClassNames.CheckboxIcon);

packages/react/src/primitives/Checkbox/__tests__/Checkbox.test.tsx

@@ -116,6 +116,21 @@ describe('Checkbox test suite', () => {
   });
 
   describe('Button test suite', () => {
+    let updateCheckedFunction;
+    const ControlledCheckbox = () => {
+      const [checked, setChecked] = React.useState(false);
+      updateCheckedFunction = setChecked;
+      return (
+        <Checkbox
+          {...basicProps}
+          onChange={(event) => {
+            setChecked(event.target.checked);
+          }}
+          checked={checked}
+        />
+      );
+    };
+
     it('should render basic props correctly', async () => {
       render(getCheckbox({ ...basicProps }));
 
@@ -127,6 +142,23 @@ describe('Checkbox test suite', () => {
       expect(button).toHaveAttribute('data-focus', 'false');
       expect(button).toHaveClass(ComponentClassNames.CheckboxButton);
     });
+
+    it('should update the checked button with a change to the controlled value', async () => {
+      render(<ControlledCheckbox />);
+
+      let button = await screen.findByTestId(
+        `${basicProps.testId}-${ComponentClassNames.CheckboxButton}`
+      );
+      expect(button).toHaveAttribute('data-checked', 'false');
+      expect(button).not.toHaveAttribute('data-disabled');
+      expect(button).toHaveAttribute('data-focus', 'false');
+      expect(button).toHaveClass(ComponentClassNames.CheckboxButton);
+      updateCheckedFunction(true);
+      button = await screen.findByTestId(
+        `${basicProps.testId}-${ComponentClassNames.CheckboxButton}`
+      );
+      expect(button).toHaveAttribute('data-checked', 'true');
+    });
   });
 
   describe('Icon test suite', () => {

packages/react/src/primitives/Checkbox/useCheckbox.ts

@@ -25,5 +25,5 @@ export const useCheckbox = (
     setDataFocus(false);
   };
 
-  return { dataChecked, dataFocus, onBlur, onChange, onFocus };
+  return { dataChecked, dataFocus, onBlur, onChange, onFocus, setDataChecked };
 };

packages/react/src/primitives/StepperField/StepperField.tsx

@@ -59,10 +59,18 @@ const StepperFieldPrimitive: Primitive<StepperFieldProps, 'input'> = (
     handleOnBlur,
     handleOnChange,
     handleOnWheel,
+    setInputValue,
     shouldDisableDecreaseButton,
     shouldDisableIncreaseButton,
   } = useStepper(props);
 
+  React.useEffect(() => {
+    const isControlled = controlledValue !== undefined;
+    if (isControlled) {
+      setInputValue(controlledValue);
+    }
+  }, [controlledValue, setInputValue]);
+
   return (
     <Flex
       className={classNames(

packages/react/src/primitives/StepperField/__tests__/StepperField.test.tsx

@@ -70,9 +70,11 @@ describe('StepperField: ', () => {
 
   describe('Input field', () => {
     const label = 'stepper';
+    let updateValueFunction;
 
     const ControlledStepper = () => {
       const [value, setValue] = React.useState(0);
+      updateValueFunction = setValue;
       return (
         <StepperField
           label={label}
@@ -202,6 +204,15 @@ describe('StepperField: ', () => {
       expect(onBlur).toHaveBeenCalled();
       expect(onWheel).toHaveBeenCalled();
     });
+
+    it('should update the value correctly(controlled)', async () => {
+      render(<ControlledStepper />);
+      let stepperInput = await screen.findByLabelText(label);
+      expect(stepperInput).toHaveValue(0);
+      updateValueFunction(8);
+      stepperInput = await screen.findByLabelText(label);
+      expect(stepperInput).toHaveValue(8);
+    });
   });
 
   describe('Increase/Decrease button', () => {

packages/react/src/primitives/StepperField/useStepper.ts

@@ -150,6 +150,7 @@ export const useStepper = ({
     handleOnBlur,
     handleOnChange,
     handleOnWheel,
+    setInputValue,
     shouldDisableDecreaseButton,
     shouldDisableIncreaseButton,
   };