forked from TrueCloudLab/frostfs-node
[#327] tests: replace os.MkdirTemp with t.TempDir
Signed-off-by: Alejandro Lopez <a.lopez@yadro.com>
This commit is contained in:
parent
cedd07bbc8
commit
973af12854
6 changed files with 9 additions and 36 deletions
|
@ -2,7 +2,6 @@ package blobstor
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"os"
|
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
objectCore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
|
objectCore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
|
||||||
|
@ -15,13 +14,9 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestExists(t *testing.T) {
|
func TestExists(t *testing.T) {
|
||||||
dir, err := os.MkdirTemp("", "frostfs*")
|
|
||||||
require.NoError(t, err)
|
|
||||||
t.Cleanup(func() { _ = os.RemoveAll(dir) })
|
|
||||||
|
|
||||||
const smallSizeLimit = 512
|
const smallSizeLimit = 512
|
||||||
|
|
||||||
storages, _, largeFileStorage := defaultTestStorages(dir, smallSizeLimit)
|
storages, _, largeFileStorage := defaultTestStorages(t.TempDir(), smallSizeLimit)
|
||||||
|
|
||||||
b := New(WithStorages(storages))
|
b := New(WithStorages(storages))
|
||||||
|
|
||||||
|
|
|
@ -3,7 +3,6 @@ package blobstor
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree"
|
||||||
|
@ -21,18 +20,13 @@ type storage struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s storage) open(b *testing.B) common.Storage {
|
func (s storage) open(b *testing.B) common.Storage {
|
||||||
dir, err := os.MkdirTemp(os.TempDir(), s.desc)
|
st := s.create(b.TempDir())
|
||||||
if err != nil {
|
|
||||||
b.Fatalf("creating %s root path: %v", s.desc, err)
|
|
||||||
}
|
|
||||||
st := s.create(dir)
|
|
||||||
|
|
||||||
require.NoError(b, st.Open(false))
|
require.NoError(b, st.Open(false))
|
||||||
require.NoError(b, st.Init())
|
require.NoError(b, st.Init())
|
||||||
|
|
||||||
b.Cleanup(func() {
|
b.Cleanup(func() {
|
||||||
require.NoError(b, st.Close())
|
require.NoError(b, st.Close())
|
||||||
require.NoError(b, os.RemoveAll(dir))
|
|
||||||
})
|
})
|
||||||
|
|
||||||
return st
|
return st
|
||||||
|
|
|
@ -233,8 +233,7 @@ func TestExecBlocks(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestPersistentShardID(t *testing.T) {
|
func TestPersistentShardID(t *testing.T) {
|
||||||
dir, err := os.MkdirTemp("", "*")
|
dir := t.TempDir()
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
te := newEngineWithErrorThreshold(t, dir, 1)
|
te := newEngineWithErrorThreshold(t, dir, 1)
|
||||||
|
|
||||||
|
|
|
@ -41,11 +41,7 @@ type testShard struct {
|
||||||
|
|
||||||
func newEngineWithErrorThreshold(t testing.TB, dir string, errThreshold uint32) *testEngine {
|
func newEngineWithErrorThreshold(t testing.TB, dir string, errThreshold uint32) *testEngine {
|
||||||
if dir == "" {
|
if dir == "" {
|
||||||
var err error
|
dir = t.TempDir()
|
||||||
|
|
||||||
dir, err = os.MkdirTemp("", "*")
|
|
||||||
require.NoError(t, err)
|
|
||||||
t.Cleanup(func() { _ = os.RemoveAll(dir) })
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var testShards [2]*testShard
|
var testShards [2]*testShard
|
||||||
|
@ -171,9 +167,7 @@ func TestErrorReporting(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestBlobstorFailback(t *testing.T) {
|
func TestBlobstorFailback(t *testing.T) {
|
||||||
dir, err := os.MkdirTemp("", "*")
|
dir := t.TempDir()
|
||||||
require.NoError(t, err)
|
|
||||||
t.Cleanup(func() { require.NoError(t, os.RemoveAll(dir)) })
|
|
||||||
|
|
||||||
te := newEngineWithErrorThreshold(t, dir, 1)
|
te := newEngineWithErrorThreshold(t, dir, 1)
|
||||||
|
|
||||||
|
@ -185,7 +179,7 @@ func TestBlobstorFailback(t *testing.T) {
|
||||||
var prm shard.PutPrm
|
var prm shard.PutPrm
|
||||||
prm.SetObject(obj)
|
prm.SetObject(obj)
|
||||||
te.ng.mtx.RLock()
|
te.ng.mtx.RLock()
|
||||||
_, err = te.ng.shards[te.shards[0].id.String()].Shard.Put(context.Background(), prm)
|
_, err := te.ng.shards[te.shards[0].id.String()].Shard.Put(context.Background(), prm)
|
||||||
te.ng.mtx.RUnlock()
|
te.ng.mtx.RUnlock()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
objs = append(objs, obj)
|
objs = append(objs, obj)
|
||||||
|
@ -193,7 +187,7 @@ func TestBlobstorFailback(t *testing.T) {
|
||||||
|
|
||||||
for i := range objs {
|
for i := range objs {
|
||||||
addr := object.AddressOf(objs[i])
|
addr := object.AddressOf(objs[i])
|
||||||
_, err = te.ng.Get(context.Background(), GetPrm{addr: addr})
|
_, err := te.ng.Get(context.Background(), GetPrm{addr: addr})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
_, err = te.ng.GetRange(context.Background(), RngPrm{addr: addr})
|
_, err = te.ng.GetRange(context.Background(), RngPrm{addr: addr})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
|
@ -4,7 +4,6 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -25,9 +24,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func newEngineEvacuate(t *testing.T, shardNum int, objPerShard int) (*StorageEngine, []*shard.ID, []*objectSDK.Object) {
|
func newEngineEvacuate(t *testing.T, shardNum int, objPerShard int) (*StorageEngine, []*shard.ID, []*objectSDK.Object) {
|
||||||
dir, err := os.MkdirTemp("", "*")
|
dir := t.TempDir()
|
||||||
require.NoError(t, err)
|
|
||||||
t.Cleanup(func() { _ = os.RemoveAll(dir) })
|
|
||||||
|
|
||||||
te := testNewEngine(t, WithShardPoolSize(1)).
|
te := testNewEngine(t, WithShardPoolSize(1)).
|
||||||
setShardsNumOpts(t, shardNum, func(id int) []shard.Option {
|
setShardsNumOpts(t, shardNum, func(id int) []shard.Option {
|
||||||
|
|
|
@ -4,7 +4,6 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"sync"
|
"sync"
|
||||||
|
@ -31,19 +30,14 @@ var providers = []struct {
|
||||||
return f
|
return f
|
||||||
}},
|
}},
|
||||||
{"bbolt", func(t testing.TB, opts ...Option) Forest {
|
{"bbolt", func(t testing.TB, opts ...Option) Forest {
|
||||||
// Use `os.TempDir` because we construct multiple times in the same test.
|
|
||||||
tmpDir, err := os.MkdirTemp(os.TempDir(), "*")
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
f := NewBoltForest(
|
f := NewBoltForest(
|
||||||
append([]Option{
|
append([]Option{
|
||||||
WithPath(filepath.Join(tmpDir, "test.db")),
|
WithPath(filepath.Join(t.TempDir(), "test.db")),
|
||||||
WithMaxBatchSize(1)}, opts...)...)
|
WithMaxBatchSize(1)}, opts...)...)
|
||||||
require.NoError(t, f.Open(false))
|
require.NoError(t, f.Open(false))
|
||||||
require.NoError(t, f.Init())
|
require.NoError(t, f.Init())
|
||||||
t.Cleanup(func() {
|
t.Cleanup(func() {
|
||||||
require.NoError(t, f.Close())
|
require.NoError(t, f.Close())
|
||||||
require.NoError(t, os.RemoveAll(tmpDir))
|
|
||||||
})
|
})
|
||||||
return f
|
return f
|
||||||
}},
|
}},
|
||||||
|
|
Loading…
Reference in a new issue