[VL] Fix streaming UTs by copying tags in EnsureLocalSortRequirements

When EnsureLocalSortRequirements adds a local SortExec to satisfy
required child ordering, the new SortExec node was missing tags from
the original child. This caused streaming test failures because
StateStoreWriter and other stateful operators rely on tags (e.g.,
isStatefulOperatorStreamingRestore) to propagate streaming execution
metadata.

Fix: Call newChild.copyTagsFrom(originalChild) to preserve tags when
wrapping a child plan with SortExec.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: Minni Mittal

gluten-substrait/src/main/scala/org/apache/gluten/extension/columnar/EnsureLocalSortRequirements.scala

@@ -38,6 +38,7 @@ object EnsureLocalSortRequirements extends Rule[SparkPlan] {
       requiredOrdering: Seq[SortOrder]): SparkPlan = {
     // FIXME: HeuristicTransform is costly. Re-applying it may cause performance issues.
     val newChild = SortExec(requiredOrdering, global = false, child = originalChild)
+    newChild.copyTagsFrom(originalChild)
     transform.apply(newChild)
   }
 

gluten-ut/spark40/src/test/scala/org/apache/gluten/utils/velox/VeloxTestSettings.scala

@@ -1200,15 +1200,15 @@ class VeloxTestSettings extends BackendTestSettings {
   enableSuite[GlutenFileStreamStressSuite]
   // TODO: 4.x enableSuite[GlutenFlatMapGroupsInPandasWithStateDistributionSuite]  // failures with GlutenPlugin
   enableSuite[GlutenFlatMapGroupsInPandasWithStateSuite]
-  // TODO: 4.x enableSuite[GlutenFlatMapGroupsWithStateDistributionSuite]
-  // TODO: 4.x enableSuite[GlutenFlatMapGroupsWithStateSuite]
+  enableSuite[GlutenFlatMapGroupsWithStateDistributionSuite]
+  enableSuite[GlutenFlatMapGroupsWithStateSuite]
   enableSuite[GlutenFlatMapGroupsWithStateWithInitialStateSuite]
   enableSuite[GlutenGroupStateSuite]
   enableSuite[GlutenLongOffsetSuite]
   enableSuite[GlutenMemorySourceStressSuite]
   // TODO: 4.x enableSuite[GlutenMultiStatefulOperatorsSuite]  // 2 failures
   enableSuite[GlutenReportSinkMetricsSuite]
-  // TODO: 4.x enableSuite[GlutenRocksDBStateStoreFlatMapGroupsWithStateSuite]
+  enableSuite[GlutenRocksDBStateStoreFlatMapGroupsWithStateSuite]
   // TODO: 4.x enableSuite[GlutenRocksDBStateStoreStreamingAggregationSuite]
   // TODO: 4.x enableSuite[GlutenRocksDBStateStoreStreamingDeduplicationSuite]
   // TODO: 4.x enableSuite[GlutenStreamSuite]

gluten-ut/spark41/src/test/scala/org/apache/gluten/utils/velox/VeloxTestSettings.scala

@@ -1214,15 +1214,15 @@ class VeloxTestSettings extends BackendTestSettings {
   enableSuite[GlutenFileStreamStressSuite]
   // TODO: 4.x enableSuite[GlutenFlatMapGroupsInPandasWithStateDistributionSuite]  // failures with GlutenPlugin
   enableSuite[GlutenFlatMapGroupsInPandasWithStateSuite]
-  // TODO: 4.x enableSuite[GlutenFlatMapGroupsWithStateDistributionSuite]
-  // TODO: 4.x enableSuite[GlutenFlatMapGroupsWithStateSuite]
+  enableSuite[GlutenFlatMapGroupsWithStateDistributionSuite]
+  enableSuite[GlutenFlatMapGroupsWithStateSuite]
   enableSuite[GlutenFlatMapGroupsWithStateWithInitialStateSuite]
   enableSuite[GlutenGroupStateSuite]
   enableSuite[GlutenLongOffsetSuite]
   enableSuite[GlutenMemorySourceStressSuite]
   // TODO: 4.x enableSuite[GlutenMultiStatefulOperatorsSuite]  // 2 failures
   enableSuite[GlutenReportSinkMetricsSuite]
-  // TODO: 4.x enableSuite[GlutenRocksDBStateStoreFlatMapGroupsWithStateSuite]
+  enableSuite[GlutenRocksDBStateStoreFlatMapGroupsWithStateSuite]
   // TODO: 4.x enableSuite[GlutenRocksDBStateStoreStreamingAggregationSuite]
   // TODO: 4.x enableSuite[GlutenRocksDBStateStoreStreamingDeduplicationSuite]
   // TODO: 4.x enableSuite[GlutenStreamSuite]