Vendor dependencies for GCS

vendor/google.golang.org/api/internal/pool_test.go

@@ -0,0 +1,82 @@
+/*
+Copyright 2016 Google Inc. All Rights Reserved.
+
+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
+
+    http://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 (
+	"google.golang.org/grpc/naming"
+	"testing"
+	"time"
+)
+
+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)
+	closedc := make(chan bool)
+	go func() {
+		next, err := watcher.Next()
+		if err != nil {
+			t.Errorf("Next: expected success, got %v", err)
+		}
+		nextc <- next
+	}()
+	go func() {
+		time.Sleep(50 * time.Millisecond)
+		watcher.Close()
+		close(closedc)
+	}()
+
+	select {
+	case <-nextc:
+		t.Fatalf("Next: second invocation didn't block, returned before Close()")
+	case <-closedc:
+		// OK, watcher was closed before Next() returned.
+	}
+	select {
+	case next := <-nextc:
+		if next != nil {
+			t.Errorf("Next: expected nil, got %v", next)
+		}
+	case <-time.After(100 * time.Millisecond):
+		t.Error("Next: did not return after 100ms")
+	}
+}