add unit tests

AbhineetJain · AbhineetJain · commit 2726aceb5258 · 2022-06-16T21:27:32.000Z
diff --git a/coderd/autobuild/executor/lifecycle_executor_test.go b/coderd/autobuild/executor/lifecycle_executor_test.go
@@ -51,6 +51,10 @@ func TestExecutorAutostartOK(t *testing.T) {
 	assert.Len(t, stats.Transitions, 1)
 	assert.Contains(t, stats.Transitions, workspace.ID)
 	assert.Equal(t, database.WorkspaceTransitionStart, stats.Transitions[workspace.ID])
+
+	workspace, err := client.Workspace(context.Background(), workspace.ID)
+	assert.NoError(t, err)
+	assert.Equal(t, codersdk.BuildReasonAutostart, workspace.LatestBuild.Reason)
 }
 
 func TestExecutorAutostartTemplateUpdated(t *testing.T) {
@@ -200,6 +204,10 @@ func TestExecutorAutostopOK(t *testing.T) {
 	assert.Len(t, stats.Transitions, 1)
 	assert.Contains(t, stats.Transitions, workspace.ID)
 	assert.Equal(t, database.WorkspaceTransitionStop, stats.Transitions[workspace.ID])
+
+	workspace, err := client.Workspace(context.Background(), workspace.ID)
+	assert.NoError(t, err)
+	assert.Equal(t, codersdk.BuildReasonAutostop, workspace.LatestBuild.Reason)
 }
 
 func TestExecutorAutostopExtend(t *testing.T) {
diff --git a/coderd/workspacebuilds.go b/coderd/workspacebuilds.go
@@ -640,6 +640,7 @@ func convertWorkspaceBuild(
 		InitiatorUsername:  initiatorName,
 		Job:                convertProvisionerJob(job),
 		Deadline:           workspaceBuild.Deadline,
+		Reason:             codersdk.BuildReason(workspaceBuild.Reason),
 	}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -640,6 +640,7 @@ func convertWorkspaceBuild(`
`640`	`640`	`InitiatorUsername: initiatorName,`
`641`	`641`	`Job: convertProvisionerJob(job),`
`642`	`642`	`Deadline: workspaceBuild.Deadline,`
	`643`	`+ Reason: codersdk.BuildReason(workspaceBuild.Reason),`
`643`	`644`	`}`
`644`	`645`	`}`
`645`	`646`