controller: add more tests for flavor merge

tsuru · Jan 22, 2024 · 449357b · 449357b
1 parent 25278c0
commit 449357b
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/controllers/controller_test.go b/controllers/controller_test.go
@@ -498,11 +498,17 @@ func TestReconcileRpaasInstance_getRpaasInstance(t *testing.T) {
 						Namespace: "default",
 					},
 					Spec: v1alpha1.RpaasFlavorSpec{
+						Default: true,
 						InstanceTemplate: &v1alpha1.RpaasInstanceSpec{
 							DNS: &v1alpha1.DNSConfig{
 								Zone: "apps.example.com",
 								TTL:  func(n int32) *int32 { return &n }(300),
 							},
+							PodTemplate: nginxv1alpha1.NginxPodTemplateSpec{
+								Annotations: map[string]string{
+									"rpaas.extensions.tsuru.io/is-default-pod-annotation": "true",
+								},
+							},
 						},
 					},
 				},
@@ -530,6 +536,11 @@ func TestReconcileRpaasInstance_getRpaasInstance(t *testing.T) {
 					Zone: "apps.test",
 					TTL:  func(n int32) *int32 { return &n }(30),
 				}
+				i.Spec.PodTemplate = nginxv1alpha1.NginxPodTemplateSpec{
+					Annotations: map[string]string{
+						"rpaas.extensions.tsuru.io/is-default-pod-annotation": "true",
+					},
+				}
 				return i
 			},
 		},
@@ -721,6 +732,21 @@ func TestReconcileRpaasInstance_getRpaasInstance(t *testing.T) {
 						},
 					},
 				},
+				&v1alpha1.RpaasFlavor{
+					ObjectMeta: metav1.ObjectMeta{
+						Name:      "flavor-d",
+						Namespace: "default",
+					},
+					Spec: v1alpha1.RpaasFlavorSpec{
+						InstanceTemplate: &v1alpha1.RpaasInstanceSpec{
+							PodTemplate: nginxv1alpha1.NginxPodTemplateSpec{
+								Annotations: map[string]string{
+									"donotuse.tsuru.io": "flavor-d",
+								},
+							},
+						},
+					},
+				},
 			},
 			instance: func(i *v1alpha1.RpaasInstance) *v1alpha1.RpaasInstance {
 				i.Spec.Flavors = []string{"flavor-a", "flavor-b", "flavor-c"}