make gen impl

hugodutka · hugodutka · commit 7c5eba8ae299 · 2025-06-16T14:20:51.000Z
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
@@ -3452,7 +3452,8 @@ func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now ti
 }
 
 func (q *querier) HasTemplateVersionsWithAITask(ctx context.Context) (bool, error) {
-	panic("not implemented")
+	// Anyone can call HasTemplateVersionsWithAITask.
+	return q.db.HasTemplateVersionsWithAITask(ctx)
 }
 
 func (q *querier) InsertAPIKey(ctx context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -4566,6 +4566,9 @@ func (s *MethodTestSuite) TestSystemFunctions() {
 	s.Run("GetProvisionerJobByIDForUpdate", s.Subtest(func(db database.Store, check *expects) {
 		check.Args(uuid.New()).Asserts(rbac.ResourceProvisionerJobs, policy.ActionRead).Errors(sql.ErrNoRows)
 	}))
+	s.Run("HasTemplateVersionsWithAITask", s.Subtest(func(db database.Store, check *expects) {
+		check.Args().Asserts()
+	}))
 }
 
 func (s *MethodTestSuite) TestNotifications() {
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -8491,8 +8491,17 @@ func (q *FakeQuerier) GetWorkspacesEligibleForTransition(ctx context.Context, no
 	return workspaces, nil
 }
 
-func (q *FakeQuerier) HasTemplateVersionsWithAITask(ctx context.Context) (bool, error) {
-	panic("not implemented")
+func (q *FakeQuerier) HasTemplateVersionsWithAITask(_ context.Context) (bool, error) {
+	q.mutex.RLock()
+	defer q.mutex.RUnlock()
+
+	for _, templateVersion := range q.templateVersions {
+		if templateVersion.HasAITask {
+			return true, nil
+		}
+	}
+
+	return false, nil
 }
 
 func (q *FakeQuerier) InsertAPIKey(_ context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {

Original file line number	Diff line number	Diff line change
`@@ -3452,7 +3452,8 @@ func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now ti`
`3452`	`3452`	`}`
`3453`	`3453`
`3454`	`3454`	`func (q *querier) HasTemplateVersionsWithAITask(ctx context.Context) (bool, error) {`
`3455`		`- panic("not implemented")`
	`3455`	`+ // Anyone can call HasTemplateVersionsWithAITask.`
	`3456`	`+ return q.db.HasTemplateVersionsWithAITask(ctx)`
`3456`	`3457`	`}`
`3457`	`3458`
`3458`	`3459`	`func (q *querier) InsertAPIKey(ctx context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {`
Original file line number	Diff line number	Diff line change
`@@ -4566,6 +4566,9 @@ func (s *MethodTestSuite) TestSystemFunctions() {`
`4566`	`4566`	`s.Run("GetProvisionerJobByIDForUpdate", s.Subtest(func(db database.Store, check *expects) {`
`4567`	`4567`	`check.Args(uuid.New()).Asserts(rbac.ResourceProvisionerJobs, policy.ActionRead).Errors(sql.ErrNoRows)`
`4568`	`4568`	`}))`
	`4569`	`+ s.Run("HasTemplateVersionsWithAITask", s.Subtest(func(db database.Store, check *expects) {`
	`4570`	`+ check.Args().Asserts()`
	`4571`	`+ }))`
`4569`	`4572`	`}`
`4570`	`4573`
`4571`	`4574`	`func (s *MethodTestSuite) TestNotifications() {`