package endpoint_test

import (
	"errors"
	"testing"
	"time"

	. "github.com/weaveworks/scope/probe/endpoint"
	"github.com/weaveworks/scope/test"
)

func TestReverseResolver(t *testing.T) {
	tests := map[string]string{
		"1.2.3.4": "test.domain.name",
		"4.3.2.1": "im.a.little.tea.pot",
	}

	revRes := NewReverseResolver()
	defer revRes.Stop()

	// Use a mocked resolver function.
	revRes.Resolver = func(addr string) (names []string, err error) {
		if name, ok := tests[addr]; ok {
			return []string{name}, nil
		}
		return []string{}, errors.New("invalid IP")
	}

	// Up the rate limit so the test runs faster.
	revRes.Throttle = time.Tick(time.Millisecond)

	for ip, hostname := range tests {
		test.Poll(t, 100*time.Millisecond, hostname, func() interface{} {
			result, _ := revRes.Get(ip)
			return result
		})
	}
}