Add test

Author: aledbf

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

@@ -70,7 +70,7 @@ func TestMaintenance(t *testing.T) {
 				api.Done(t)
 			})
 
-			addInitTask := func(swr *wsmanapi.StartWorkspaceRequest) error {
+			customizeWorkspace := func(swr *wsmanapi.StartWorkspaceRequest) error {
 				swr.Spec.Initializer = &csapi.WorkspaceInitializer{
 					Spec: &csapi.WorkspaceInitializer_Git{
 						Git: &csapi.GitInitializer{
@@ -84,7 +84,7 @@ func TestMaintenance(t *testing.T) {
 				return nil
 			}
 
-			_, _, err = integration.LaunchWorkspaceDirectly(t, ctx, api, integration.WithRequestModifier(addInitTask))
+			_, _, err = integration.LaunchWorkspaceDirectly(t, ctx, api, integration.WithRequestModifier(customizeWorkspace))
 			if err == nil {
 				t.Fatalf("expected under maintenance error")
 			} else {
@@ -100,6 +100,103 @@ func TestMaintenance(t *testing.T) {
 	testEnv.Test(t, f)
 }
 
+func TestMaintenanceUpdate(t *testing.T) {
+	testRepo := "https://github.com/gitpod-io/empty"
+	testRepoName := "empty"
+
+	f := features.New("ws-manager").
+		WithLabel("component", "ws-manager").
+		WithLabel("type", "maintenance").
+		Assess("should display a maintenance message when configured and not when disabled", func(testCtx context.Context, t *testing.T, cfg *envconf.Config) context.Context {
+			kubeClient, err := cfg.NewClient()
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			untilTime := time.Now().Add(1 * time.Hour)
+			cmap, err := maintenanceConfigmap(&untilTime)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			err = kubeClient.Resources().Create(testCtx, cmap)
+			if err != nil && !apierrors.IsAlreadyExists(err) {
+				t.Fatal(err)
+			}
+
+			disableMaintenanceMode := func() {
+				cmap, err := maintenanceConfigmap(nil)
+				if err != nil {
+					t.Error(err)
+					return
+				}
+
+				err = kubeClient.Resources().Update(testCtx, cmap)
+				if err != nil {
+					t.Error(err)
+				}
+			}
+
+			disableMaintenanceMode()
+
+			ctx, cancel := context.WithTimeout(testCtx, time.Duration(5*time.Minute))
+			defer cancel()
+
+			api := integration.NewComponentAPI(ctx, cfg.Namespace(), kubeconfig, cfg.Client())
+			t.Cleanup(func() {
+				api.Done(t)
+			})
+
+			_, _, err = integration.LaunchWorkspaceDirectly(t, ctx, api, integration.WithRequestModifier(func(swr *wsmanapi.StartWorkspaceRequest) error {
+				swr.Spec.Initializer = &csapi.WorkspaceInitializer{
+					Spec: &csapi.WorkspaceInitializer_Git{
+						Git: &csapi.GitInitializer{
+							RemoteUri:        testRepo,
+							CheckoutLocation: testRepoName,
+							Config:           &csapi.GitConfig{},
+						},
+					},
+				}
+				swr.Spec.WorkspaceLocation = testRepoName
+				return nil
+			}))
+			if err == nil {
+				t.Fatalf("expected under maintenance error")
+			} else {
+				if !errors.Is(err, status.Error(codes.FailedPrecondition, "under maintenance")) {
+					t.Fatal(err)
+				}
+			}
+
+			disableMaintenanceMode()
+
+			_, stopWs, err := integration.LaunchWorkspaceDirectly(t, ctx, api, integration.WithRequestModifier(func(swr *wsmanapi.StartWorkspaceRequest) error {
+				swr.Spec.Initializer = &csapi.WorkspaceInitializer{
+					Spec: &csapi.WorkspaceInitializer_Git{
+						Git: &csapi.GitInitializer{
+							RemoteUri:        testRepo,
+							CheckoutLocation: testRepoName,
+							Config:           &csapi.GitConfig{},
+						},
+					},
+				}
+				swr.Spec.WorkspaceLocation = testRepoName
+				return nil
+			}))
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			if err := stopWorkspace(t, cfg, stopWs); err != nil {
+				t.Errorf("cannot stop workspace: %q", err)
+			}
+
+			return testCtx
+		}).
+		Feature()
+
+	testEnv.Test(t, f)
+}
 func maintenanceConfigmap(untilTime *time.Time) (*corev1.ConfigMap, error) {
 	mcfg := config.MaintenanceConfig{}
 	if untilTime != nil {