Function TestWorkspaceBuildWithTask

coderd/wsbuilder/wsbuilder_test.go:1166–1218

(t *testing.T)

}

func TestWorkspaceBuildWithTask(t *testing.T) {
	t.Parallel()
	req := require.New(t)
	asrt := assert.New(t)

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	testTask := database.Task{
		ID:                taskID,
		OrganizationID:    orgID,
		OwnerID:           userID,
		Name:              "test-task",
		WorkspaceID:       uuid.NullUUID{UUID: workspaceID, Valid: true},
		TemplateVersionID: activeVersionID,
		CreatedAt:         dbtime.Now(),
	}

	mDB := expectDB(t,
		// Inputs
		withTemplate,
		withInactiveVersion(nil),
		withLastBuildFound,
		withLastBuildState,
		withTemplateVersionVariables(inactiveVersionID, nil),
		withRichParameters(nil),
		withParameterSchemas(inactiveJobID, nil),
		withWorkspaceTags(inactiveVersionID, nil),
		withProvisionerDaemons([]database.GetEligibleProvisionerDaemonsByProvisionerJobIDsRow{}),

		// Outputs
		expectProvisionerJob(func(job database.InsertProvisionerJobParams) {}),
		withInTx,
		expectFindMatchingPresetID(uuid.Nil, sql.ErrNoRows),
		expectBuild(func(bld database.InsertWorkspaceBuildParams) {}),
		withBuild,
		withTask(testTask),
		expectUpsertTaskWorkspaceApp(func(params database.UpsertTaskWorkspaceAppParams) {
			asrt.Equal(taskID, params.TaskID)
			asrt.Equal(int32(2), params.WorkspaceBuildNumber)
			asrt.False(params.WorkspaceAgentID.Valid, "workspace_agent_id should be NULL initially")
			asrt.False(params.WorkspaceAppID.Valid, "workspace_app_id should be NULL initially")
		}),
		expectBuildParameters(func(params database.InsertWorkspaceBuildParametersParams) {}),
	)
	fc := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})

	ws := database.Workspace{ID: workspaceID, TemplateID: templateID, OwnerID: userID}
	uut := wsbuilder.New(ws, database.WorkspaceTransitionStart, wsbuilder.NoopUsageChecker{})
	// nolint: dogsled
	_, _, _, err := uut.Build(ctx, mDB, fc, nil, audit.WorkspaceBuildBaggage{})
	req.NoError(err)
}

func TestWsbuildError(t *testing.T) {
	t.Parallel()