argo/test/workflow_template_test.go

package test

import (
	"git.sxidc.com/go-tools/argo-api"
	"git.sxidc.com/go-tools/argo-api/client"
	"github.com/pkg/errors"
	"testing"
)

func TestWorkflowTemplate(t *testing.T) {
	err := argo.Init(kubeConfig)
	if err != nil {
		t.Fatalf("%+v\n", err)
	}

	defer argo.Destroy()

	err = argo.GetInstance().LintWorkflowTemplate(client.LintWorkflowTemplateParams{
		Namespace:          namespace,
		TemplateDefinition: templateYamlStr,
	})
	if err != nil {
		t.Fatalf("%+v\n", err)
	}

	err = argo.GetInstance().CreateWorkflowTemplate(client.CreateWorkflowTemplateParams{
		Namespace:          namespace,
		TemplateDefinition: templateYamlStr,
	})
	if err != nil {
		t.Fatalf("%+v\n", err)
	}

	defer func() {
		err := argo.GetInstance().DeleteWorkflowTemplate(client.DeleteWorkflowTemplateParams{
			Namespace: namespace,
			Name:      workflowTemplateName,
		})
		if err != nil {
			t.Fatalf("%+v\n", err)
		}
	}()

	wfs, err := argo.GetInstance().GetWorkflowTemplatesInNamespace(client.GetWorkflowTemplatesInNamespaceParams{
		Namespace: namespace,
	})
	if err != nil {
		t.Fatalf("%+v\n", err)
	}

	if len(wfs) != 1 {
		t.Fatalf("%+v\n", errors.Errorf("数量不一致: %v", len(wfs)))
	}

	wf, err := argo.GetInstance().GetWorkflowTemplate(client.GetWorkflowTemplateParams{
		Namespace: namespace,
		Name:      workflowTemplateName,
	})
	if err != nil {
		t.Fatalf("%+v\n", err)
	}

	if wf.Namespace != wfs[0].Namespace {
		t.Fatalf("%+v\n", errors.Errorf("命名空间不一致: one: %s other: %s", wf.Namespace, wfs[0].Namespace))
	}

	if wf.Name != wfs[0].Name {
		t.Fatalf("%+v\n", errors.Errorf("名称不一致: one: %s other: %s", wf.Name, wfs[0].Name))
	}

	err = argo.GetInstance().UpdateWorkflowTemplate(client.UpdateWorkflowTemplateParams{
		Namespace:        namespace,
		Name:             workflowTemplateName,
		WorkflowTemplate: wf,
	})
	if err != nil {
		t.Fatalf("%+v\n", err)
	}
}