internal/pool_test.go

// Copyright 2016 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

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")
	}
}