more tests

Emyrk · Emyrk · commit 25aa0464f08b · 2025-12-02T11:56:19.000-06:00
diff --git a/cli/create_test.go b/cli/create_test.go
@@ -1573,8 +1573,10 @@ func TestCreateValidateRichParameters(t *testing.T) {
 func TestCreateWithGitAuth(t *testing.T) {
 	t.Parallel()
 	echoResponses := &echo.Responses{
-		Parse: echo.ParseComplete,
-		ProvisionPlan: []*proto.Response{
+		Parse:         echo.ParseComplete,
+		ProvisionInit: echo.InitComplete,
+		ProvisionPlan: echo.PlanComplete,
+		ProvisionGraph: []*proto.Response{
 			{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
diff --git a/cli/restart_test.go b/cli/restart_test.go
@@ -306,7 +306,7 @@ func TestRestartWithParameters(t *testing.T) {
 	echoResponses := func() *echo.Responses {
 		return &echo.Responses{
 			Parse: echo.ParseComplete,
-			ProvisionPlan: []*proto.Response{
+			ProvisionGraph: []*proto.Response{
 				{
 					Type: &proto.Response_Graph{
 						Graph: &proto.GraphComplete{
diff --git a/cli/start_test.go b/cli/start_test.go
@@ -36,7 +36,7 @@ const (
 func mutableParamsResponse() *echo.Responses {
 	return &echo.Responses{
 		Parse: echo.ParseComplete,
-		ProvisionPlan: []*proto.Response{
+		ProvisionGraph: []*proto.Response{
 			{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
@@ -59,7 +59,7 @@ func mutableParamsResponse() *echo.Responses {
 func immutableParamsResponse() *echo.Responses {
 	return &echo.Responses{
 		Parse: echo.ParseComplete,
-		ProvisionPlan: []*proto.Response{
+		ProvisionGraph: []*proto.Response{
 			{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
diff --git a/cli/task_test.go b/cli/task_test.go
@@ -285,16 +285,7 @@ func createAITaskTemplate(t *testing.T, client *codersdk.Client, orgID uuid.UUID
 	taskAppID := uuid.New()
 	version := coderdtest.CreateTemplateVersion(t, client, orgID, &echo.Responses{
 		Parse: echo.ParseComplete,
-		ProvisionPlan: []*proto.Response{
-			{
-				Type: &proto.Response_Graph{
-					Graph: &proto.GraphComplete{
-						HasAiTasks: true,
-					},
-				},
-			},
-		},
-		ProvisionApply: []*proto.Response{
+		ProvisionGraph: []*proto.Response{
 			{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
@@ -321,6 +312,7 @@ func createAITaskTemplate(t *testing.T, client *codersdk.Client, orgID uuid.UUID
 								},
 							},
 						},
+						HasAiTasks: true,
 						AiTasks: []*proto.AITask{
 							{
 								AppId: taskAppID.String(),
diff --git a/provisioner/echo/serve_test.go b/provisioner/echo/serve_test.go
@@ -56,7 +56,8 @@ func TestEcho(t *testing.T) {
 			},
 		}
 		data, err := echo.Tar(&echo.Responses{
-			Parse: responses,
+			Parse:         responses,
+			ProvisionInit: echo.InitComplete,
 		})
 		require.NoError(t, err)
 		client, err := api.Session(ctx)
@@ -296,6 +297,7 @@ func TestEcho(t *testing.T) {
 			},
 		}}
 		data, err := echo.Tar(&echo.Responses{
+			ProvisionInit:  echo.InitComplete,
 			ProvisionPlan:  echo.PlanComplete,
 			ProvisionApply: echo.ApplyComplete,
 			ProvisionGraph: responses,

Original file line number	Diff line number	Diff line change
`@@ -306,7 +306,7 @@ func TestRestartWithParameters(t *testing.T) {`
`306`	`306`	`echoResponses := func() *echo.Responses {`
`307`	`307`	`return &echo.Responses{`
`308`	`308`	`Parse: echo.ParseComplete,`
`309`		`- ProvisionPlan: []*proto.Response{`
	`309`	`+ ProvisionGraph: []*proto.Response{`
`310`	`310`	`{`
`311`	`311`	`Type: &proto.Response_Graph{`
`312`	`312`	`Graph: &proto.GraphComplete{`