more tests

Emyrk · Emyrk · commit b93321e4c69f · 2025-12-02T11:56:19.000-06:00
diff --git a/enterprise/coderd/workspacequota_test.go b/enterprise/coderd/workspacequota_test.go
@@ -223,7 +223,10 @@ func TestWorkspaceQuota(t *testing.T) {
 		verifyQuota(ctx, t, client, user.OrganizationID.String(), 0, 4)
 
 		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
-			Parse: echo.ParseComplete,
+			Parse:          echo.ParseComplete,
+			ProvisionInit:  echo.InitComplete,
+			ProvisionPlan:  echo.PlanComplete,
+			ProvisionApply: echo.ApplyComplete,
 			ProvisionPlanMap: map[proto.WorkspaceTransition][]*proto.Response{
 				proto.WorkspaceTransition_START: planWithCost(2),
 				proto.WorkspaceTransition_STOP:  planWithCost(1),
@@ -429,7 +432,16 @@ func TestWorkspaceQuota(t *testing.T) {
 		// Create a template with a workspace that costs 1 credit
 		authToken := uuid.NewString()
 		version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
-			Parse: echo.ParseComplete,
+			Parse:         echo.ParseComplete,
+			ProvisionInit: echo.InitComplete,
+			ProvisionPlan: []*proto.Response{{
+				Type: &proto.Response_Plan{
+					Plan: &proto.PlanComplete{
+						DailyCost: 1,
+					},
+				},
+			}},
+			ProvisionApply: echo.ApplyComplete,
 			ProvisionGraph: []*proto.Response{{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
@@ -465,7 +477,16 @@ func TestWorkspaceQuota(t *testing.T) {
 
 		// Test with a template that has zero cost - should pass
 		versionZeroCost := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
-			Parse: echo.ParseComplete,
+			Parse:         echo.ParseComplete,
+			ProvisionInit: echo.InitComplete,
+			ProvisionPlan: []*proto.Response{{
+				Type: &proto.Response_Plan{
+					Plan: &proto.PlanComplete{
+						DailyCost: 0,
+					},
+				},
+			}},
+			ProvisionApply: echo.ApplyComplete,
 			ProvisionGraph: []*proto.Response{{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
@@ -549,7 +570,16 @@ func TestWorkspaceQuota(t *testing.T) {
 		// Create templates for both organizations
 		authToken := uuid.NewString()
 		version1 := coderdtest.CreateTemplateVersion(t, owner, first.OrganizationID, &echo.Responses{
-			Parse: echo.ParseComplete,
+			Parse:         echo.ParseComplete,
+			ProvisionInit: echo.InitComplete,
+			ProvisionPlan: []*proto.Response{{
+				Type: &proto.Response_Plan{
+					Plan: &proto.PlanComplete{
+						DailyCost: 1,
+					},
+				},
+			}},
+			ProvisionApply: echo.ApplyComplete,
 			ProvisionGraph: []*proto.Response{{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
@@ -573,7 +603,16 @@ func TestWorkspaceQuota(t *testing.T) {
 		template1 := coderdtest.CreateTemplate(t, owner, first.OrganizationID, version1.ID)
 
 		version2 := coderdtest.CreateTemplateVersion(t, owner, second.ID, &echo.Responses{
-			Parse: echo.ParseComplete,
+			Parse:         echo.ParseComplete,
+			ProvisionInit: echo.InitComplete,
+			ProvisionPlan: []*proto.Response{{
+				Type: &proto.Response_Plan{
+					Plan: &proto.PlanComplete{
+						DailyCost: 1,
+					},
+				},
+			}},
+			ProvisionApply: echo.ApplyComplete,
 			ProvisionGraph: []*proto.Response{{
 				Type: &proto.Response_Graph{
 					Graph: &proto.GraphComplete{
