internal/pool_test.go

// Copyright 2016 Google LLC.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package internal

import (
	"testing"
	"time"

	"google.golang.org/grpc/naming"
)

func TestConnectionPool(t *testing.T) {
	addr := "127.0.0.1:123"
	ds := DialSettings{Endpoint: addr}
	pr := NewPoolResolver(4, &ds)
	watcher, err := pr.Resolve(addr)
	if err != nil {
		t.Fatalf("Resolve: %v", err)
	}

	updates, err := watcher.Next()
	if err != nil {
		t.Fatalf("Next: %v", err)
	}
	if len(updates) != 4 {
		t.Fatalf("Update count: %v", err)
	}
	metaSeen := make(map[interface{}]bool)
	for _, u := range updates {
		if u.Addr != addr {
			t.Errorf("Addr from update: wanted %v, got %v", addr, u.Addr)
		}
		// Metadata must be unique
		if metaSeen[u.Metadata] {
			t.Errorf("Wanted %v to be unique, got %v", u.Metadata, metaSeen)
		}
		metaSeen[u.Metadata] = true
	}

	// Test that Next blocks until Close and returns nil.
	nextc := make(chan []*naming.Update)
	go func() {
		next, err := watcher.Next()
		if err != nil {
			t.Errorf("Next: expected success, got %v", err)
		}
		nextc <- next
	}()
	time.Sleep(50 * time.Millisecond) // wait for watcher.Next goroutine
	select {
	case <-nextc:
		t.Fatal("next should not have been called yet")
	default:
	}
	watcher.Close()
	select {
	case next := <-nextc:
		if next != nil {
			t.Errorf("Next: expected nil, got %v", next)
		}
	case <-time.After(50 * time.Millisecond):
		t.Error("Next: did not return after 100ms")
	}
}