package agent

import (
	"io"
	"math/rand"
	"net"
	"testing"
	"time"

	"basic.com/valib/serf.git/serf"
	"basic.com/valib/serf.git/testutil"
)

func init() {
	// Seed the random number generator
	rand.Seed(time.Now().UnixNano())
}

func drainEventCh(ch <-chan string) {
	for {
		select {
		case <-ch:
		default:
			return
		}
	}
}

func testAgent(t *testing.T, ip net.IP, logOutput io.Writer) *Agent {
	return testAgentWithConfig(t, ip, DefaultConfig(), serf.DefaultConfig(), logOutput)
}

func testAgentWithConfig(t *testing.T, ip net.IP, agentConfig *Config, serfConfig *serf.Config, logOutput io.Writer) *Agent {
	serfConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond
	serfConfig.MemberlistConfig.BindAddr = ip.String()
	serfConfig.NodeName = serfConfig.MemberlistConfig.BindAddr

	// Activate the strictest version of memberlist validation to ensure
	// we properly pass node names through the serf layer.
	serfConfig.MemberlistConfig.RequireNodeNames = true

	if logOutput == nil {
		logOutput = testutil.TestWriter(t)
	}

	agent, err := Create(agentConfig, serfConfig, logOutput)
	if err != nil {
		t.Fatalf("err: %v", err)
	}
	return agent
}