| // Copyright Istio Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package manifest |
| |
| import ( |
| "bytes" |
| "fmt" |
| "os" |
| "path/filepath" |
| "testing" |
| ) |
| |
| import ( |
| "github.com/apache/dubbo-go-pixiu/operator/pkg/util" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/env" |
| ) |
| |
| func TestReadLayeredYAMLs(t *testing.T) { |
| testDataDir := filepath.Join(env.IstioSrc, "operator/pkg/util/testdata/yaml") |
| tests := []struct { |
| name string |
| overlays []string |
| wantErr bool |
| stdin bool |
| }{ |
| { |
| name: "layer1", |
| overlays: []string{"yaml_layer1"}, |
| wantErr: false, |
| }, |
| { |
| name: "layer1_stdin", |
| overlays: []string{"yaml_layer1"}, |
| wantErr: false, |
| stdin: true, |
| }, |
| { |
| name: "layer1_2", |
| overlays: []string{"yaml_layer1", "yaml_layer2"}, |
| wantErr: false, |
| }, |
| { |
| name: "layer1_2_3", |
| overlays: []string{"yaml_layer1", "yaml_layer2", "yaml_layer3"}, |
| wantErr: false, |
| }, |
| } |
| for _, tt := range tests { |
| t.Run(fmt.Sprintf("%s stdin=%v", tt.name, tt.stdin), func(t *testing.T) { |
| inDir := filepath.Join(testDataDir, "input") |
| outPath := filepath.Join(testDataDir, "output", tt.name+".yaml") |
| wantBytes, err := os.ReadFile(outPath) |
| want := string(wantBytes) |
| if err != nil { |
| t.Errorf("os.ReadFile() error = %v, filename: %v", err, outPath) |
| } |
| |
| stdinReader := &bytes.Buffer{} |
| |
| var filenames []string |
| for _, ol := range tt.overlays { |
| filename := filepath.Join(inDir, ol+".yaml") |
| if tt.stdin { |
| b, err := os.ReadFile(filename) |
| if err != nil { |
| t.Fatalf("os.ReadFile() error = %v, filenaem: %v", err, filename) |
| } |
| if _, err := stdinReader.Write(b); err != nil { |
| t.Fatalf("failed to populate fake sdtin") |
| } |
| filenames = append(filenames, "-") |
| } else { |
| filenames = append(filenames, filename) |
| } |
| } |
| got, err := readLayeredYAMLs(filenames, stdinReader) |
| if (err != nil) != tt.wantErr { |
| t.Errorf("ReadLayeredYAMLs() error = %v, wantErr %v", err, tt.wantErr) |
| return |
| } |
| |
| diff := util.YAMLDiff(got, want) |
| if diff != "" { |
| t.Errorf("ReadLayeredYAMLs() got:\n%s\nwant:\n%s\ndiff:\n%s", got, want, diff) |
| } |
| }) |
| } |
| } |