test.go

package test

import (
	"io/ioutil"
	"path/filepath"
	"runtime"

	wfv1 "github.com/argoproj/argo/pkg/apis/workflow/v1alpha1"
	"github.com/ghodss/yaml"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
)

var (
	testDir string
)

func init() {
	_, filename, _, ok := runtime.Caller(0)
	if !ok {
		panic("could not determine test directory")
	}
	testDir = filepath.Dir(filename)
}

// LoadE2EWorkflow returns a test workflow by it's path
func LoadE2EWorkflow(path string) *wfv1.Workflow {
	yamlBytes, err := ioutil.ReadFile(filepath.Join(testDir, "e2e", path))
	if err != nil {
		panic(err)
	}
	return LoadWorkflowFromBytes(yamlBytes)
}

// LoadTestWorkflow returns a workflow relative to the test file
func LoadTestWorkflow(path string) *wfv1.Workflow {
	yamlBytes, err := ioutil.ReadFile(path)
	if err != nil {
		panic(err)
	}
	return LoadWorkflowFromBytes(yamlBytes)
}

// LoadWorkflowFromBytes returns a workflow unmarshalled from an yaml byte array
func LoadWorkflowFromBytes(yamlBytes []byte) *wfv1.Workflow {
	var wf wfv1.Workflow
	err := yaml.Unmarshal(yamlBytes, &wf)
	if err != nil {
		panic(err)
	}
	return &wf
}

// LoadUnstructuredFromBytes returns an Unstructured unmarshalled from an yaml byte array
func LoadUnstructuredFromBytes(yamlBytes []byte) *unstructured.Unstructured {
	var un unstructured.Unstructured
	err := yaml.Unmarshal(yamlBytes, &un)
	if err != nil {
		panic(err)
	}
	return &un
}