mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-08 23:18:15 -05:00
28 lines
1016 B
Go
28 lines
1016 B
Go
package peerdas_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/OffchainLabs/prysm/v7/beacon-chain/core/peerdas"
|
|
"github.com/OffchainLabs/prysm/v7/testing/require"
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
)
|
|
|
|
func TestInfo(t *testing.T) {
|
|
nodeID := enode.ID{}
|
|
custodyGroupCount := uint64(7)
|
|
|
|
expectedCustodyGroup := map[uint64]bool{1: true, 17: true, 19: true, 42: true, 75: true, 87: true, 102: true}
|
|
expectedCustodyColumns := map[uint64]bool{1: true, 17: true, 19: true, 42: true, 75: true, 87: true, 102: true}
|
|
expectedDataColumnsSubnets := map[uint64]bool{1: true, 17: true, 19: true, 42: true, 75: true, 87: true, 102: true}
|
|
|
|
for _, cached := range []bool{false, true} {
|
|
actual, ok, err := peerdas.Info(nodeID, custodyGroupCount)
|
|
require.NoError(t, err)
|
|
require.Equal(t, cached, ok)
|
|
require.DeepEqual(t, expectedCustodyGroup, actual.CustodyGroups)
|
|
require.DeepEqual(t, expectedCustodyColumns, actual.CustodyColumns)
|
|
require.DeepEqual(t, expectedDataColumnsSubnets, actual.DataColumnsSubnets)
|
|
}
|
|
}
|