package registry

import (
	"fmt"
	"testing"
	"time"
)

func TestZooKeeperRegistry_Register_Discovery(t *testing.T) {

	z := ZooKeeperRegistry{
		Host: []string{""},
	}
	chanNodes, chanNodesErr := z.Discovery("/test")
	go func() {
		for i := 0; i < 3; i++ {

			if err := z.Register("/test", fmt.Sprintf("192.168.1.1%d", i), []byte{}); err != nil {
				t.Fatal("failed register host at path")
			} else {
				t.Logf("register host successfully")
			}
			time.Sleep(time.Millisecond * 300)

		}

	}()
	for {
		select {
		case err := <-chanNodesErr:
			t.Fatalf("chanNodes occur error, info: %s", err.Error())
		case changeNodes := <-chanNodes:
			t.Logf("node list changed, new list: %v", changeNodes)
		default:
			time.Sleep(time.Millisecond * 100)
		}
	}
}