mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
f19770b2c6
* chore: Rename ProjectHistory to ProjectVersion Version more accurately represents version storage. This forks from the WorkspaceHistory name, but I think it's easier to understand Workspace history. * Rename files
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package coderd_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"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"
|
|
"github.com/coder/coder/provisioner/echo"
|
|
"github.com/coder/coder/provisionersdk/proto"
|
|
)
|
|
|
|
func TestWorkspaceHistoryLogs(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.ProvisionerTypeEcho,
|
|
})
|
|
require.NoError(t, err)
|
|
workspace, err := client.CreateWorkspace(context.Background(), "", coderd.CreateWorkspaceRequest{
|
|
Name: "example",
|
|
ProjectID: project.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
return project, workspace
|
|
}
|
|
|
|
setupProjectVersion := func(t *testing.T, client *codersdk.Client, user coderd.CreateInitialUserRequest, project coderd.Project, data []byte) coderd.ProjectVersion {
|
|
projectVersion, err := client.CreateProjectVersion(context.Background(), user.Organization, project.Name, coderd.CreateProjectVersionRequest{
|
|
StorageMethod: database.ProjectStorageMethodInlineArchive,
|
|
StorageSource: data,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Eventually(t, func() bool {
|
|
hist, err := client.ProjectVersion(context.Background(), user.Organization, project.Name, projectVersion.Name)
|
|
require.NoError(t, err)
|
|
return hist.Import.Status.Completed()
|
|
}, 15*time.Second, 50*time.Millisecond)
|
|
return projectVersion
|
|
}
|
|
|
|
server := coderdtest.New(t)
|
|
user := server.RandomInitialUser(t)
|
|
_ = server.AddProvisionerd(t)
|
|
project, workspace := setupProjectAndWorkspace(t, server.Client, user)
|
|
data, err := echo.Tar(echo.ParseComplete, []*proto.Provision_Response{{
|
|
Type: &proto.Provision_Response_Log{
|
|
Log: &proto.Log{
|
|
Output: "test",
|
|
},
|
|
},
|
|
}, {
|
|
Type: &proto.Provision_Response_Complete{
|
|
Complete: &proto.Provision_Complete{},
|
|
},
|
|
}})
|
|
require.NoError(t, err)
|
|
projectVersion := setupProjectVersion(t, server.Client, user, project, data)
|
|
|
|
workspaceHistory, err := server.Client.CreateWorkspaceHistory(context.Background(), "", workspace.Name, coderd.CreateWorkspaceHistoryRequest{
|
|
ProjectVersionID: projectVersion.ID,
|
|
Transition: database.WorkspaceTransitionCreate,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
now := database.Now()
|
|
logChan, err := server.Client.FollowWorkspaceHistoryLogsAfter(context.Background(), "", workspace.Name, workspaceHistory.Name, now)
|
|
require.NoError(t, err)
|
|
|
|
for {
|
|
log, more := <-logChan
|
|
if !more {
|
|
break
|
|
}
|
|
t.Logf("Output: %s", log.Output)
|
|
}
|
|
|
|
t.Run("ReturnAll", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
_, err := server.Client.WorkspaceHistoryLogs(context.Background(), "", workspace.Name, workspaceHistory.Name)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Between", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
_, err := server.Client.WorkspaceHistoryLogsBetween(context.Background(), "", workspace.Name, workspaceHistory.Name, time.Time{}, database.Now())
|
|
require.NoError(t, err)
|
|
})
|
|
}
|