return node spec config in node outputs (#1753)

* transform raw TOML for all override mode

* changeset

Author: skudasov

framework/.changeset/v0.6.9.md

@@ -0,0 +1 @@
+- Copy node specs in mode all to make tests easier

framework/config.go

@@ -88,6 +88,11 @@ func mergeInputs[T any]() (*T, error) {
 			fmt.Println(string(data))
 		}
 
+		data, err = transformAllOverrideModeForNodeSet(data)
+		if err != nil {
+			return nil, fmt.Errorf("error transforming node specs: %w", err)
+		}
+
 		decoder := toml.NewDecoder(strings.NewReader(string(data)))
 		decoder.DisallowUnknownFields()
 
@@ -141,6 +146,41 @@ func validate(s interface{}) error {
 	return nil
 }
 
+// transformAllOverrideModeForNodeSet we need this function so the test logic can be the same in both "each" and "all" override modes
+// we can't do UnmarshalTOML or UnmarshalText because our TOML library do not support it
+func transformAllOverrideModeForNodeSet(data []byte) ([]byte, error) {
+	var config map[string]interface{}
+	if err := toml.Unmarshal(data, &config); err != nil {
+		return nil, err
+	}
+	nodeset, ok := config["nodeset"].(map[string]interface{})
+	if !ok {
+		return data, nil
+	}
+	if nodeset["override_mode"] != "all" {
+		return data, nil
+	}
+	nodes, ok := nodeset["nodes"].(int64)
+	if !ok || nodes <= 0 {
+		return nil, fmt.Errorf("invalid nodes count")
+	}
+	specs, ok := nodeset["node_specs"].([]interface{})
+	if !ok || len(specs) == 0 {
+		return nil, fmt.Errorf("node_specs must be provided")
+	}
+	firstSpec := specs[0].(map[string]interface{})
+	expanded := make([]interface{}, nodes)
+	for i := range expanded {
+		newSpec := make(map[string]interface{})
+		for k, v := range firstSpec {
+			newSpec[k] = v
+		}
+		expanded[i] = newSpec
+	}
+	nodeset["node_specs"] = expanded
+	return toml.Marshal(config)
+}
+
 func Load[X any](t *testing.T) (*X, error) {
 	input, err := mergeInputs[X]()
 	if err != nil {