10

Author: aledbf

test/tests/components/ws-manager/maintenence_test.go

@@ -6,14 +6,19 @@ package wsmanager
 
 import (
 	"context"
+	"encoding/json"
 	"testing"
 	"time"
 
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"sigs.k8s.io/e2e-framework/pkg/envconf"
+	"sigs.k8s.io/e2e-framework/pkg/features"
+
 	csapi "github.com/gitpod-io/gitpod/content-service/api"
 	"github.com/gitpod-io/gitpod/test/pkg/integration"
 	wsmanapi "github.com/gitpod-io/gitpod/ws-manager/api"
-	"sigs.k8s.io/e2e-framework/pkg/envconf"
-	"sigs.k8s.io/e2e-framework/pkg/features"
+	"github.com/gitpod-io/gitpod/ws-manager/api/config"
 )
 
 func TestMaintenance(t *testing.T) {
@@ -24,6 +29,43 @@ func TestMaintenance(t *testing.T) {
 		WithLabel("component", "ws-manager").
 		WithLabel("type", "maintenance").
 		Assess("should display maintenance message", func(testCtx context.Context, t *testing.T, cfg *envconf.Config) context.Context {
+			kubeClient, err := cfg.NewClient()
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			futureTime := time.Now().Add(1 * time.Hour)
+			mcfg := config.MaintenanceConfig{
+				EnabledUntil: &futureTime,
+			}
+
+			data, err := json.Marshal(mcfg)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			configMap := &corev1.ConfigMap{
+				ObjectMeta: metav1.ObjectMeta{
+					Name:      "ws-manager-mk2-maintenance-mode",
+					Namespace: "default",
+				},
+				Data: map[string]string{
+					"config.json": string(data),
+				},
+			}
+
+			err = kubeClient.Resources().Create(testCtx, configMap)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			t.Cleanup(func() {
+				err := kubeClient.Resources().Delete(testCtx, configMap)
+				if err != nil {
+					t.Error(err)
+				}
+			})
+
 			ctx, cancel := context.WithTimeout(testCtx, time.Duration(5*time.Minute))
 			defer cancel()