Fix context builder test failure

runatlantis · Sep 27, 2021 · c9a3f17 · c9a3f17
1 parent dd0a3e6
commit c9a3f17
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/server/events/project_command_context_builder.go b/server/events/project_command_context_builder.go
@@ -16,17 +16,19 @@ func NewProjectCommandContextBulder(policyCheckEnabled bool, commentBuilder Comm
 		CommentBuilder: commentBuilder,
 	}
 
+	contextBuilderWithStats := &CommandScopedStatsProjectCommandContextBuilder{
+		ProjectCommandContextBuilder: projectCommandContextBuilder,
+		ProjectCounter:               scope.NewCounter("projects"),
+	}
+
 	if policyCheckEnabled {
 		return &PolicyCheckProjectCommandContextBuilder{
 			CommentBuilder:               commentBuilder,
-			ProjectCommandContextBuilder: projectCommandContextBuilder,
+			ProjectCommandContextBuilder: contextBuilderWithStats,
 		}
 	}
 
-	return &CommandScopedStatsProjectCommandContextBuilder{
-		ProjectCommandContextBuilder: projectCommandContextBuilder,
-		ProjectCounter:               scope.NewCounter("projects"),
-	}
+	return projectCommandContextBuilder
 }
 
 type ContextFlags struct {
@@ -143,8 +145,8 @@ func (cb *DefaultProjectCommandContextBuilder) BuildProjectContext(
 }
 
 type PolicyCheckProjectCommandContextBuilder struct {
-	ProjectCommandContextBuilder *DefaultProjectCommandContextBuilder
-	CommentBuilder               CommentBuilder
+	ProjectCommandContextBuilder
+	CommentBuilder CommentBuilder
 }
 
 func (cb *PolicyCheckProjectCommandContextBuilder) BuildProjectContext(