forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-15 16:33:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: unit test error

Browse source
This commit is contained in:
fearlessfe 2024-11-21 21:54:02 +08:00 committed by Chen Kai
parent ad2fd55e7a
commit 6388d8b3db
1 changed files with 23 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
portalnetwork/storage/ethpepple/storage_test.go
View file

@ -1,6 +1,8 @@
package ethpepple package ethpepple
import ( import (
"crypto/rand"
"encoding/hex"
"os" "os"
"testing" "testing"
@ -10,16 +12,24 @@ import (
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
const dataDir = "./node1"
var testRadius = uint256.NewInt(100000) var testRadius = uint256.NewInt(100000)
func clearNodeData() { func clearNodeData(path string) {
_ = os.RemoveAll(dataDir) _ = os.RemoveAll(path)
} }
func getTestDb() (storage.ContentStorage, error) { func getRandomPath() string {
db, err := NewPeppleDB(dataDir, 100, 100, "history") // gen a random hex string
bytes := make([]byte, 32)
_, err := rand.Read(bytes)
if err != nil {
panic(err)
}
return hex.EncodeToString(bytes)
}
func getTestDb(path string) (storage.ContentStorage, error) {
db, err := NewPeppleDB(path, 100, 100, "history")
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -33,9 +43,10 @@ func getTestDb() (storage.ContentStorage, error) {
} }
func TestReadRadius(t *testing.T) { func TestReadRadius(t *testing.T) {
db, err := getTestDb() path := getRandomPath()
db, err := getTestDb(path)
assert.NoError(t, err) assert.NoError(t, err)
defer clearNodeData() defer clearNodeData(path)
assert.True(t, db.Radius().Eq(storage.MaxDistance)) assert.True(t, db.Radius().Eq(storage.MaxDistance))
data, err := testRadius.MarshalSSZ() data, err := testRadius.MarshalSSZ()
@ -46,15 +57,16 @@ func TestReadRadius(t *testing.T) {
err = store.db.Close() err = store.db.Close()
assert.NoError(t, err) assert.NoError(t, err)
db, err = getTestDb() db, err = getTestDb(path)
assert.NoError(t, err) assert.NoError(t, err)
assert.True(t, db.Radius().Eq(testRadius)) assert.True(t, db.Radius().Eq(testRadius))
} }
func TestStorage(t *testing.T) { func TestStorage(t *testing.T) {
db, err := getTestDb() path := getRandomPath()
db, err := getTestDb(path)
assert.NoError(t, err) assert.NoError(t, err)
defer clearNodeData() defer clearNodeData(path)
testcases := map[string][]byte{ testcases := map[string][]byte{
"test1": []byte("test1"), "test1": []byte("test1"),
"test2": []byte("test2"), "test2": []byte("test2"),