mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
201 lines
8.0 KiB
Go
201 lines
8.0 KiB
Go
// Copyright (c) 2022 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 server
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/bufbuild/connect-go"
|
|
"github.com/gitpod-io/gitpod/common-go/baseserver"
|
|
"github.com/gitpod-io/gitpod/public-api-server/pkg/auth"
|
|
v1 "github.com/gitpod-io/gitpod/public-api/v1"
|
|
"github.com/gitpod-io/gitpod/public-api/v1/v1connect"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/metadata"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
func TestPublicAPIServer_v1_WorkspaceService(t *testing.T) {
|
|
ctx := metadata.AppendToOutgoingContext(context.Background(), "authorization", "some-token")
|
|
srv := baseserver.NewForTests(t,
|
|
baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)),
|
|
)
|
|
|
|
gitpodAPI, err := url.Parse("wss://main.preview.gitpod-dev.com/api/v1")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, register(srv, gitpodAPI))
|
|
baseserver.StartServerForTests(t, srv)
|
|
|
|
conn, err := grpc.Dial(srv.GRPCAddress(), grpc.WithTransportCredentials(insecure.NewCredentials()))
|
|
require.NoError(t, err)
|
|
|
|
workspaceClient := v1.NewWorkspacesServiceClient(conn)
|
|
|
|
_, err = workspaceClient.CreateAndStartWorkspace(ctx, &v1.CreateAndStartWorkspaceRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
_, err = workspaceClient.StartWorkspace(ctx, &v1.StartWorkspaceRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
_, err = workspaceClient.GetActiveWorkspaceInstance(ctx, &v1.GetActiveWorkspaceInstanceRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
_, err = workspaceClient.GetWorkspaceInstanceOwnerToken(ctx, &v1.GetWorkspaceInstanceOwnerTokenRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
stopWorkspaceStream, err := workspaceClient.StopWorkspace(ctx, &v1.StopWorkspaceRequest{})
|
|
require.NoError(t, err)
|
|
_, err = stopWorkspaceStream.Recv()
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
listenWorkspaceStream, err := workspaceClient.ListenToWorkspaceInstance(ctx, &v1.ListenToWorkspaceInstanceRequest{})
|
|
require.NoError(t, err)
|
|
_, err = listenWorkspaceStream.Recv()
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
listenImageBuildStream, err := workspaceClient.ListenToImageBuildLogs(ctx, &v1.ListenToImageBuildLogsRequest{})
|
|
require.NoError(t, err)
|
|
_, err = listenImageBuildStream.Recv()
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
}
|
|
|
|
func TestPublicAPIServer_v1_PrebuildService(t *testing.T) {
|
|
ctx := context.Background()
|
|
srv := baseserver.NewForTests(t, baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)))
|
|
|
|
gitpodAPI, err := url.Parse("wss://main.preview.gitpod-dev.com/api/v1")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, register(srv, gitpodAPI))
|
|
|
|
baseserver.StartServerForTests(t, srv)
|
|
|
|
conn, err := grpc.Dial(srv.GRPCAddress(), grpc.WithTransportCredentials(insecure.NewCredentials()))
|
|
require.NoError(t, err)
|
|
|
|
prebuildClient := v1.NewPrebuildsServiceClient(conn)
|
|
|
|
_, err = prebuildClient.GetPrebuild(ctx, &v1.GetPrebuildRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
_, err = prebuildClient.GetRunningPrebuild(ctx, &v1.GetRunningPrebuildRequest{})
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
listenToStatusStream, err := prebuildClient.ListenToPrebuildStatus(ctx, &v1.ListenToPrebuildStatusRequest{})
|
|
require.NoError(t, err)
|
|
_, err = listenToStatusStream.Recv()
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
|
|
listenToLogsStream, err := prebuildClient.ListenToPrebuildLogs(ctx, &v1.ListenToPrebuildLogsRequest{})
|
|
require.NoError(t, err)
|
|
_, err = listenToLogsStream.Recv()
|
|
requireErrorStatusCode(t, codes.Unimplemented, err)
|
|
}
|
|
|
|
func TestPublicAPIServer_WorkspaceServiceHandler(t *testing.T) {
|
|
ctx := context.Background()
|
|
srv := baseserver.NewForTests(t,
|
|
baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)),
|
|
baseserver.WithHTTP(baseserver.MustUseRandomLocalAddress(t)),
|
|
)
|
|
|
|
gitpodAPI, err := url.Parse("wss://main.preview.gitpod-dev.com/api/v1")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, register(srv, gitpodAPI))
|
|
baseserver.StartServerForTests(t, srv)
|
|
|
|
client := v1connect.NewWorkspacesServiceClient(http.DefaultClient, srv.HTTPAddress(), connect.WithInterceptors(auth.NewClientInterceptor("token")))
|
|
|
|
_, err = client.ListWorkspaces(ctx, connect.NewRequest(&v1.ListWorkspacesRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.GetWorkspace(ctx, connect.NewRequest(&v1.GetWorkspaceRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.GetOwnerToken(ctx, connect.NewRequest(&v1.GetOwnerTokenRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.CreateAndStartWorkspace(ctx, connect.NewRequest(&v1.CreateAndStartWorkspaceRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.StartWorkspace(ctx, connect.NewRequest(&v1.StartWorkspaceRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.GetActiveWorkspaceInstance(ctx, connect.NewRequest(&v1.GetActiveWorkspaceInstanceRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
_, err = client.GetWorkspaceInstanceOwnerToken(ctx, connect.NewRequest(&v1.GetWorkspaceInstanceOwnerTokenRequest{}))
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(err).String())
|
|
|
|
stream, err := client.ListenToWorkspaceInstance(ctx, connect.NewRequest(&v1.ListenToWorkspaceInstanceRequest{}))
|
|
require.NoError(t, err)
|
|
stream.Receive()
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(stream.Err()).String())
|
|
|
|
logsStream, err := client.ListenToImageBuildLogs(ctx, connect.NewRequest(&v1.ListenToImageBuildLogsRequest{}))
|
|
require.NoError(t, err)
|
|
logsStream.Receive()
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(logsStream.Err()).String())
|
|
|
|
stopStream, err := client.StopWorkspace(ctx, connect.NewRequest(&v1.StopWorkspaceRequest{}))
|
|
require.NoError(t, err)
|
|
stopStream.Receive()
|
|
require.Equal(t, connect.CodeUnimplemented.String(), connect.CodeOf(stopStream.Err()).String())
|
|
}
|
|
|
|
func requireErrorStatusCode(t *testing.T, expected codes.Code, err error) {
|
|
require.Error(t, err)
|
|
st, ok := status.FromError(err)
|
|
require.True(t, ok)
|
|
require.Equalf(t, expected, st.Code(), "expected: %s but got: %s", expected.String(), st.String())
|
|
}
|
|
|
|
func TestConnectWorkspaceService_RequiresAuth(t *testing.T) {
|
|
srv := baseserver.NewForTests(t,
|
|
baseserver.WithHTTP(baseserver.MustUseRandomLocalAddress(t)),
|
|
baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)),
|
|
)
|
|
|
|
gitpodAPI, err := url.Parse("wss://main.preview.gitpod-dev.com/api/v1")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, register(srv, gitpodAPI))
|
|
|
|
baseserver.StartServerForTests(t, srv)
|
|
|
|
clientWithoutAuth := v1connect.NewWorkspacesServiceClient(http.DefaultClient, srv.HTTPAddress())
|
|
_, err = clientWithoutAuth.GetWorkspace(context.Background(), connect.NewRequest(&v1.GetWorkspaceRequest{WorkspaceId: "123"}))
|
|
require.Error(t, err)
|
|
require.Equal(t, connect.CodeUnauthenticated, connect.CodeOf(err))
|
|
|
|
}
|
|
|
|
func TestConnectPrebuildsService_RequiresAuth(t *testing.T) {
|
|
srv := baseserver.NewForTests(t,
|
|
baseserver.WithHTTP(baseserver.MustUseRandomLocalAddress(t)),
|
|
baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)),
|
|
)
|
|
|
|
gitpodAPI, err := url.Parse("wss://main.preview.gitpod-dev.com/api/v1")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, register(srv, gitpodAPI))
|
|
|
|
baseserver.StartServerForTests(t, srv)
|
|
|
|
clientWithoutAuth := v1connect.NewPrebuildsServiceClient(http.DefaultClient, srv.HTTPAddress())
|
|
_, err = clientWithoutAuth.GetPrebuild(context.Background(), connect.NewRequest(&v1.GetPrebuildRequest{PrebuildId: "123"}))
|
|
require.Error(t, err)
|
|
require.Equal(t, connect.CodeUnauthenticated, connect.CodeOf(err))
|
|
}
|