9

Author: aledbf

components/ws-manager-mk2/controllers/maintenance_controller_test.go

@@ -0,0 +1,73 @@
+// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License.AGPL.txt in the project root for license information.
+
+package wsmanager
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	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"
+)
+
+func TestMaintenance(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 maintenance message", func(testCtx context.Context, t *testing.T, cfg *envconf.Config) context.Context {
+			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)
+			})
+
+			addInitTask := 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
+			}
+
+			nfo, stopWs, err := integration.LaunchWorkspaceDirectly(t, ctx, api, integration.WithRequestModifier(addInitTask))
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			t.Logf("status: %v", nfo.LastStatus)
+
+			t.Cleanup(func() {
+				sctx, scancel := context.WithTimeout(context.Background(), 5*time.Minute)
+				defer scancel()
+
+				sapi := integration.NewComponentAPI(sctx, cfg.Namespace(), kubeconfig, cfg.Client())
+				defer sapi.Done(t)
+
+				if _, err = stopWs(true, sapi); err != nil {
+					t.Errorf("cannot stop workspace: %q", err)
+				}
+			})
+
+			return testCtx
+		}).
+		Feature()
+
+	testEnv.Test(t, f)
+}