coder
diff --git a/‎coderd/database/check_constraint.go
Lines changed: 2 additions & 2 deletions b/‎coderd/database/check_constraint.go
Lines changed: 2 additions & 2 deletions
diff --git a/‎coderd/database/dbauthz/dbauthz.go
Lines changed: 12 additions & 0 deletions b/‎coderd/database/dbauthz/dbauthz.go
Lines changed: 12 additions & 0 deletions
diff --git a/‎coderd/database/dbgen/dbgen.go
Lines changed: 38 additions & 0 deletions b/‎coderd/database/dbgen/dbgen.go
Lines changed: 38 additions & 0 deletions
diff --git a/‎coderd/database/dbmetrics/querymetrics.go
Lines changed: 21 additions & 0 deletions b/‎coderd/database/dbmetrics/querymetrics.go
Lines changed: 21 additions & 0 deletions
diff --git a/‎coderd/database/dbmock/dbmock.go
Lines changed: 45 additions & 0 deletions b/‎coderd/database/dbmock/dbmock.go
Lines changed: 45 additions & 0 deletions
@@ -2781,6 +2781,10 @@ func (q *querier) GetTailnetTunnelPeerIDs(ctx context.Context, srcID uuid.UUID)
 	return q.db.GetTailnetTunnelPeerIDs(ctx, srcID)
 }
 
+func (q *querier) GetTaskByID(ctx context.Context, id uuid.UUID) (database.Task, error) {
+	panic("not implemented")
+}
+
 func (q *querier) GetTelemetryItem(ctx context.Context, key string) (database.TelemetryItem, error) {
 	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil {
 		return database.TelemetryItem{}, err
@@ -3978,6 +3982,14 @@ func (q *querier) InsertReplica(ctx context.Context, arg database.InsertReplicaP
 	return q.db.InsertReplica(ctx, arg)
 }
 
+func (q *querier) InsertTask(ctx context.Context, arg database.InsertTaskParams) (database.TaskTable, error) {
+	panic("not implemented")
+}
+
+func (q *querier) InsertTaskWorkspaceApp(ctx context.Context, arg database.InsertTaskWorkspaceAppParams) (database.TaskWorkspaceApp, error) {
+	panic("not implemented")
+}
+
 func (q *querier) InsertTelemetryItemIfNotExists(ctx context.Context, arg database.InsertTelemetryItemIfNotExistsParams) error {
 	if err := q.authorizeContext(ctx, policy.ActionCreate, rbac.ResourceSystem); err != nil {
 		return err
 
@@ -27,6 +27,7 @@ import (
 	"github.com/coder/coder/v2/coderd/database/provisionerjobs"
 	"github.com/coder/coder/v2/coderd/database/pubsub"
 	"github.com/coder/coder/v2/coderd/rbac"
+	"github.com/coder/coder/v2/coderd/taskname"
 	"github.com/coder/coder/v2/codersdk"
 	"github.com/coder/coder/v2/cryptorand"
 	"github.com/coder/coder/v2/provisionerd/proto"
@@ -1470,6 +1471,43 @@ func ClaimPrebuild(
 	return claimedWorkspace
 }
 
+func Task(t testing.TB, db database.Store, orig database.TaskTable) database.TaskTable {
+	t.Helper()
+
+	parameters := orig.TemplateParameters
+	if parameters == nil {
+		parameters = json.RawMessage([]byte("{}"))
+	}
+
+	task, err := db.InsertTask(genCtx, database.InsertTaskParams{
+		OrganizationID:     orig.OrganizationID,
+		OwnerID:            orig.OwnerID,
+		Name:               takeFirst(orig.Name, taskname.GenerateFallback()),
+		WorkspaceID:        orig.WorkspaceID,
+		TemplateVersionID:  orig.TemplateVersionID,
+		TemplateParameters: parameters,
+		Prompt:             orig.Prompt,
+		CreatedAt:          takeFirst(orig.CreatedAt, dbtime.Now()),
+	})
+	require.NoError(t, err, "failed to insert task")
+
+	return task
+}
+
+func TaskWorkspaceApp(t testing.TB, db database.Store, orig database.TaskWorkspaceApp) database.TaskWorkspaceApp {
+	t.Helper()
+
+	app, err := db.InsertTaskWorkspaceApp(genCtx, database.InsertTaskWorkspaceAppParams{
+		TaskID:           orig.TaskID,
+		WorkspaceBuildID: orig.WorkspaceBuildID,
+		WorkspaceAgentID: orig.WorkspaceAgentID,
+		WorkspaceAppID:   orig.WorkspaceAppID,
+	})
+	require.NoError(t, err, "failed to insert task workspace app")
+
+	return app
+}
+
 func provisionerJobTiming(t testing.TB, db database.Store, seed database.ProvisionerJobTiming) database.ProvisionerJobTiming {
 	timing, err := db.InsertProvisionerJobTimings(genCtx, database.InsertProvisionerJobTimingsParams{
 		JobID:     takeFirst(seed.JobID, uuid.New()),