builder/pkg/gitreceive/k8s_util_test.go at e72b96ed8b7bc7b24b1684b1712cd4b0b1fdb592 · drycc/builder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package gitreceive

import (
	"reflect"
	"strings"
	"testing"

	"k8s.io/kubernetes/pkg/api"
)

func DockerBuilderPodName(t *testing.T) {
	name := dockerBuilderPodName("demo", "12345678")
	if !strings.HasPrefix(name, "dockerbuild-demo-12345678-") {
		t.Fatalf("expected pod name dockerbuild-demo-12345678-, got %s", name)
	}
}

func SlugBuilderPodName(t *testing.T) {
	name := dockerBuilderPodName("demo", "12345678")
	if !strings.HasPrefix(name, "slugbuild-demo-12345678-") {
		t.Fatalf("expected pod name slugbuild-demo-12345678-, got %s", name)
	}
}

type buildCase struct {
	buildType string
	withAuth  bool
	name      string
	namespace string
	env       map[string]interface{}
	templ     string
}

func TestBuildPod(t *testing.T) {
	emptyEnv := make(map[string]interface{})

	env := make(map[string]interface{})
	env["KEY"] = "VALUE"

	var pod api.Pod
	var expPod api.Pod

	yamlTempl = "../../rootfs/etc/deis-%v%v.yaml"

	builds := []buildCase{
		{"slugbuilder", true, "test", "default", emptyEnv, "../../rootfs/etc/deis-slugbuilder.yaml"},
		{"slugbuilder", false, "test", "default", emptyEnv, "../../rootfs/etc/deis-slugbuilder-no-creds.yaml"},
		{"slugbuilder", true, "test", "default", env, "../../rootfs/etc/deis-slugbuilder.yaml"},
		{"slugbuilder", false, "test", "default", env, "../../rootfs/etc/deis-slugbuilder-no-creds.yaml"},
		{"dockerbuilder", true, "test", "default", emptyEnv, "../../rootfs/etc/deis-dockerbuilder.yaml"},
		{"dockerbuilder", false, "test", "default", emptyEnv, "../../rootfs/etc/deis-dockerbuilder-no-creds.yaml"},
		{"dockerbuilder", true, "test", "default", env, "../../rootfs/etc/deis-dockerbuilder.yaml"},
		{"dockerbuilder", false, "test", "default", env, "../../rootfs/etc/deis-dockerbuilder-no-creds.yaml"},
	}

	for _, build := range builds {
		pod = buildPod(build.buildType, build.withAuth, build.name, build.namespace, build.env)
		expPod = buildTestPod(build.name, build.namespace, build.env, build.templ)
		if !reflect.DeepEqual(expPod, pod) {
			t.Errorf("%v and %v are not equal", expPod, pod)
		}
	}
}

func buildTestPod(name, namespace string, env map[string]interface{}, templ string) api.Pod {
	pod := podFromFile(templ)
	pod.ObjectMeta.Name = name
	pod.ObjectMeta.Namespace = namespace
	addEnvToPod(pod, env)
	return pod
}