mirror of
https://github.com/btcsuite/btcd.git
synced 2024-11-19 18:00:11 +01:00
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.
65 lines
1.6 KiB
Go
65 lines
1.6 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 (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestMemPool(t *testing.T) {
|
|
pver := ProtocolVersion
|
|
|
|
// Ensure the command is expected value.
|
|
wantCmd := "mempool"
|
|
msg := NewMsgMemPool()
|
|
if cmd := msg.Command(); cmd != wantCmd {
|
|
t.Errorf("NewMsgMemPool: wrong command - got %v want %v",
|
|
cmd, wantCmd)
|
|
}
|
|
|
|
// Ensure max payload is expected value.
|
|
wantPayload := uint32(0)
|
|
maxPayload := msg.MaxPayloadLength(pver)
|
|
if maxPayload != wantPayload {
|
|
t.Errorf("MaxPayloadLength: wrong max payload length for "+
|
|
"protocol version %d - got %v, want %v", pver,
|
|
maxPayload, wantPayload)
|
|
}
|
|
|
|
// Test encode with latest protocol version.
|
|
var buf bytes.Buffer
|
|
err := msg.BtcEncode(&buf, pver)
|
|
if err != nil {
|
|
t.Errorf("encode of MsgMemPool failed %v err <%v>", msg, err)
|
|
}
|
|
|
|
// Older protocol versions should fail encode since message didn't
|
|
// exist yet.
|
|
oldPver := BIP0035Version - 1
|
|
err = msg.BtcEncode(&buf, oldPver)
|
|
if err == nil {
|
|
s := "encode of MsgMemPool passed for old protocol version %v err <%v>"
|
|
t.Errorf(s, msg, err)
|
|
}
|
|
|
|
// Test decode with latest protocol version.
|
|
readmsg := NewMsgMemPool()
|
|
err = readmsg.BtcDecode(&buf, pver)
|
|
if err != nil {
|
|
t.Errorf("decode of MsgMemPool failed [%v] err <%v>", buf, err)
|
|
}
|
|
|
|
// Older protocol versions should fail decode since message didn't
|
|
// exist yet.
|
|
err = readmsg.BtcDecode(&buf, oldPver)
|
|
if err == nil {
|
|
s := "decode of MsgMemPool passed for old protocol version %v err <%v>"
|
|
t.Errorf(s, msg, err)
|
|
}
|
|
|
|
return
|
|
}
|