Function TestExecutorTaskWorkspace

coderd/autobuild/lifecycle_executor_test.go:1860–2087

(t *testing.T)

}

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

	createTaskTemplate := func(t *testing.T, client *codersdk.Client, orgID uuid.UUID, ctx context.Context, defaultTTL time.Duration) codersdk.Template {
		t.Helper()

		taskAppID := uuid.New()
		version := coderdtest.CreateTemplateVersion(t, client, orgID, &echo.Responses{
			Parse: echo.ParseComplete,
			ProvisionGraph: []*proto.Response{
				{
					Type: &proto.Response_Graph{
						Graph: &proto.GraphComplete{
							Resources: []*proto.Resource{
								{
									Agents: []*proto.Agent{
										{
											Id:   uuid.NewString(),
											Name: "dev",
											Auth: &proto.Agent_Token{
												Token: uuid.NewString(),
											},
											Apps: []*proto.App{
												{
													Id:   taskAppID.String(),
													Slug: "task-app",
												},
											},
										},
									},
								},
							},
							HasAiTasks: true,
							AiTasks: []*proto.AITask{
								{
									AppId: taskAppID.String(),
								},
							},
						},
					},
				},
			},
		})
		coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)
		template := coderdtest.CreateTemplate(t, client, orgID, version.ID)

		if defaultTTL > 0 {
			_, err := client.UpdateTemplateMeta(ctx, template.ID, codersdk.UpdateTemplateMeta{
				DefaultTTLMillis: ptr.Ref(defaultTTL.Milliseconds()),
			})
			require.NoError(t, err)
		}

		return template
	}

	createTaskWorkspace := func(t *testing.T, client *codersdk.Client, template codersdk.Template, ctx context.Context, input string) codersdk.Workspace {
		t.Helper()