Method createWorkspace

scripts/workspace-runtime-audit/runtimeaudit_test.go:374–416

(t *testing.T, db database.Store, builds []workspaceBuildArgs)

}

func (s *setup) createWorkspace(t *testing.T, db database.Store, builds []workspaceBuildArgs) database.WorkspaceTable {
	// Create template version first
	tv := dbfake.TemplateVersion(t, db).
		Seed(database.TemplateVersion{
			OrganizationID: s.org.ID,
			CreatedBy:      s.usr.ID,
		}).
		Do()

	wrk := dbgen.Workspace(t, db, database.WorkspaceTable{
		OwnerID:        s.usr.ID,
		OrganizationID: s.org.ID,
		TemplateID:     tv.Template.ID,
		CreatedAt:      builds[0].at,
	})

	for i, b := range builds {
		builder := dbfake.WorkspaceBuild(t, db, wrk).
			Seed(database.WorkspaceBuild{
				CreatedAt:         b.at,
				UpdatedAt:         b.at,
				TemplateVersionID: tv.TemplateVersion.ID,
				//nolint:gosec // this will not overflow
				BuildNumber: int32(i) + 1,
				Transition:  b.transition,
				InitiatorID: s.usr.ID,
			}).
			Succeeded(dbfake.WithJobCompletedAt(b.at))

		// Set job status based on the build args
		switch b.jobStatus {
		case database.ProvisionerJobStatusCanceled:
			builder = builder.Canceled(dbfake.WithJobCompletedAt(b.at))
		case database.ProvisionerJobStatusFailed:
			builder = builder.Failed(dbfake.WithJobError("fake error"), dbfake.WithJobCompletedAt(b.at))
			// default: Succeeded (the builder's default)
		}

		builder.Do()
	}

	return wrk
}