60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
// Copyright (c) 2013-2014 Conformal Systems LLC.
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package btcwire_test
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/btcsuite/btcwire"
|
|
)
|
|
|
|
// fakeMessage implements the btcwire.Message interface and is used to force
|
|
// encode errors in messages.
|
|
type fakeMessage struct {
|
|
command string
|
|
payload []byte
|
|
forceEncodeErr bool
|
|
forceLenErr bool
|
|
}
|
|
|
|
// BtcDecode doesn't do anything. It just satisfies the btcwire.Message
|
|
// interface.
|
|
func (msg *fakeMessage) BtcDecode(r io.Reader, pver uint32) error {
|
|
return nil
|
|
}
|
|
|
|
// BtcEncode writes the payload field of the fake message or forces an error
|
|
// if the forceEncodeErr flag of the fake message is set. It also satisfies the
|
|
// btcwire.Message interface.
|
|
func (msg *fakeMessage) BtcEncode(w io.Writer, pver uint32) error {
|
|
if msg.forceEncodeErr {
|
|
err := &btcwire.MessageError{
|
|
Func: "fakeMessage.BtcEncode",
|
|
Description: "intentional error",
|
|
}
|
|
return err
|
|
}
|
|
|
|
_, err := w.Write(msg.payload)
|
|
return err
|
|
}
|
|
|
|
// Command returns the command field of the fake message and satisfies the
|
|
// btcwire.Message interface.
|
|
func (msg *fakeMessage) Command() string {
|
|
return msg.command
|
|
}
|
|
|
|
// MaxPayloadLength returns the length of the payload field of fake message
|
|
// or a smaller value if the forceLenErr flag of the fake message is set. It
|
|
// satisfies the btcwire.Message interface.
|
|
func (msg *fakeMessage) MaxPayloadLength(pver uint32) uint32 {
|
|
lenp := uint32(len(msg.payload))
|
|
if msg.forceLenErr {
|
|
return lenp - 1
|
|
}
|
|
|
|
return lenp
|
|
}
|