mirror of
https://github.com/libp2p/go-libp2p-core.git
synced 2024-12-26 23:30:27 +08:00
7b2888dfdb
* add SignedEnvelope type * use struct for SignedEnvelope instead of exposing protobuf directly * doc comments for envelopes * tests for SignedEnvelopes * add helpers to make routing records for Host * fix doc comment * go fmt * add method to peerstore to retrieve signed routing records * update to match spec changes * just use nanoseconds * use proto3 & rename fields to match spec changes * use proto3 for routing records * make envelope fields private & validate on unmarshal * use buffer pool for envelope signatures * tests for RoutingState * go fmt * rename Equals -> Equal, add some comments * use test helpers * get rid of unsigned RoutingState struct, only expose SignedRoutingState * rm batching SignedRoutingStates accessor in peerstore the datastore peerstore implementation doesn't support batched reads, so it's no more efficient to get a bunch of states at once than it is to call SignedRoutingState multiple times. * whitespace * expose struct fields & remove accessors * use camelCase in protos for consistency * use multiformats uvarint for length-prefixes * remove payloadType check when unmarhaling * rm stray ref to golang/protobuf * define CertifiedAddrBook to avoid breaking API change * add events for updated addresses and routing state * remove SignedRoutingStateFromHost helper moving this to go-libp2p * add routing state records, extend peerstore API * fix: rebuild protos with new gogofaster generator * filter private addrs from signed routing records * envelope: use byte slices from pool; adjust interface. * move envelope to record package. * move protobuf files; adjust imports everywhere. * rename RoutingStateRecord -> PeerRecord also removes embedded reference to Envelope from the record, as that was confusing. as a result, the CertifiedAddrBook now accepts/returns record.SignedEnvelope instead of a specialized type. * hoist Seq from PeerRecord to SignedEnvelope * test that PeerRecords can't be signed by wrong key * commit go.sum * add Seq field to envelope signature * fix proto_path in Makefile * fix import ordering * comments for PeerRecord proto message also removes the seq field from PeerMessage proto, since it was moved to the SignedEnvelope * use Record type for envelope payloads * rename SignedEnvelope -> Envelope, unmarshal payload in ConsumeEnvelope * return buffer to pool before early return * doc comments * rename CertifiedAddrBook methods, update comments * cache unmarshalled Record payload inside Envelope * doc comments * store reflect.Type when registering Record * Revert "return buffer to pool before early return"8d8da386f2misread this - unsigned will be nil if there's an error, so it was right the way it was * use a DefaultRecord for unregistered PayloadTypes instead of returning an error if we don't have a registered Record for a given PayloadType, we can have a catch-all DefaultRecord type that just preserves the original payload as a []byte * cleanup DefaultRecord code a bit - removes unused error return from blankRecordForPayloadType - just references instead of copying in DefaultRecord.UnmarshalRecord I figure this is likely safe, since we'll be unmarshalling from the payload of an Envelope, which shouldn't get altered after it's created. * use explicit payloadType in MakeEnvelopeWithRecord * Revert DefaultRecord commitsae3bc7bdfba26c845a76* doc comments * move Seq field back to PeerRecord * make diffs optional in EvtLocalAddressesUpdated * more envelope tests * replace MakeEnvelope with record.Seal also: - add Domain and Codec fields to Record interface * fix import * add interface check * rename ProcessPeerRecord -> ConsumePeerRecord also, adds bool `accepted` return value * rename event field, add doc comment * peer record protobuf: fix field casing. * record protobuf: add docs and fix casing. * cleanup: group imports. * nit: split test/utils.go => test/{addrs,errors}.go. Co-authored-by: Raúl Kripalani <raul.kripalani@gmail.com>
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package record
|
|
|
|
import "testing"
|
|
|
|
var testPayloadType = []byte("/libp2p/test/record/payload-type")
|
|
|
|
type testPayload struct {
|
|
unmarshalPayloadCalled bool
|
|
}
|
|
|
|
func (p *testPayload) Domain() string {
|
|
return "testing"
|
|
}
|
|
|
|
func (p *testPayload) Codec() []byte {
|
|
return testPayloadType
|
|
}
|
|
|
|
func (p *testPayload) MarshalRecord() ([]byte, error) {
|
|
return []byte("hello"), nil
|
|
}
|
|
|
|
func (p *testPayload) UnmarshalRecord(bytes []byte) error {
|
|
p.unmarshalPayloadCalled = true
|
|
return nil
|
|
}
|
|
|
|
func TestUnmarshalPayload(t *testing.T) {
|
|
t.Run("fails if payload type is unregistered", func(t *testing.T) {
|
|
_, err := unmarshalRecordPayload([]byte("unknown type"), []byte{})
|
|
if err != ErrPayloadTypeNotRegistered {
|
|
t.Error("Expected error when unmarshalling payload with unregistered payload type")
|
|
}
|
|
})
|
|
|
|
t.Run("calls UnmarshalRecord on concrete Record type", func(t *testing.T) {
|
|
RegisterType(&testPayload{})
|
|
|
|
payload, err := unmarshalRecordPayload(testPayloadType, []byte{})
|
|
if err != nil {
|
|
t.Errorf("unexpected error unmarshalling registered payload type: %v", err)
|
|
}
|
|
typedPayload, ok := payload.(*testPayload)
|
|
if !ok {
|
|
t.Error("expected unmarshalled payload to be of the correct type")
|
|
}
|
|
if !typedPayload.unmarshalPayloadCalled {
|
|
t.Error("expected UnmarshalRecord to be called on concrete Record instance")
|
|
}
|
|
})
|
|
}
|