Function TestEncryptIncCounterSimpleArgs

rpc/mux/server_test.go:188–220

(t *testing.T)

}

func TestEncryptIncCounterSimpleArgs(t *testing.T) {
	defer utils.RemoveAll(PubKeyStorePath + "*")
	log.SetLevel(log.FatalLevel)
	addr := "127.0.0.1:0"
	masterKey := []byte("abc")
	server, err := NewServerWithService(ServiceMap{"Test": NewTestService()})
	if err != nil {
		log.Fatal(err)
	}

	_, _ = route.NewDHTService(PubKeyStorePath, new(consistent.KMSStorage), true)
	_ = 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)

	repSimple := new(int32)
	err = client.Call("Test.IncCounterSimpleArgs", 10, repSimple)
	if err != nil {
		log.Fatal(err)
	}
	CheckNum(*repSimple, 10, t)

	_ = client.Close()
	server.Stop()
}

func TestETLSBug(t *testing.T) {
	defer utils.RemoveAll(PubKeyStorePath + "*")