Function TestDHTService_FindNeighbor_FindNode

route/service_test.go:41–203

(t *testing.T)

const DHTStorePath = "./DHTStore.keystore"

func TestDHTService_FindNeighbor_FindNode(t *testing.T) {
	utils.RemoveAll(DHTStorePath + "*")
	defer utils.RemoveAll(DHTStorePath + "*")
	log.SetLevel(log.DebugLevel)
	addr := "127.0.0.1:0"
	dht, _ := NewDHTService(DHTStorePath+"1", new(consistent.KMSStorage), false)
	kms.ResetBucket()
	rpc.RegisterName(DHTRPCName, dht)
	ln, err := net.Listen("tcp", addr)
	if err != nil {
		fmt.Println(err)
		return
	}
	go func() {
		for {
			c, err := ln.Accept()
			if err != nil {
				continue
			}
			go rpc.ServeConn(c)
		}
	}()

	client, err := rpc.Dial("tcp", ln.Addr().String())
	if err != nil {
		log.Error(err)
	}

	req := &FindNeighborReq{
		ID:    "123",
		Count: 2,
	}
	resp := new(FindNeighborResp)
	err = client.Call("DHT.FindNeighbor", req, resp)
	if err != nil {
		log.Error(err)
	}
	log.Debugf("resp: %v", resp)

	Convey("test FindNeighbor empty", t, func() {
		So(resp.Nodes, ShouldBeEmpty)
		So(err.Error(), ShouldContainSubstring, consistent.ErrEmptyCircle.Error())
	})

	reqFN1 := &FindNodeReq{
		ID: "123",
	}
	respFN1 := new(FindNodeResp)
	err = client.Call("DHT.FindNode", reqFN1, respFN1)
	if err != nil {
		log.Error(err)
	}
	log.Debugf("respFN1: %v", respFN1)
	Convey("test FindNode", t, func() {
		So(respFN1.Node, ShouldBeNil)
		So(err.Error(), ShouldContainSubstring, consistent.ErrKeyNotFound.Error())
	})