mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-22 03:30:37 +00:00
51 lines
1.6 KiB
Go
51 lines
1.6 KiB
Go
package sentinel
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/golang/snappy"
|
|
"github.com/ledgerwatch/erigon/cl/clparams"
|
|
"github.com/ledgerwatch/erigon/cl/utils"
|
|
pubsubpb "github.com/libp2p/go-libp2p-pubsub/pb"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMsgID(t *testing.T) {
|
|
g := clparams.GenesisConfigs[clparams.MainnetNetwork]
|
|
n := clparams.NetworkConfigs[clparams.MainnetNetwork]
|
|
s := &Sentinel{
|
|
ctx: context.TODO(),
|
|
cfg: &SentinelConfig{
|
|
BeaconConfig: &clparams.MainnetBeaconConfig,
|
|
GenesisConfig: &g,
|
|
NetworkConfig: &n,
|
|
},
|
|
}
|
|
d := [4]byte{108, 122, 33, 65}
|
|
tpc := fmt.Sprintf("/eth2/%x/beacon_block", d)
|
|
topicLen := uint64(len(tpc))
|
|
topicLenBytes := utils.Uint64ToLE(topicLen)
|
|
invalidSnappy := [32]byte{'J', 'U', 'N', 'K'}
|
|
pMsg := &pubsubpb.Message{Data: invalidSnappy[:], Topic: &tpc}
|
|
// Create object to hash
|
|
combinedObj := append(n.MessageDomainInvalidSnappy[:], topicLenBytes...)
|
|
combinedObj = append(combinedObj, tpc...)
|
|
combinedObj = append(combinedObj, pMsg.Data...)
|
|
hashedData := utils.Sha256(combinedObj)
|
|
msgID := string(hashedData[:20])
|
|
require.Equal(t, msgID, s.msgId(pMsg), "Got incorrect msg id")
|
|
|
|
validObj := [32]byte{'v', 'a', 'l', 'i', 'd'}
|
|
enc := snappy.Encode(nil, validObj[:])
|
|
nMsg := &pubsubpb.Message{Data: enc, Topic: &tpc}
|
|
// Create object to hash
|
|
combinedObj = append(n.MessageDomainValidSnappy[:], topicLenBytes...)
|
|
combinedObj = append(combinedObj, tpc...)
|
|
combinedObj = append(combinedObj, validObj[:]...)
|
|
hashedData = utils.Sha256(combinedObj)
|
|
msgID = string(hashedData[:20])
|
|
require.Equal(t, msgID, s.msgId(nMsg), "Got incorrect msg id")
|
|
}
|