Function TestEncPingFindNeighbor

rpc/mux/server_test.go:272–352

(t *testing.T)

}

func TestEncPingFindNeighbor(t *testing.T) {
	utils.RemoveAll(PubKeyStorePath + "*")
	defer utils.RemoveAll(PubKeyStorePath + "*")
	log.SetLevel(log.FatalLevel)
	addr := "127.0.0.1:0"
	masterKey := []byte("abc")
	dht, err := route.NewDHTService(PubKeyStorePath, new(consistent.KMSStorage), true)

	server, err := NewServerWithService(ServiceMap{route.DHTRPCName: dht})
	if err != nil {
		log.Fatal(err)
	}

	_ = server.InitRPCServer(addr, "../keys/test.key", masterKey)
	go server.Serve()

	publicKey, err := kms.GetLocalPublicKey()
	nonce := asymmetric.GetPubKeyNonce(publicKey, 10, 100*time.Millisecond, nil)
	serverNodeID := proto.NodeID(nonce.Hash.String())
	_ = kms.SetPublicKey(serverNodeID, nonce.Nonce, publicKey)

	kms.SetLocalNodeIDNonce(nonce.Hash.CloneBytes(), &nonce.Nonce)
	_ = route.SetNodeAddrCache(&proto.RawNodeID{Hash: nonce.Hash}, server.Listener.Addr().String())

	client, err := rpc.DialToNodeWithPool(&nilSessionPool{}, serverNodeID, false)
	if err != nil {
		log.Fatal(err)
	}

	node1 := proto.NewNode()
	_ = node1.InitNodeCryptoInfo(100 * time.Millisecond)

	reqA := &proto.PingReq{
		Node: *node1,
	}

	respA := new(proto.PingResp)
	err = client.Call("DHT.Ping", reqA, respA)
	if err != nil {
		log.Fatal(err)
	}
	log.Debugf("respA: %v", respA)

	node2 := proto.NewNode()
	_ = node2.InitNodeCryptoInfo(100 * time.Millisecond)

	reqB := &proto.PingReq{
		Node: *node2,
	}

	respB := new(proto.PingResp)
	err = client.Call("DHT.Ping", reqB, respB)
	if err != nil {
		log.Fatal(err)
	}
	log.Debugf("respB: %v", respB)

	req := &proto.FindNeighborReq{