pass trait when shadow node is cloned from update state (#44355)

Summary:
Pull Request resolved: #44355

changelog: [internal]

Reviewed By: cipolleschi

Differential Revision: D56765820

fbshipit-source-id: 5bfd2231883c25a0d9149bb07d529da796fd921e

packages/react-native/ReactCommon/react/renderer/uimanager/UIManager.cpp

@@ -366,6 +366,9 @@ void UIManager::updateState(const StateUpdate& stateUpdate) const {
   auto& callback = stateUpdate.callback;
   auto& family = stateUpdate.family;
   auto& componentDescriptor = family->getComponentDescriptor();
+  auto clonedByNativeStateTraits = ShadowNodeTraits();
+  clonedByNativeStateTraits.set(
+      ShadowNodeTraits::Trait::ClonedByNativeStateUpdate);
 
   shadowTreeRegistry_.visit(
       family->getSurfaceId(), [&](const ShadowTree& shadowTree) {
@@ -374,7 +377,8 @@ void UIManager::updateState(const StateUpdate& stateUpdate) const {
               auto isValid = true;
 
               auto rootNode = oldRootShadowNode.cloneTree(
-                  *family, [&](const ShadowNode& oldShadowNode) {
+                  *family,
+                  [&](const ShadowNode& oldShadowNode) {
                     auto newData =
                         callback(oldShadowNode.getState()->getDataPointer());
 
@@ -387,13 +391,13 @@ void UIManager::updateState(const StateUpdate& stateUpdate) const {
                     auto newState =
                         componentDescriptor.createState(*family, newData);
 
-                    return oldShadowNode.clone({
-                        /* .props = */ ShadowNodeFragment::propsPlaceholder(),
-                        /* .children = */
-                        ShadowNodeFragment::childrenPlaceholder(),
-                        /* .state = */ newState,
-                    });
-                  });
+                    return oldShadowNode.clone(
+                        {.props = ShadowNodeFragment::propsPlaceholder(),
+                         .children = ShadowNodeFragment::childrenPlaceholder(),
+                         .state = newState,
+                         .traits = clonedByNativeStateTraits});
+                  },
+                  clonedByNativeStateTraits);
 
               return isValid
                   ? std::static_pointer_cast<RootShadowNode>(rootNode)