internal/database: replace os.MkdirTemp with T.TempDir
Updates [#45402](https://go.dev/issue/45402).
Change-Id: I7e14a6923af21fd5ce4979c648734155a3468558
GitHub-Last-Rev: 81004b22c5414997153b951de71ba7783d6dcb1d
GitHub-Pull-Request: golang/vulndb#1508
Reviewed-on: https://go-review.googlesource.com/c/vulndb/+/463098
Reviewed-by: Tatiana Bradley <tatianabradley@google.com>
TryBot-Result: Gopher Robot <gobot@golang.org>
Run-TryBot: Tim King <taking@google.com>
Reviewed-by: Tim King <taking@google.com>
diff --git a/internal/database/write_test.go b/internal/database/write_test.go
index 30bf4eb..d0e0c96 100644
--- a/internal/database/write_test.go
+++ b/internal/database/write_test.go
@@ -6,7 +6,6 @@
import (
"fmt"
- "os"
"testing"
"github.com/google/go-cmp/cmp"
@@ -14,22 +13,13 @@
)
func TestWrite(t *testing.T) {
- tempDir, err := os.MkdirTemp("", "testWrite")
- if err != nil {
- t.Fatal(err)
- }
-
- defer func() {
- if err := os.RemoveAll(tempDir); err != nil {
- t.Log(err)
- }
- }()
+ tempDir := t.TempDir()
if err := valid.Write(tempDir, true); err != nil {
t.Fatal(err)
}
- if err = cmpDirHashes(tempDir, validDir); err != nil {
+ if err := cmpDirHashes(tempDir, validDir); err != nil {
t.Error(err)
}
}
@@ -53,16 +43,7 @@
// Check that Write and Load are inverses.
func TestWriteLoad(t *testing.T) {
- tempDir, err := os.MkdirTemp("", "testWriteLoad")
- if err != nil {
- t.Fatal(err)
- }
-
- defer func() {
- if err := os.RemoveAll(tempDir); err != nil {
- t.Log(err)
- }
- }()
+ tempDir := t.TempDir()
written := valid
if err := written.Write(tempDir, false); err != nil {
