update expectations in TestWorkspaceBuildWithPreset

SasSwart · SasSwart · commit 86d93c5a4ea6 · 2025-04-15T14:25:26.000Z
diff --git a/coderd/wsbuilder/wsbuilder_test.go b/coderd/wsbuilder/wsbuilder_test.go
@@ -789,6 +789,7 @@ func TestWorkspaceBuildWithPreset(t *testing.T) {
 		// Inputs
 		withTemplate,
 		withActiveVersion(nil),
+		withTemplateVersionPreset(presetID),
 		withLastBuildNotFound,
 		withTemplateVersionVariables(activeVersionID, nil),
 		withParameterSchemas(activeJobID, nil),
@@ -960,6 +961,12 @@ func withInactiveVersion(params []database.TemplateVersionParameter) func(mTx *d
 	}
 }
 
+func withTemplateVersionPreset(presetID uuid.UUID) func(mTx *dbmock.MockStore) {
+	return func(mTx *dbmock.MockStore) {
+		mTx.EXPECT().GetPresetParametersByPresetID(gomock.Any(), presetID).Return(nil, nil)
+	}
+}
+
 func withLastBuildFound(mTx *dbmock.MockStore) {
 	mTx.EXPECT().GetLatestWorkspaceBuildByWorkspaceID(gomock.Any(), workspaceID).
 		Times(1).
