c229842f57
The directory created by `T.TempDir` is automatically removed when the test and all its subtests complete. Reference: https://pkg.go.dev/testing#T.TempDir Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
96 lines
1.7 KiB
Go
96 lines
1.7 KiB
Go
package stores
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/filecoin-project/lotus/extern/sector-storage/fsutil"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const pathSize = 16 << 20
|
|
|
|
type TestingLocalStorage struct {
|
|
root string
|
|
c StorageConfig
|
|
}
|
|
|
|
func (t *TestingLocalStorage) DiskUsage(path string) (int64, error) {
|
|
return 1, nil
|
|
}
|
|
|
|
func (t *TestingLocalStorage) GetStorage() (StorageConfig, error) {
|
|
return t.c, nil
|
|
}
|
|
|
|
func (t *TestingLocalStorage) SetStorage(f func(*StorageConfig)) error {
|
|
f(&t.c)
|
|
return nil
|
|
}
|
|
|
|
func (t *TestingLocalStorage) Stat(path string) (fsutil.FsStat, error) {
|
|
return fsutil.FsStat{
|
|
Capacity: pathSize,
|
|
Available: pathSize,
|
|
FSAvailable: pathSize,
|
|
}, nil
|
|
}
|
|
|
|
func (t *TestingLocalStorage) init(subpath string) error {
|
|
path := filepath.Join(t.root, subpath)
|
|
if err := os.Mkdir(path, 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
metaFile := filepath.Join(path, MetaFile)
|
|
|
|
meta := &LocalStorageMeta{
|
|
ID: ID(uuid.New().String()),
|
|
Weight: 1,
|
|
CanSeal: true,
|
|
CanStore: true,
|
|
}
|
|
|
|
mb, err := json.MarshalIndent(meta, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := ioutil.WriteFile(metaFile, mb, 0644); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var _ LocalStorage = &TestingLocalStorage{}
|
|
|
|
func TestLocalStorage(t *testing.T) {
|
|
ctx := context.TODO()
|
|
|
|
root := t.TempDir()
|
|
|
|
tstor := &TestingLocalStorage{
|
|
root: root,
|
|
}
|
|
|
|
index := NewIndex()
|
|
|
|
st, err := NewLocal(ctx, tstor, index, nil)
|
|
require.NoError(t, err)
|
|
|
|
p1 := "1"
|
|
require.NoError(t, tstor.init("1"))
|
|
|
|
err = st.OpenPath(ctx, filepath.Join(tstor.root, p1))
|
|
require.NoError(t, err)
|
|
|
|
// TODO: put more things here
|
|
}
|