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

func TestNodePoolSpec_ApplyNode(t *testing.T) {
	type fields struct {
		Taints  []corev1.Taint
		Labels  map[string]string
		Handler string
	}
	type args struct {
		node v1.Node
	}
	tests := []struct {
		name   string
		fields fields
		args   args
		want   *corev1.Node
	}{
		{
			name: "label",
			fields: fields{
				Labels: map[string]string{
					"node-pool.lailin.xyz/test": "",
				},
			},
			args: args{
				node: v1.Node{
					ObjectMeta: metav1.ObjectMeta{
						Name: "worker",
						Labels: map[string]string{
							"kubernetes.io/arch": "amd64",
							"a":                  "b",
						},
					},
				},
			},
			want: &v1.Node{
				ObjectMeta: metav1.ObjectMeta{
					Name: "worker",
					Labels: map[string]string{
						"kubernetes.io/arch":        "amd64",
						"node-pool.lailin.xyz/test": "",
					},
				},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s := &NodePoolSpec{
				Taints:  tt.fields.Taints,
				Labels:  tt.fields.Labels,
				Handler: tt.fields.Handler,
			}
			assert.Equal(t, tt.want, s.ApplyNode(tt.args.node))
		})
	}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14

import sigs.k8s.io/controller-runtime/pkg/envtest
//指定 testEnv 配置
testEnv = &envtest.Environment{
    CRDDirectoryPaths: []string{filepath.Join("..", "config", "crd", "bases")},
}
//启动 testEnv
cfg, err = testEnv.Start()
//编写测试逻辑
//停止 testEnv
err = testEnv.Stop()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

var _ = Describe("node labels", func() {
	pool := &nodesv1.NodePool{
		ObjectMeta: metav1.ObjectMeta{
			Name: "test",
		},
		Spec: nodesv1.NodePoolSpec{
			Labels: map[string]string{
				"node-pool.lailin.xyz/xxx": "",
			},
			Handler: "",
		},
	}
	It("create pool", func() {
		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		err := k8sClient.Create(ctx, pool)
		Expect(err).NotTo(HaveOccurred())
	})
})

1
2
3
4
5

Using cached envtest tools from blog-code/k8s-operator/07-node-pool-operator/testbin
setting up env vars
?       github.com/mohuishou/blog-code/k8s-operator/node-pool-operator  [no test files]
ok      github.com/mohuishou/blog-code/k8s-operator/node-pool-operator/api/v1   9.403s  coverage: 24.5% of statements
ok      github.com/mohuishou/blog-code/k8s-operator/node-pool-operator/controllers      10.390s coverage: 0.0% of statements