Function TestServer

p2p/server_test.go:25–83

(t *testing.T)

}

func TestServer(t *testing.T) {
	listener := []byte("9")
	os.Setenv("PUBLICIP", "127.0.0.1")
	log.Init(listener[:])

	pListener, _ := CreateP2PNetwork(listener, "9905", NoDiscover)
	pListener.Listen()

	go receiveEvent(pListener, t)

	var wgForPeer sync.WaitGroup
	wgForPeer.Add(1)
	for c := 9904; c < 9905; c++ {
		go func(c int) {
			defer wgForPeer.Done()
			id := []byte(strconv.Itoa(c))
			p, _ := CreateP2PNetwork(id, strconv.Itoa(c), NoDiscover)
			p.Listen()

			var count uint64
			var wgForMsg sync.WaitGroup
			var connected []byte
			var err error
			wgForMsg.Add(10)
			for count = 0; count < 10; count++ {
				go func(count uint64) {
					defer wgForMsg.Done()
					p.SetPort("9905")
					connected, err = p.ConnectTo("127.0.0.1", nil)
					if err != nil {
						t.Errorf("ConnectTo ,Error %s", err)
					}
					cmd := &Ping{Count: count}
					pb := proto.Message(cmd)
					reply, _ := p.Request(connected, pb)
					pong, _ := reply.Msg.Message.(*Pong)
					if pong.Count-count != 10 {
						t.Errorf("TestRequest ,Expected %d Actual %d", count+10, pong.Count)
					}
					p.DisConnectTo(connected)
				}(count)
			}
			wgForMsg.Wait()
		}(c)
	}
	wgForPeer.Wait()
	retryLimit := 5
	for {
		prNum, pcNum := pListener.numOfClient()
		if prNum == 0 && pcNum == 0 {
			break
		}
		retryLimit--
		if retryLimit == 0 {
			t.Errorf("TestServer ,Expected %d Actual %d", 0, prNum)
		}
		time.Sleep(1 * time.Second)
	}