@@ -157,20 +157,21 @@ func TestTasks(t *testing.T) {
157157appURL string
158158authToken string
159159}
160+
160161type aiTemplateOpt func (* aiTemplateOpts )
162+
161163withSidebarURL := func (url string )aiTemplateOpt {return func (o * aiTemplateOpts ) {o .appURL = url } }
162164withAgentToken := func (token string )aiTemplateOpt {return func (o * aiTemplateOpts ) {o .authToken = token } }
165+
163166createAITemplate := func (t * testing.T ,client * codersdk.Client ,user codersdk.CreateFirstUserResponse ,opts ... aiTemplateOpt ) codersdk.Template {
164167t .Helper ()
165- var opt aiTemplateOpts
168+
169+ opt := aiTemplateOpts {
170+ authToken :uuid .New ().String (),
171+ }
166172for _ ,o := range opts {
167173o (& opt )
168174}
169- appURL := opt .appURL
170- var agentAuth * proto.Agent_Token
171- if opt .authToken != "" {
172- agentAuth = & proto.Agent_Token {Token :opt .authToken }
173- }
174175
175176// Create a template version that supports AI tasks with the AI Prompt parameter.
176177taskAppID := uuid .New ()
@@ -198,13 +199,15 @@ func TestTasks(t *testing.T) {
198199{
199200Id :uuid .NewString (),
200201Name :"example" ,
201- Auth :agentAuth ,
202+ Auth :& proto.Agent_Token {
203+ Token :opt .authToken ,
204+ },
202205Apps : []* proto.App {
203206{
204207Id :taskAppID .String (),
205208Slug :"task-sidebar" ,
206209DisplayName :"Task Sidebar" ,
207- Url :appURL ,
210+ Url :opt . appURL ,
208211},
209212},
210213},