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

fix: test case error

Browse source
This commit is contained in:
fearlessfe 2024-09-04 07:52:10 +08:00
parent fbe6854076
commit 1b58d9431b
1 changed files with 31 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
portalnetwork/beacon/beacon_network_test.go
View file

@ -76,7 +76,7 @@ func setupBeaconNetwork(addr string, bootNodes []*enode.Node) (*BeaconNetwork, e
} }
func TestBeaconNetworkContent(t *testing.T) { func TestBeaconNetworkContent(t *testing.T) {
logger := testlog.Logger(t, log.LvlTrace) logger := testlog.Logger(t, log.LevelError)
node1, err := setupBeaconNetwork(":6666", nil) node1, err := setupBeaconNetwork(":6666", nil)
assert.NoError(t, err) assert.NoError(t, err)
node1.log = logger node1.log = logger
@ -84,7 +84,6 @@ func TestBeaconNetworkContent(t *testing.T) {
err = node1.Start() err = node1.Start()
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(10 * time.Second)
node2, err := setupBeaconNetwork(":6667", []*enode.Node{node1.portalProtocol.Self()}) node2, err := setupBeaconNetwork(":6667", []*enode.Node{node1.portalProtocol.Self()})
assert.NoError(t, err) assert.NoError(t, err)
@ -93,8 +92,6 @@ func TestBeaconNetworkContent(t *testing.T) {
err = node2.Start() err = node2.Start()
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(10 * time.Second)
node3, err := setupBeaconNetwork(":6668", []*enode.Node{node1.portalProtocol.Self()}) node3, err := setupBeaconNetwork(":6668", []*enode.Node{node1.portalProtocol.Self()})
assert.NoError(t, err) assert.NoError(t, err)
node3.log = logger node3.log = logger
@ -102,8 +99,6 @@ func TestBeaconNetworkContent(t *testing.T) {
err = node3.Start() err = node3.Start()
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(10 * time.Second)
filePath := "testdata/light_client_updates_by_range.json" filePath := "testdata/light_client_updates_by_range.json"
jsonStr, _ := os.ReadFile(filePath) jsonStr, _ := os.ReadFile(filePath)
@ -115,13 +110,13 @@ func TestBeaconNetworkContent(t *testing.T) {
for _, v := range result { for _, v := range result {
kBytes, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string)) kBytes, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string))
vBytes, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string)) vBytes, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string))
contentKey := storage.NewContentKey(LightClientUpdate, kBytes).Encode() // contentKey := storage.NewContentKey(LightClientUpdate, kBytes).Encode()
_, err = node1.portalProtocol.Gossip(&id, [][]byte{contentKey}, [][]byte{vBytes}) _, err = node1.portalProtocol.Gossip(&id, [][]byte{kBytes}, [][]byte{vBytes})
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(3 * time.Second) time.Sleep(3 * time.Second)
contentId := node2.portalProtocol.ToContentId(contentKey) contentId := node2.portalProtocol.ToContentId(kBytes)
get, err := node2.portalProtocol.Get(contentKey, contentId) get, err := node2.portalProtocol.Get(kBytes, contentId)
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, vBytes, get) assert.Equal(t, vBytes, get)
} }
@ -137,16 +132,16 @@ func TestBeaconNetworkContent(t *testing.T) {
kBytes1, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string)) kBytes1, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string))
vBytes1, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string)) vBytes1, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string))
contentKey1 := storage.NewContentKey(LightClientFinalityUpdate, kBytes1).Encode() // contentKey1 := storage.NewContentKey(LightClientFinalityUpdate, kBytes1).Encode()
id = node1.portalProtocol.Self().ID() id = node1.portalProtocol.Self().ID()
_, err = node1.portalProtocol.Gossip(&id, [][]byte{contentKey1}, [][]byte{vBytes1}) _, err = node1.portalProtocol.Gossip(&id, [][]byte{kBytes1}, [][]byte{vBytes1})
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(3 * time.Second) time.Sleep(3 * time.Second)
contentId1 := node2.portalProtocol.ToContentId(contentKey1) contentId1 := node2.portalProtocol.ToContentId(kBytes1)
get1, err := node2.portalProtocol.Get(contentKey1, contentId1) get1, err := node2.portalProtocol.Get(kBytes1, contentId1)
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, vBytes1, get1) assert.Equal(t, vBytes1, get1)
} }
@ -162,44 +157,44 @@ func TestBeaconNetworkContent(t *testing.T) {
kBytes2, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string)) kBytes2, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string))
vBytes2, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string)) vBytes2, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string))
contentKey2 := storage.NewContentKey(LightClientOptimisticUpdate, kBytes2).Encode() // contentKey2 := storage.NewContentKey(LightClientOptimisticUpdate, kBytes2).Encode()
id = node1.portalProtocol.Self().ID() id = node1.portalProtocol.Self().ID()
_, err = node1.portalProtocol.Gossip(&id, [][]byte{contentKey2}, [][]byte{vBytes2}) _, err = node1.portalProtocol.Gossip(&id, [][]byte{kBytes2}, [][]byte{vBytes2})
assert.NoError(t, err) assert.NoError(t, err)
time.Sleep(3 * time.Second) time.Sleep(3 * time.Second)
contentId2 := node2.portalProtocol.ToContentId(contentKey2) contentId2 := node2.portalProtocol.ToContentId(kBytes2)
get2, err := node2.portalProtocol.Get(contentKey2, contentId2) get2, err := node2.portalProtocol.Get(kBytes2, contentId2)
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, vBytes2, get2) assert.Equal(t, vBytes2, get2)
} }
filePath3 := "testdata/light_client_bootstrap.json" // filePath3 := "testdata/light_client_bootstrap.json"
jsonStr3, _ := os.ReadFile(filePath3) // jsonStr3, _ := os.ReadFile(filePath3)
var result3 map[string]interface{} // var result3 map[string]interface{}
err = json.Unmarshal(jsonStr3, &result3) // err = json.Unmarshal(jsonStr3, &result3)
assert.NoError(t, err) // assert.NoError(t, err)
for _, v := range result3 { // for _, v := range result3 {
kBytes3, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string)) // kBytes3, _ := hexutil.Decode(v.(map[string]interface{})["content_key"].(string))
vBytes3, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string)) // vBytes3, _ := hexutil.Decode(v.(map[string]interface{})["content_value"].(string))
contentKey3 := storage.NewContentKey(LightClientBootstrap, kBytes3).Encode() // contentKey3 := storage.NewContentKey(LightClientBootstrap, kBytes3).Encode()
id = node1.portalProtocol.Self().ID() // id = node1.portalProtocol.Self().ID()
_, err = node1.portalProtocol.Gossip(&id, [][]byte{contentKey3}, [][]byte{vBytes3}) // _, err = node1.portalProtocol.Gossip(&id, [][]byte{contentKey3}, [][]byte{vBytes3})
assert.NoError(t, err) // assert.NoError(t, err)
time.Sleep(3 * time.Second) // time.Sleep(3 * time.Second)
contentId3 := node2.portalProtocol.ToContentId(contentKey3) // contentId3 := node2.portalProtocol.ToContentId(contentKey3)
get3, err := node2.portalProtocol.Get(contentKey3, contentId3) // get3, err := node2.portalProtocol.Get(contentKey3, contentId3)
assert.NoError(t, err) // assert.NoError(t, err)
assert.Equal(t, vBytes3, get3) // assert.Equal(t, vBytes3, get3)
} // }
} }
type Entry struct { type Entry struct {