diff --git a/bitcask_test.go b/bitcask_test.go index 23d2ea9..8548801 100644 --- a/bitcask_test.go +++ b/bitcask_test.go @@ -3,6 +3,7 @@ package bitcask import ( "fmt" "io/ioutil" + "os" "reflect" "sort" "strings" @@ -503,11 +504,17 @@ type benchmarkTestCase struct { } func BenchmarkGet(b *testing.B) { - testdir, err := ioutil.TempDir("", "bitcask") + currentDir, err := os.Getwd() if err != nil { b.Fatal(err) } + testdir, err := ioutil.TempDir(currentDir, "bitcask_bench") + if err != nil { + b.Fatal(err) + } + defer os.RemoveAll(testdir) + db, err := Open(testdir) if err != nil { b.Fatal(err) @@ -553,11 +560,17 @@ func BenchmarkGet(b *testing.B) { } func BenchmarkPut(b *testing.B) { - testdir, err := ioutil.TempDir("", "bitcask") + currentDir, err := os.Getwd() if err != nil { b.Fatal(err) } + testdir, err := ioutil.TempDir(currentDir, "bitcask") + if err != nil { + b.Fatal(err) + } + defer os.RemoveAll(testdir) + db, err := Open(testdir) if err != nil { b.Fatal(err) @@ -594,11 +607,17 @@ func BenchmarkPut(b *testing.B) { } func BenchmarkScan(b *testing.B) { - testdir, err := ioutil.TempDir("", "bitcask") + currentDir, err := os.Getwd() if err != nil { b.Fatal(err) } + testdir, err := ioutil.TempDir(currentDir, "bitcask") + if err != nil { + b.Fatal(err) + } + defer os.RemoveAll(testdir) + db, err := Open(testdir) if err != nil { b.Fatal(err)