Function TestWorkspaceCreateWithImplicitPreset

coderd/workspaces_test.go:6131–6411

(t *testing.T)

}

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

	// Helper function to create template with presets
	createTemplateWithPresets := func(t *testing.T, client *codersdk.Client, user codersdk.CreateFirstUserResponse, presets []*proto.Preset) (codersdk.Template, codersdk.TemplateVersion) {
		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
			Parse: echo.ParseComplete,
			ProvisionGraph: []*proto.Response{
				{
					Type: &proto.Response_Graph{
						Graph: &proto.GraphComplete{
							Presets: presets,
						},
					},
				},
			},
		})
		coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)
		template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
		return template, version
	}

	// Helper function to create workspace and verify preset usage
	createWorkspaceAndVerifyPreset := func(t *testing.T, client *codersdk.Client, template codersdk.Template, expectedPresetID *uuid.UUID, params []codersdk.WorkspaceBuildParameter) codersdk.Workspace {
		wsName := testutil.GetRandomNameHyphenated(t)
		var ws codersdk.Workspace
		if len(params) > 0 {
			ws = coderdtest.CreateWorkspace(t, client, template.ID, func(cwr *codersdk.CreateWorkspaceRequest) {
				cwr.Name = wsName
				cwr.RichParameterValues = params
			})
		} else {
			ws = coderdtest.CreateWorkspace(t, client, template.ID, func(cwr *codersdk.CreateWorkspaceRequest) {
				cwr.Name = wsName
			})
		}
		require.Equal(t, wsName, ws.Name)

		coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, ws.LatestBuild.ID)

		// Verify the preset was used if expected
		if expectedPresetID != nil {
			require.NotNil(t, ws.LatestBuild.TemplateVersionPresetID)
			require.Equal(t, *expectedPresetID, *ws.LatestBuild.TemplateVersionPresetID)
		} else {
			require.Nil(t, ws.LatestBuild.TemplateVersionPresetID)
		}

		return ws
	}

	t.Run("NoPresets", func(t *testing.T) {
		t.Parallel()

		client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})
		user := coderdtest.CreateFirstUser(t, client)

		// Create template with no presets