airship/airshipctl
Code Issues Proposed changes

Convert workflow_init_test to a proper table test

Browse Source
This commit is contained in:
Ian Howell 2019-05-29 15:34:38 -05:00
parent 3dae793f70
commit 3b31b47158
1 changed files with 42 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
cmd/workflow/workflow_init_test.go
View File

@ -3,7 +3,13 @@ package workflow_test
import (
"testing"
v1beta2 "k8s.io/api/apps/v1beta2"
v1 "k8s.io/api/core/v1"
rbacv1 "k8s.io/api/rbac/v1"
apixv1beta1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1beta1"
apixv1beta1fake "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset/fake"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
kubefake "k8s.io/client-go/kubernetes/fake"
"github.com/ian-howell/airshipctl/cmd"
@ -22,27 +28,43 @@ func TestWorkflowInit(t *testing.T) {
workflowRoot.AddCommand(workflow.NewWorkflowInitCommand(settings))
rootCmd.AddCommand(workflowRoot)
argoClient := argofake.NewSimpleClientset()
crdClient := apixv1beta1fake.NewSimpleClientset()
kubeClient := kubefake.NewSimpleClientset()
settings.PluginSettings[workflow.PluginSettingsID] = &wfenv.Settings{
ArgoClient: argoClient,
CRDClient: crdClient,
KubeClient: kubeClient,
cmdTests := []WorkflowCmdTest{
{
CmdTest: &test.CmdTest{
Name: "workflow-init",
CmdLine: "workflow init",
Objs: []runtime.Object{},
},
},
{
CmdTest: &test.CmdTest{
Name: "workflow-init-already-initialized",
CmdLine: "workflow init",
Objs: []runtime.Object{
&v1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: "argo"}},
&v1.ServiceAccount{ObjectMeta: metav1.ObjectMeta{Name: "argo", Namespace: "argo"}},
&rbacv1.ClusterRole{ObjectMeta: metav1.ObjectMeta{Name: "argo-aggregate-to-admin"}},
&rbacv1.ClusterRole{ObjectMeta: metav1.ObjectMeta{Name: "argo-aggregate-to-edit"}},
&rbacv1.ClusterRole{ObjectMeta: metav1.ObjectMeta{Name: "argo-aggregate-to-view"}},
&rbacv1.ClusterRole{ObjectMeta: metav1.ObjectMeta{Name: "argo-cluster-role"}},
&rbacv1.ClusterRoleBinding{ObjectMeta: metav1.ObjectMeta{Name: "argo-binding"}},
&v1.ConfigMap{ObjectMeta: metav1.ObjectMeta{Name: "workflow-controller-configmap", Namespace: "argo"}},
&v1beta2.Deployment{ObjectMeta: metav1.ObjectMeta{Name: "workflow-controller", Namespace: "argo"}},
},
},
ArgoObjs: []runtime.Object{},
CRDObjs: []runtime.Object{
&apixv1beta1.CustomResourceDefinition{ObjectMeta: metav1.ObjectMeta{Name: "workflows.argoproj.io"}},
},
},
}
var tt *test.CmdTest
tt = &test.CmdTest{
Name: "workflow-init",
CmdLine: "workflow init",
for _, tt := range cmdTests {
settings.PluginSettings[workflow.PluginSettingsID] = &wfenv.Settings{
KubeClient: kubefake.NewSimpleClientset(tt.CmdTest.Objs...),
ArgoClient: argofake.NewSimpleClientset(tt.ArgoObjs...),
CRDClient: apixv1beta1fake.NewSimpleClientset(tt.CRDObjs...),
}
test.RunTest(t, tt.CmdTest, rootCmd)
}
test.RunTest(t, tt, rootCmd)
tt = &test.CmdTest{
Name: "workflow-init-already-initialized",
CmdLine: "workflow init",
}
test.RunTest(t, tt, rootCmd)
}