mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
* feat: Add history middleware parameters These will be used for streaming logs, checking status, and other operations related to workspace and project history. * refactor: Move all HTTP routes to top-level struct Nesting all structs behind their respective structures is leaky, and promotes naming conflicts between handlers. Our HTTP routes cannot have conflicts, so neither should function naming.
136 lines
4.9 KiB
Go
136 lines
4.9 KiB
Go
package coderd_test
|
|
|
|
import (
|
|
"archive/tar"
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/coderd"
|
|
"github.com/coder/coder/coderd/coderdtest"
|
|
"github.com/coder/coder/codersdk"
|
|
"github.com/coder/coder/database"
|
|
)
|
|
|
|
func TestWorkspaceHistory(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
setupProjectAndWorkspace := func(t *testing.T, client *codersdk.Client, user coderd.CreateInitialUserRequest) (coderd.Project, coderd.Workspace) {
|
|
project, err := client.CreateProject(context.Background(), user.Organization, coderd.CreateProjectRequest{
|
|
Name: "banana",
|
|
Provisioner: database.ProvisionerTypeTerraform,
|
|
})
|
|
require.NoError(t, err)
|
|
workspace, err := client.CreateWorkspace(context.Background(), "", coderd.CreateWorkspaceRequest{
|
|
Name: "example",
|
|
ProjectID: project.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
return project, workspace
|
|
}
|
|
|
|
setupProjectHistory := func(t *testing.T, client *codersdk.Client, user coderd.CreateInitialUserRequest, project coderd.Project) coderd.ProjectHistory {
|
|
var buffer bytes.Buffer
|
|
writer := tar.NewWriter(&buffer)
|
|
err := writer.WriteHeader(&tar.Header{
|
|
Name: "file",
|
|
Size: 1 << 10,
|
|
})
|
|
require.NoError(t, err)
|
|
_, err = writer.Write(make([]byte, 1<<10))
|
|
require.NoError(t, err)
|
|
projectHistory, err := client.CreateProjectHistory(context.Background(), user.Organization, project.Name, coderd.CreateProjectHistoryRequest{
|
|
StorageMethod: database.ProjectStorageMethodInlineArchive,
|
|
StorageSource: buffer.Bytes(),
|
|
})
|
|
require.NoError(t, err)
|
|
return projectHistory
|
|
}
|
|
|
|
t.Run("AllHistory", func(t *testing.T) {
|
|
t.Parallel()
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
project, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
history, err := server.Client.WorkspaceHistory(context.Background(), "", workspace.Name)
|
|
require.NoError(t, err)
|
|
require.Len(t, history, 0)
|
|
projectVersion := setupProjectHistory(t, server.Client, user, project)
|
|
_, err = server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: projectVersion.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.NoError(t, err)
|
|
history, err = server.Client.WorkspaceHistory(context.Background(), "", workspace.Name)
|
|
require.NoError(t, err)
|
|
require.Len(t, history, 1)
|
|
})
|
|
|
|
t.Run("LatestHistory", func(t *testing.T) {
|
|
t.Parallel()
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
project, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
_, err := server.Client.LatestWorkspaceHistory(context.Background(), "", workspace.Name)
|
|
require.Error(t, err)
|
|
projectVersion := setupProjectHistory(t, server.Client, user, project)
|
|
_, err = server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: projectVersion.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.NoError(t, err)
|
|
_, err = server.Client.LatestWorkspaceHistory(context.Background(), "", workspace.Name)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("CreateHistory", func(t *testing.T) {
|
|
t.Parallel()
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
project, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
projectHistory := setupProjectHistory(t, server.Client, user, project)
|
|
|
|
_, err := server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: projectHistory.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("CreateHistoryAlreadyInProgress", func(t *testing.T) {
|
|
t.Parallel()
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
project, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
projectHistory := setupProjectHistory(t, server.Client, user, project)
|
|
|
|
_, err := server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: projectHistory.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: projectHistory.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("CreateHistoryInvalidProjectVersion", func(t *testing.T) {
|
|
t.Parallel()
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
_, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
|
|
_, err := server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectHistoryID: uuid.New(),
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.Error(t, err)
|
|
})
|
|
}
|