mirror of https://github.com/coder/coder.git
233 lines
9.0 KiB
Go
233 lines
9.0 KiB
Go
package coderd_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/coderd/coderdtest"
|
|
"github.com/coder/coder/v2/coderd/database"
|
|
"github.com/coder/coder/v2/coderd/database/dbauthz"
|
|
"github.com/coder/coder/v2/coderd/database/dbfake"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/provisionersdk/proto"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
func TestPostWorkspaceAgentPortShare(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
|
|
defer cancel()
|
|
dep := coderdtest.DeploymentValues(t)
|
|
dep.Experiments = append(dep.Experiments, string(codersdk.ExperimentSharedPorts))
|
|
ownerClient, db := coderdtest.NewWithDatabase(t, &coderdtest.Options{
|
|
DeploymentValues: dep,
|
|
})
|
|
owner := coderdtest.CreateFirstUser(t, ownerClient)
|
|
client, user := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)
|
|
|
|
tmpDir := t.TempDir()
|
|
r := dbfake.WorkspaceBuild(t, db, database.Workspace{
|
|
OrganizationID: owner.OrganizationID,
|
|
OwnerID: user.ID,
|
|
}).WithAgent(func(agents []*proto.Agent) []*proto.Agent {
|
|
agents[0].Directory = tmpDir
|
|
return agents
|
|
}).Do()
|
|
agents, err := db.GetWorkspaceAgentsInLatestBuildByWorkspaceID(dbauthz.As(ctx, coderdtest.AuthzUserSubject(user, owner.OrganizationID)), r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
|
|
// owner level should fail
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevel("owner"),
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.Error(t, err)
|
|
|
|
// invalid level should fail
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevel("invalid"),
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.Error(t, err)
|
|
|
|
// invalid protocol should fail
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocol("invalid"),
|
|
})
|
|
require.Error(t, err)
|
|
|
|
// invalid port should fail
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 0,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.Error(t, err)
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 90000000,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
})
|
|
require.Error(t, err)
|
|
|
|
// OK, ignoring template max port share level because we are AGPL
|
|
ps, err := client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTPS,
|
|
})
|
|
require.NoError(t, err)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelPublic, ps.ShareLevel)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareProtocolHTTPS, ps.Protocol)
|
|
|
|
// list
|
|
list, err := client.GetWorkspaceAgentPortShares(ctx, r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
require.Len(t, list.Shares, 1)
|
|
require.EqualValues(t, agents[0].Name, list.Shares[0].AgentName)
|
|
require.EqualValues(t, 8080, list.Shares[0].Port)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelPublic, list.Shares[0].ShareLevel)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareProtocolHTTPS, list.Shares[0].Protocol)
|
|
|
|
// update share level and protocol
|
|
ps, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelAuthenticated,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.NoError(t, err)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelAuthenticated, ps.ShareLevel)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareProtocolHTTP, ps.Protocol)
|
|
|
|
// list
|
|
list, err = client.GetWorkspaceAgentPortShares(ctx, r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
require.Len(t, list.Shares, 1)
|
|
require.EqualValues(t, agents[0].Name, list.Shares[0].AgentName)
|
|
require.EqualValues(t, 8080, list.Shares[0].Port)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelAuthenticated, list.Shares[0].ShareLevel)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareProtocolHTTP, list.Shares[0].Protocol)
|
|
|
|
// list 2 ordered by port
|
|
ps, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8081,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTPS,
|
|
})
|
|
require.NoError(t, err)
|
|
list, err = client.GetWorkspaceAgentPortShares(ctx, r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
require.Len(t, list.Shares, 2)
|
|
require.EqualValues(t, 8080, list.Shares[0].Port)
|
|
require.EqualValues(t, 8081, list.Shares[1].Port)
|
|
}
|
|
|
|
func TestGetWorkspaceAgentPortShares(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
|
|
defer cancel()
|
|
|
|
dep := coderdtest.DeploymentValues(t)
|
|
dep.Experiments = append(dep.Experiments, string(codersdk.ExperimentSharedPorts))
|
|
ownerClient, db := coderdtest.NewWithDatabase(t, &coderdtest.Options{
|
|
DeploymentValues: dep,
|
|
})
|
|
owner := coderdtest.CreateFirstUser(t, ownerClient)
|
|
client, user := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)
|
|
|
|
tmpDir := t.TempDir()
|
|
r := dbfake.WorkspaceBuild(t, db, database.Workspace{
|
|
OrganizationID: owner.OrganizationID,
|
|
OwnerID: user.ID,
|
|
}).WithAgent(func(agents []*proto.Agent) []*proto.Agent {
|
|
agents[0].Directory = tmpDir
|
|
return agents
|
|
}).Do()
|
|
agents, err := db.GetWorkspaceAgentsInLatestBuildByWorkspaceID(dbauthz.As(ctx, coderdtest.AuthzUserSubject(user, owner.OrganizationID)), r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
|
|
_, err = client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
ps, err := client.GetWorkspaceAgentPortShares(ctx, r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
require.Len(t, ps.Shares, 1)
|
|
require.EqualValues(t, agents[0].Name, ps.Shares[0].AgentName)
|
|
require.EqualValues(t, 8080, ps.Shares[0].Port)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelPublic, ps.Shares[0].ShareLevel)
|
|
}
|
|
|
|
func TestDeleteWorkspaceAgentPortShare(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
|
|
defer cancel()
|
|
|
|
dep := coderdtest.DeploymentValues(t)
|
|
dep.Experiments = append(dep.Experiments, string(codersdk.ExperimentSharedPorts))
|
|
ownerClient, db := coderdtest.NewWithDatabase(t, &coderdtest.Options{
|
|
DeploymentValues: dep,
|
|
})
|
|
owner := coderdtest.CreateFirstUser(t, ownerClient)
|
|
client, user := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)
|
|
|
|
tmpDir := t.TempDir()
|
|
r := dbfake.WorkspaceBuild(t, db, database.Workspace{
|
|
OrganizationID: owner.OrganizationID,
|
|
OwnerID: user.ID,
|
|
}).WithAgent(func(agents []*proto.Agent) []*proto.Agent {
|
|
agents[0].Directory = tmpDir
|
|
return agents
|
|
}).Do()
|
|
agents, err := db.GetWorkspaceAgentsInLatestBuildByWorkspaceID(dbauthz.As(ctx, coderdtest.AuthzUserSubject(user, owner.OrganizationID)), r.Workspace.ID)
|
|
require.NoError(t, err)
|
|
|
|
// create
|
|
ps, err := client.UpsertWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.UpsertWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
ShareLevel: codersdk.WorkspaceAgentPortShareLevelPublic,
|
|
Protocol: codersdk.WorkspaceAgentPortShareProtocolHTTP,
|
|
})
|
|
require.NoError(t, err)
|
|
require.EqualValues(t, codersdk.WorkspaceAgentPortShareLevelPublic, ps.ShareLevel)
|
|
|
|
// delete
|
|
err = client.DeleteWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.DeleteWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// delete missing
|
|
err = client.DeleteWorkspaceAgentPortShare(ctx, r.Workspace.ID, codersdk.DeleteWorkspaceAgentPortShareRequest{
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
})
|
|
require.Error(t, err)
|
|
|
|
_, err = db.GetWorkspaceAgentPortShare(dbauthz.As(ctx, coderdtest.AuthzUserSubject(user, owner.OrganizationID)), database.GetWorkspaceAgentPortShareParams{
|
|
WorkspaceID: r.Workspace.ID,
|
|
AgentName: agents[0].Name,
|
|
Port: 8080,
|
|
})
|
|
require.Error(t, err)
|
|
}
|