d406d9e52b
Putting the test code in the same package makes it easier for forks since they don't have to change the import paths as much and it also gets rid of the need for internal_test.go to bridge. This same thing should probably be done for the majority of the code base.
56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
// Copyright (c) 2013-2016 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package wire
|
|
|
|
import "io"
|
|
|
|
// fakeMessage implements the 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 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
|
|
// Message interface.
|
|
func (msg *fakeMessage) BtcEncode(w io.Writer, pver uint32) error {
|
|
if msg.forceEncodeErr {
|
|
err := &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
|
|
// 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 Message interface.
|
|
func (msg *fakeMessage) MaxPayloadLength(pver uint32) uint32 {
|
|
lenp := uint32(len(msg.payload))
|
|
if msg.forceLenErr {
|
|
return lenp - 1
|
|
}
|
|
|
|
return lenp
|
|
}
|