forked from TrueCloudLab/neoneo-go
6e3f1ec43e
* refactored tcp transport * return errors on outgoing messages * TCP transport should report its error after reading from connection * handle error returned from peer transport * bump version * cleaned up error
76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package network
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/network/payload"
|
|
)
|
|
|
|
func TestHandleVersionFailWrongPort(t *testing.T) {
|
|
s := NewServer(ModeDevNet)
|
|
go s.loop()
|
|
|
|
p := NewLocalPeer(s)
|
|
|
|
version := payload.NewVersion(1337, 1, "/NEO:0.0.0/", 0, true)
|
|
msg := newMessage(ModeDevNet, cmdVersion, version)
|
|
if err := s.handleVersionCmd(msg, p); err == nil {
|
|
t.Fatal("expected error got nil")
|
|
}
|
|
}
|
|
|
|
func TestHandleVersionFailIdenticalNonce(t *testing.T) {
|
|
s := NewServer(ModeDevNet)
|
|
go s.loop()
|
|
|
|
p := NewLocalPeer(s)
|
|
|
|
version := payload.NewVersion(s.id, 1, "/NEO:0.0.0/", 0, true)
|
|
msg := newMessage(ModeDevNet, cmdVersion, version)
|
|
if err := s.handleVersionCmd(msg, p); err == nil {
|
|
t.Fatal("expected error got nil")
|
|
}
|
|
}
|
|
|
|
func TestHandleVersion(t *testing.T) {
|
|
s := NewServer(ModeDevNet)
|
|
go s.loop()
|
|
|
|
p := NewLocalPeer(s)
|
|
|
|
version := payload.NewVersion(1337, p.addr().Port, "/NEO:0.0.0/", 0, true)
|
|
msg := newMessage(ModeDevNet, cmdVersion, version)
|
|
|
|
if err := s.handleVersionCmd(msg, p); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestPeerCount(t *testing.T) {
|
|
s := NewServer(ModeDevNet)
|
|
go s.loop()
|
|
|
|
lenPeers := 10
|
|
for i := 0; i < lenPeers; i++ {
|
|
s.register <- NewLocalPeer(s)
|
|
}
|
|
|
|
if have, want := s.peerCount(), lenPeers; want != have {
|
|
t.Fatalf("expected %d connected peers got %d", want, have)
|
|
}
|
|
}
|
|
|
|
func TestHandleAddrCmd(t *testing.T) {
|
|
// todo
|
|
}
|
|
|
|
func TestHandleGetAddrCmd(t *testing.T) {
|
|
// todo
|
|
}
|
|
|
|
func TestHandleInv(t *testing.T) {
|
|
// todo
|
|
}
|
|
func TestHandleBlockCmd(t *testing.T) {
|
|
// todo
|
|
}
|