Function TestNewPersistentCaller

rpc/mux/wrapper_test.go:167–313

(t *testing.T)

}

func TestNewPersistentCaller(t *testing.T) {
	log.SetLevel(log.FatalLevel)
	utils.RemoveAll(PubKeyStorePath + "*")
	defer utils.RemoveAll(PubKeyStorePath + "*")
	utils.RemoveAll(publicKeyStore + "*")
	defer utils.RemoveAll(publicKeyStore + "*")

	var d string
	var err error
	if d, err = ioutil.TempDir("", "rpcutil_test_"); err != nil {
		return
	}

	// init conf
	_, testFile, _, _ := runtime.Caller(0)
	dupConfFile := filepath.Join(d, "config.yaml")
	confFile := filepath.Join(filepath.Dir(testFile), "../../test/node_standalone/config.yaml")
	if err = utils.DupConf(confFile, dupConfFile); err != nil {
		return
	}
	privateKeyPath := filepath.Join(filepath.Dir(testFile), "../../test/node_standalone/private.key")

	conf.GConf, _ = conf.LoadConfig(dupConfFile)
	log.Debugf("GConf: %#v", conf.GConf)
	// reset the once
	route.Once.Reset()
	_ = GetSessionPoolInstance().Close()
	route.InitKMS(publicKeyStore)

	addr := conf.GConf.ListenAddr // see ../test/node_standalone/config.yaml
	masterKey := []byte("")
	dht, err := route.NewDHTService(PubKeyStorePath, new(consistent.KMSStorage), true)

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

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
	defer cancel()
	err = utils.WaitForPorts(ctx, "127.0.0.1", []int{
		12230,
	}, time.Millisecond*200)

	if err != nil {
		log.Fatalf("wait for port ready timeout: %v", err)
	}

	err = server.InitRPCServer(addr, privateKeyPath, masterKey)
	if err != nil {
		t.Fatal(err)
	}
	go server.Serve()

	client := NewPersistentCaller(conf.GConf.BP.NodeID)
	node1 := proto.NewNode()
	_ = node1.InitNodeCryptoInfo(100 * time.Millisecond)
	node1.Addr = "1.1.1.1:1"