update tests

replicatedhq · Jul 14, 2023 · 5b6a629 · 5b6a629
1 parent 3e9b0a6
commit 5b6a629
Showing 1 changed file with 52 additions and 0 deletions.
diff --git a/pkg/integration/integration_test.go b/pkg/integration/integration_test.go
@@ -113,6 +113,58 @@ func TestIntegration_IsEnabled(t *testing.T) {
 			want:    false,
 			wantErr: false,
 		},
+		{
+			name: "enabled for a dev license because key doesn't exist",
+			args: args{
+				clientset: fake.NewSimpleClientset(&corev1.SecretList{
+					TypeMeta: metav1.TypeMeta{},
+					ListMeta: metav1.ListMeta{},
+					Items: []corev1.Secret{{
+						TypeMeta: metav1.TypeMeta{},
+						ObjectMeta: metav1.ObjectMeta{
+							Name:      replicatedSecretName,
+							Namespace: "default",
+						},
+						Data: map[string][]byte{},
+					}},
+				}),
+				namespace: "default",
+				license: &kotsv1beta1.License{
+					Spec: kotsv1beta1.LicenseSpec{
+						LicenseType: "dev",
+					},
+				},
+			},
+			want:    true,
+			wantErr: false,
+		},
+		{
+			name: "enabled for a dev license because value is empty",
+			args: args{
+				clientset: fake.NewSimpleClientset(&corev1.SecretList{
+					TypeMeta: metav1.TypeMeta{},
+					ListMeta: metav1.ListMeta{},
+					Items: []corev1.Secret{{
+						TypeMeta: metav1.TypeMeta{},
+						ObjectMeta: metav1.ObjectMeta{
+							Name:      replicatedSecretName,
+							Namespace: "default",
+						},
+						Data: map[string][]byte{
+							replicatedIntegrationEnabledKey: []byte(""),
+						},
+					}},
+				}),
+				namespace: "default",
+				license: &kotsv1beta1.License{
+					Spec: kotsv1beta1.LicenseSpec{
+						LicenseType: "dev",
+					},
+				},
+			},
+			want:    true,
+			wantErr: false,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {