Function TestMultiple

kite_test.go:103–175

(t *testing.T)

}

func TestMultiple(t *testing.T) {
	testDuration := time.Second * 10

	// number of available mathworker kites to be called
	kiteNumber := 3

	// number of exp kites that will call mathwork kites
	clientNumber := 3

	// ports are starting from 6000 up to 6000 + kiteNumber
	port := 6000

	transport := transportFromEnv()

	for i := 0; i < kiteNumber; i++ {
		m := New("mathworker"+strconv.Itoa(i), "0.1."+strconv.Itoa(i))
		m.Config.DisableAuthentication = true
		m.Config.Transport = transport
		m.Config.Port = port + i

		m.OnConnect(panicHandler)
		m.OnRegister(panicRegisterHandler)
		m.OnDisconnect(panicHandler)
		m.OnFirstRequest(panicHandler)

		m.HandleFunc("square", Square)
		go m.Run()
		<-m.ServerReadyNotify()
		defer m.Close()
	}

	clients := make([]*Client, clientNumber)
	for i := 0; i < clientNumber; i++ {
		cn := New("exp"+strconv.Itoa(i), "0.0.1")
		cn.Config.Transport = transport

		c := cn.NewClient("http://127.0.0.1:" + strconv.Itoa(port+i) + "/kite")
		if err := c.Dial(); err != nil {
			t.Fatal(err)
		}

		clients[i] = c
		defer c.Close()
	}

	timeout := time.After(testDuration)

	// every one second
	for {
		select {
		case <-time.Tick(time.Second):
			var wg sync.WaitGroup

			for i := 0; i < clientNumber; i++ {
				wg.Add(1)

				go func(i int, t *testing.T) {
					defer wg.Done()