Add NetworkBoundaryBuilder argument to inject_network_boundaries.rs

Author: gabotechs

src/distributed_planner/inject_network_boundaries.rs

@@ -678,15 +678,16 @@ mod tests {
         "#;
         let annotated = sql_to_annotated(query).await;
         assert_snapshot!(annotated, @r"
-        SortPreservingMergeExec: task_count=Maximum(1)
-          NetworkCoalesceExec: task_count=Maximum(1)
-            SortExec: task_count=Desired(2)
-              ProjectionExec: task_count=Desired(2)
-                AggregateExec: task_count=Desired(2)
-                  NetworkShuffleExec: task_count=Desired(2)
-                    RepartitionExec: task_count=Desired(3)
-                      AggregateExec: task_count=Desired(3)
-                        DataSourceExec: task_count=Desired(3)
+        ProjectionExec: task_count=Maximum(1)
+          SortPreservingMergeExec: task_count=Maximum(1)
+            NetworkCoalesceExec: task_count=Maximum(1)
+              SortExec: task_count=Desired(2)
+                ProjectionExec: task_count=Desired(2)
+                  AggregateExec: task_count=Desired(2)
+                    NetworkShuffleExec: task_count=Desired(2)
+                      RepartitionExec: task_count=Desired(3)
+                        AggregateExec: task_count=Desired(3)
+                          DataSourceExec: task_count=Desired(3)
         ")
     }