Function TestVersionHandle

p2pserver/message/utils/msg_handler_test.go:155–195

TestVersionHandle tests Function VersionHandle handling a version message

(t *testing.T)

// TestVersionHandle tests Function VersionHandle handling a version message
func TestVersionHandle(t *testing.T) {
	// Simulate a remote peer to connect to the local
	remotePeer := peer.NewPeer()
	assert.NotNil(t, remotePeer)

	network.AddPeerAddress("127.0.0.1:50010", remotePeer)

	var testID uint64
	_, testPub, _ := keypair.GenerateKeyPair(keypair.PK_ECDSA, keypair.P256)
	key := keypair.SerializePublicKey(testPub)
	err := binary.Read(bytes.NewBuffer(key[:8]), binary.LittleEndian, &(testID))
	assert.Nil(t, err)

	// Construct a version packet
	buf := msgpack.NewVersion(network, 12345)
	version := buf.(*types.Version)
	version.P.Nonce = testID

	msg := &types.MsgPayload{
		Id:      testID,
		Addr:    "127.0.0.1:50010",
		Payload: buf,
	}

	// Invoke VersionHandle to handle the msg
	VersionHandle(msg, network, nil)

	// Get the remote peer from the neighbor peers by peer id
	tempPeer := network.GetPeer(testID)
	assert.NotNil(t, tempPeer)

	assert.Equal(t, tempPeer.GetID(), testID)
	assert.Equal(t, tempPeer.GetVersion(), network.GetVersion())
	assert.Equal(t, tempPeer.GetServices(), network.GetServices())
	assert.Equal(t, tempPeer.GetPort(), network.GetPort())
	assert.Equal(t, tempPeer.GetHttpInfoPort(), network.GetHttpInfoPort())
	assert.Equal(t, tempPeer.GetHeight(), uint64(12345))
	assert.Equal(t, tempPeer.GetState(), uint32(msgCommon.HAND_SHAKE))

	network.DelNbrNode(testID)
}

// TestVerAckHandle tests Function VerAckHandle handling a version ack
func TestVerAckHandle(t *testing.T) {