Commit 2a610086 authored by Hordur Freyr Yngvason's avatar Hordur Freyr Yngvason
Browse files

Merge branch 'dcroft-service_template_test' into 'master' 

Moves service template test cases to separate file

See merge request gitlab-org/cluster-integration/auto-deploy-image!180
parents 72ca54e6 a952ee80

assets/auto-deploy-app/test/template_test.go

+0 −91
Original line number Diff line number Diff line
@@ -17,97 +17,6 @@ const ( 
	helmChartPath = ".."

)



func TestServiceTemplate_DifferentTracks(t *testing.T) {

	templates := []string{"templates/service.yaml"}

	tcs := []struct {

		name        string

		releaseName string

		values      map[string]string



		expectedName        string

		expectedLabels      map[string]string

		expectedSelector    map[string]string

		expectedErrorRegexp *regexp.Regexp

	}{

		{

			name:             "defaults",

			releaseName:      "production",

			expectedName:     "production-auto-deploy",

			expectedLabels:   map[string]string{"app": "production", "release": "production", "track": "stable"},

			expectedSelector: map[string]string{"app": "production", "tier": "web", "track": "stable"},

		},

		{

			name:             "with canary track",

			releaseName:      "production-canary",

			values:           map[string]string{"application.track": "canary"},

			expectedName:     "production-canary-auto-deploy",

			expectedLabels:   map[string]string{"app": "production-canary", "release": "production-canary", "track": "canary"},

			expectedSelector: map[string]string{"app": "production-canary", "tier": "web", "track": "canary"},

		},

	}



	for _, tc := range tcs {

		t.Run(tc.name, func(t *testing.T) {

			output, ret := renderTemplate(t, tc.values, tc.releaseName, templates, tc.expectedErrorRegexp)



			if ret == false {

				return

			}



			service := new(coreV1.Service)

			helm.UnmarshalK8SYaml(t, output, service)

			require.Equal(t, tc.expectedName, service.ObjectMeta.Name)

			for key, value := range tc.expectedLabels {

				require.Equal(t, service.ObjectMeta.Labels[key], value)

			}

			for key, value := range tc.expectedSelector {

				require.Equal(t, service.Spec.Selector[key], value)

			}

		})

	}

}



func TestServiceTemplate_Disable(t *testing.T) {

	templates := []string{"templates/service.yaml"}

	releaseName := "service-disable-test"

	tcs := []struct {

		name   string

		values map[string]string



		expectedName        string

		expectedErrorRegexp *regexp.Regexp

	}{

		{

			name:         "defaults",

			expectedName: releaseName + "-auto-deploy",

		},

		{

			name:                "with service disabled and track stable",

			values:              map[string]string{"service.enabled": "false", "application.track": "stable"},

			expectedErrorRegexp: regexp.MustCompile("Error: could not find template templates/service.yaml in chart"),

		},

		{

			name:                "with service disabled and track non-stable",

			values:              map[string]string{"service.enabled": "false", "application.track": "non-stable"},

			expectedErrorRegexp: regexp.MustCompile("Error: could not find template templates/service.yaml in chart"),

		},

	}



	for _, tc := range tcs {

		t.Run(tc.name, func(t *testing.T) {

			output, ret := renderTemplate(t, tc.values, releaseName, templates, tc.expectedErrorRegexp)



			if ret == false {

				return

			}



			service := new(coreV1.Service)

			helm.UnmarshalK8SYaml(t, output, service)

			require.Equal(t, tc.expectedName, service.ObjectMeta.Name)

		})

	}

}



func renderTemplate(t *testing.T, values map[string]string, releaseName string, templates []string, expectedErrorRegexp *regexp.Regexp) (string, bool) {

	opts := &helm.Options{

		SetValues: values,

assets/auto-deploy-app/test/templates/service_test.go

0 → 100644
+101 −0
Original line number Diff line number Diff line 
package main



import (

	"regexp"

	"testing"



	"github.com/gruntwork-io/terratest/modules/helm"

	"github.com/stretchr/testify/require"

	coreV1 "k8s.io/api/core/v1"

)



func TestServiceTemplate_DifferentTracks(t *testing.T) {

	templates := []string{"templates/service.yaml"}

	tcs := []struct {

		name        string

		releaseName string

		values      map[string]string



		expectedName        string

		expectedLabels      map[string]string

		expectedSelector    map[string]string

		expectedErrorRegexp *regexp.Regexp

	}{

		{

			name:             "defaults",

			releaseName:      "production",

			expectedName:     "production-auto-deploy",

			expectedLabels:   map[string]string{"app": "production", "release": "production", "track": "stable"},

			expectedSelector: map[string]string{"app": "production", "tier": "web", "track": "stable"},

		},

		{

			name:             "with canary track",

			releaseName:      "production-canary",

			values:           map[string]string{"application.track": "canary"},

			expectedName:     "production-canary-auto-deploy",

			expectedLabels:   map[string]string{"app": "production-canary", "release": "production-canary", "track": "canary"},

			expectedSelector: map[string]string{"app": "production-canary", "tier": "web", "track": "canary"},

		},

	}



	for _, tc := range tcs {

		t.Run(tc.name, func(t *testing.T) {

			output, ret := renderTemplate(t, tc.values, tc.releaseName, templates, tc.expectedErrorRegexp)



			if ret == false {

				return

			}



			service := new(coreV1.Service)

			helm.UnmarshalK8SYaml(t, output, service)

			require.Equal(t, tc.expectedName, service.ObjectMeta.Name)

			for key, value := range tc.expectedLabels {

				require.Equal(t, service.ObjectMeta.Labels[key], value)

			}

			for key, value := range tc.expectedSelector {

				require.Equal(t, service.Spec.Selector[key], value)

			}

		})

	}

}



func TestServiceTemplate_Disable(t *testing.T) {

	templates := []string{"templates/service.yaml"}

	releaseName := "service-disable-test"

	tcs := []struct {

		name   string

		values map[string]string



		expectedName        string

		expectedErrorRegexp *regexp.Regexp

	}{

		{

			name:         "defaults",

			expectedName: releaseName + "-auto-deploy",

		},

		{

			name:                "with service disabled and track stable",

			values:              map[string]string{"service.enabled": "false", "application.track": "stable"},

			expectedErrorRegexp: regexp.MustCompile("Error: could not find template templates/service.yaml in chart"),

		},

		{

			name:                "with service disabled and track non-stable",

			values:              map[string]string{"service.enabled": "false", "application.track": "non-stable"},

			expectedErrorRegexp: regexp.MustCompile("Error: could not find template templates/service.yaml in chart"),

		},

	}



	for _, tc := range tcs {

		t.Run(tc.name, func(t *testing.T) {

			output, ret := renderTemplate(t, tc.values, releaseName, templates, tc.expectedErrorRegexp)



			if ret == false {

				return

			}



			service := new(coreV1.Service)

			helm.UnmarshalK8SYaml(t, output, service)

			require.Equal(t, tc.expectedName, service.ObjectMeta.Name)

		})

	}

}