troubleshoot/pkg/supportbundle/supportbundle_test.go

package supportbundle

import (
	"reflect"
	"testing"

	troubleshootv1beta2 "github.com/replicatedhq/troubleshoot/pkg/apis/troubleshoot/v1beta2"
)

func Test_LoadAndConcatSpec(t *testing.T) {

	bundle1doc, err := LoadSupportBundleSpec("test/supportbundle1.yaml")
	if err != nil {
		t.Error("couldn't load bundle1 from file")
	}

	bundle2doc, err := LoadSupportBundleSpec("test/supportbundle2.yaml")
	if err != nil {
		t.Error("couldn't load bundle2 from file")
	}

	bundle1, err := ParseSupportBundleFromDoc(bundle1doc)
	if err != nil {
		t.Error("couldn't parse bundle 1")
	}

	bundle2, err := ParseSupportBundleFromDoc(bundle2doc)
	if err != nil {
		t.Error("couldn't parse bundle 2")
	}

	fulldoc, err := LoadSupportBundleSpec("test/completebundle.yaml")
	if err != nil {
		t.Error("couldn't load full bundle from file")
	}

	fullbundle, err := ParseSupportBundleFromDoc(fulldoc)
	if err != nil {
		t.Error("couldn't parse full bundle")
	}

	bundle3 := ConcatSpec(bundle1, bundle2)

	if reflect.DeepEqual(fullbundle, bundle3) == false {
		t.Error("Full bundle and concatenated bundle are not the same.")
	}

}

func Test_LoadAndConcatSpec_WithNil(t *testing.T) {
	var bundle *troubleshootv1beta2.SupportBundle
	// both function arguments are nil
	bundle4 := ConcatSpec(bundle, bundle)
	if reflect.DeepEqual(bundle4, (*troubleshootv1beta2.SupportBundle)(nil)) == false {
		t.Error("concatenating nil pointer with nil pointer has error.")
	}

	fulldoc, _ := LoadSupportBundleSpec("test/completebundle.yaml")
	bundle, _ = ParseSupportBundleFromDoc(fulldoc)

	// targetBundle is nil pointer
	bundle5 := ConcatSpec((*troubleshootv1beta2.SupportBundle)(nil), bundle)
	if reflect.DeepEqual(bundle5, bundle) == false {
		t.Error("concatenating targetBundle of nil pointer has error.")
	}

	// sourceBundle is nil pointer
	bundle6 := ConcatSpec(bundle, (*troubleshootv1beta2.SupportBundle)(nil))
	if reflect.DeepEqual(bundle6, bundle) == false {
		t.Error("concatenating sourceBundle of nil pointer has error.")
	}
}