2eef3720a9
This commit contains the entire btcwire repository along with several changes needed to move all of the files into the wire directory in order to prepare it for merging. This does NOT update btcd or any of the other packages to use the new location as that will be done separately. - All import paths in the old btcwire test files have been changed to the new location - All references to btcwire as the package name have been chagned to wire - The coveralls badge has been removed since it unfortunately doesn't support coverage of sub-packages This is ongoing work toward #214.
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
// Copyright (c) 2013-2015 Conformal Systems LLC.
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package wire_test
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/btcsuite/btcd/wire"
|
|
)
|
|
|
|
// fakeMessage implements the wire.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 wire.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
|
|
// wire.Message interface.
|
|
func (msg *fakeMessage) BtcEncode(w io.Writer, pver uint32) error {
|
|
if msg.forceEncodeErr {
|
|
err := &wire.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
|
|
// wire.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 wire.Message interface.
|
|
func (msg *fakeMessage) MaxPayloadLength(pver uint32) uint32 {
|
|
lenp := uint32(len(msg.payload))
|
|
if msg.forceLenErr {
|
|
return lenp - 1
|
|
}
|
|
|
|
return lenp
|
|
}
|