mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 23:48:06 -05:00
7e32bbc199
* wip * wip * adding ssz marshalling * updating function for readability * adding unit tests and fixing ssz * gaz * linting * fixing test * wip * fixing mock blocker * fixing test * fixing tests and handler * updating unit test for more coverage * adding some comments * self review * gofmt * updating and consolidating tests * moving functional options so it can be used properly * gofmt * more missed gofmt * Update beacon-chain/rpc/endpoints.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Update beacon-chain/rpc/lookup/blocker.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Update beacon-chain/rpc/eth/beacon/handlers.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * radek feedback * fixing tests * fixing test * moving endpoint in test * removed unneeded comment * fixing linting from latest develop merge * fixing linting from latest develop merge * Update beacon-chain/rpc/eth/blob/handlers.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Update beacon-chain/rpc/eth/blob/handlers.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * reverting change * reverting change * adding in better error for which hashes are missing * Update beacon-chain/rpc/lookup/blocker.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * fixing unit test * gofmt --------- Co-authored-by: Radosław Kapka <rkapka@wp.pl>
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package testutil
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"github.com/OffchainLabs/prysm/v6/beacon-chain/rpc/core"
|
|
"github.com/OffchainLabs/prysm/v6/beacon-chain/rpc/options"
|
|
"github.com/OffchainLabs/prysm/v6/consensus-types/blocks"
|
|
"github.com/OffchainLabs/prysm/v6/consensus-types/interfaces"
|
|
"github.com/OffchainLabs/prysm/v6/consensus-types/primitives"
|
|
"github.com/OffchainLabs/prysm/v6/encoding/bytesutil"
|
|
)
|
|
|
|
// MockBlocker is a fake implementation of lookup.Blocker.
|
|
type MockBlocker struct {
|
|
BlockToReturn interfaces.ReadOnlySignedBeaconBlock
|
|
ErrorToReturn error
|
|
SlotBlockMap map[primitives.Slot]interfaces.ReadOnlySignedBeaconBlock
|
|
RootBlockMap map[[32]byte]interfaces.ReadOnlySignedBeaconBlock
|
|
}
|
|
|
|
// Block --
|
|
func (m *MockBlocker) Block(_ context.Context, b []byte) (interfaces.ReadOnlySignedBeaconBlock, error) {
|
|
if m.ErrorToReturn != nil {
|
|
return nil, m.ErrorToReturn
|
|
}
|
|
if m.BlockToReturn != nil {
|
|
return m.BlockToReturn, nil
|
|
}
|
|
slotNumber, parseErr := strconv.ParseUint(string(b), 10, 64)
|
|
if parseErr != nil {
|
|
//nolint:nilerr
|
|
return m.RootBlockMap[bytesutil.ToBytes32(b)], nil
|
|
}
|
|
return m.SlotBlockMap[primitives.Slot(slotNumber)], nil
|
|
}
|
|
|
|
// Blobs --
|
|
func (*MockBlocker) Blobs(_ context.Context, _ string, _ ...options.BlobsOption) ([]*blocks.VerifiedROBlob, *core.RpcError) {
|
|
return nil, &core.RpcError{}
|
|
}
|