gopls: fix a couple places where temporary files are not removed

Fix two places where gopls fails to clean up temporary files:
- In telemetry tests, the deferred cleanup was not run after os.Exit.
- In the GC details codelens, a persistent GC details directory is
  assumed; just use a temp directory instead.

Change-Id: Icef5a4b612ac1727fee7d2c65e99a90f73123081
Reviewed-on: https://go-review.googlesource.com/c/tools/+/579755
Reviewed-by: Alan Donovan <[email protected]>
Auto-Submit: Robert Findley <[email protected]>
LUCI-TryBot-Result: Go LUCI <[email protected]>

Author: findleyr

gopls/internal/golang/gc_annotations.go

@@ -17,6 +17,7 @@ import (
 	"golang.org/x/tools/gopls/internal/cache/metadata"
 	"golang.org/x/tools/gopls/internal/protocol"
 	"golang.org/x/tools/gopls/internal/settings"
+	"golang.org/x/tools/internal/event"
 	"golang.org/x/tools/internal/gocommand"
 )
 
@@ -25,11 +26,16 @@ func GCOptimizationDetails(ctx context.Context, snapshot *cache.Snapshot, mp *me
 		return nil, nil
 	}
 	pkgDir := filepath.Dir(mp.CompiledGoFiles[0].Path())
-	outDir := filepath.Join(os.TempDir(), fmt.Sprintf("gopls-%d.details", os.Getpid()))
-
-	if err := os.MkdirAll(outDir, 0700); err != nil {
+	outDir, err := os.MkdirTemp("", fmt.Sprintf("gopls-%d.details", os.Getpid()))
+	if err != nil {
 		return nil, err
 	}
+	defer func() {
+		if err := os.RemoveAll(outDir); err != nil {
+			event.Error(ctx, "cleaning gcdetails dir", err)
+		}
+	}()
+
 	tmpFile, err := os.CreateTemp(os.TempDir(), "gopls-x")
 	if err != nil {
 		return nil, err

gopls/internal/telemetry/telemetry_test.go

@@ -31,8 +31,9 @@ func TestMain(m *testing.M) {
 		panic(err)
 	}
 	countertest.Open(tmp)
-	defer os.RemoveAll(tmp)
-	Main(m)
+	code := Main(m)
+	os.RemoveAll(tmp)
+	os.Exit(code)
 }
 
 func TestTelemetry(t *testing.T) {

gopls/internal/test/integration/codelens/codelens_test.go

@@ -6,6 +6,7 @@ package codelens
 
 import (
 	"fmt"
+	"os"
 	"testing"
 
 	"golang.org/x/tools/gopls/internal/server"
@@ -20,7 +21,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	Main(m)
+	os.Exit(Main(m))
 }
 
 func TestDisablingCodeLens(t *testing.T) {

gopls/internal/test/integration/completion/completion_test.go

@@ -6,6 +6,7 @@ package completion
 
 import (
 	"fmt"
+	"os"
 	"sort"
 	"strings"
 	"testing"
@@ -24,7 +25,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	Main(m)
+	os.Exit(Main(m))
 }
 
 const proxy = `

gopls/internal/test/integration/debug/debug_test.go

@@ -9,6 +9,7 @@ import (
 	"encoding/json"
 	"io"
 	"net/http"
+	"os"
 	"strings"
 	"testing"
 
@@ -19,7 +20,7 @@ import (
 )
 
 func TestMain(m *testing.M) {
-	Main(m)
+	os.Exit(Main(m))
 }
 
 func TestBugNotification(t *testing.T) {

gopls/internal/test/integration/diagnostics/diagnostics_test.go

@@ -7,6 +7,7 @@ package diagnostics
 import (
 	"context"
 	"fmt"
+	"os"
 	"os/exec"
 	"testing"
 
@@ -20,7 +21,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	Main(m)
+	os.Exit(Main(m))
 }
 
 // Use mod.com for all go.mod files due to golang/go#35230.

gopls/internal/test/integration/doc.go

@@ -33,7 +33,7 @@
 //	)
 //
 //	func TestMain(m *testing.M) {
-//		Main(m, hooks.Options)
+//		os.Exit(Main(m, hooks.Options))
 //	}
 //
 // # Writing a simple integration test

gopls/internal/test/integration/inlayhints/inlayhints_test.go

@@ -4,6 +4,7 @@
 package inlayhint
 
 import (
+	"os"
 	"testing"
 
 	"golang.org/x/tools/gopls/internal/golang"
@@ -13,7 +14,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	Main(m)
+	os.Exit(Main(m))
 }
 
 func TestEnablingInlayHints(t *testing.T) {

gopls/internal/test/integration/misc/misc_test.go

@@ -5,6 +5,7 @@
 package misc
 
 import (
+	"os"
 	"strings"
 	"testing"
 
@@ -16,7 +17,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	integration.Main(m)
+	os.Exit(integration.Main(m))
 }
 
 // TestDocumentURIFix ensures that a DocumentURI supplied by the

gopls/internal/test/integration/modfile/modfile_test.go

@@ -5,6 +5,7 @@
 package modfile
 
 import (
+	"os"
 	"path/filepath"
 	"runtime"
 	"strings"
@@ -19,7 +20,7 @@ import (
 
 func TestMain(m *testing.M) {
 	bug.PanicOnBugs = true
-	Main(m)
+	os.Exit(Main(m))
 }
 
 const workspaceProxy = `