Switch to using the dep tool and update all the dependencies
This commit is contained in:
Nick Craig-Wood
+319
@@ -0,0 +1,319 @@
|
||||
// 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 pubsub
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"reflect"
|
||||
"sort"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
)
|
||||
|
||||
func TestKeepAliveExtendsDeadline(t *testing.T) {
|
||||
ticker := make(chan time.Time)
|
||||
deadline := time.Nanosecond * 15
|
||||
s := &testService{modDeadlineCalled: make(chan modDeadlineCall)}
|
||||
|
||||
checkModDeadlineCall := func(ackIDs []string) {
|
||||
got := <-s.modDeadlineCalled
|
||||
sort.Strings(got.ackIDs)
|
||||
|
||||
want := modDeadlineCall{
|
||||
subName: "subname",
|
||||
deadline: deadline,
|
||||
ackIDs: ackIDs,
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(got, want) {
|
||||
t.Errorf("keepalive: got:\n%v\nwant:\n%v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
ka := &keepAlive{
|
||||
s: s,
|
||||
Ctx: context.Background(),
|
||||
Sub: "subname",
|
||||
ExtensionTick: ticker,
|
||||
Deadline: deadline,
|
||||
MaxExtension: time.Hour,
|
||||
}
|
||||
ka.Start()
|
||||
|
||||
ka.Add("a")
|
||||
ka.Add("b")
|
||||
ticker <- time.Time{}
|
||||
checkModDeadlineCall([]string{"a", "b"})
|
||||
ka.Add("c")
|
||||
ka.Remove("b")
|
||||
ticker <- time.Time{}
|
||||
checkModDeadlineCall([]string{"a", "c"})
|
||||
ka.Remove("a")
|
||||
ka.Remove("c")
|
||||
ka.Add("d")
|
||||
ticker <- time.Time{}
|
||||
checkModDeadlineCall([]string{"d"})
|
||||
|
||||
ka.Remove("d")
|
||||
ka.Stop()
|
||||
}
|
||||
|
||||
func TestKeepAliveStopsWhenNoItem(t *testing.T) {
|
||||
ticker := make(chan time.Time)
|
||||
stopped := make(chan bool)
|
||||
s := &testService{modDeadlineCalled: make(chan modDeadlineCall, 3)}
|
||||
ka := &keepAlive{
|
||||
s: s,
|
||||
Ctx: context.Background(),
|
||||
ExtensionTick: ticker,
|
||||
}
|
||||
|
||||
ka.Start()
|
||||
|
||||
// There should be no call to modifyAckDeadline since there is no item.
|
||||
ticker <- time.Time{}
|
||||
|
||||
go func() {
|
||||
ka.Stop() // No items; should not block
|
||||
if len(s.modDeadlineCalled) > 0 {
|
||||
t.Errorf("unexpected extension to non-existent items: %v", <-s.modDeadlineCalled)
|
||||
}
|
||||
close(stopped)
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-stopped:
|
||||
case <-time.After(time.Second):
|
||||
t.Errorf("keepAlive timed out waiting for stop")
|
||||
}
|
||||
}
|
||||
|
||||
func TestKeepAliveStopsWhenItemsExpired(t *testing.T) {
|
||||
ticker := make(chan time.Time)
|
||||
stopped := make(chan bool)
|
||||
s := &testService{modDeadlineCalled: make(chan modDeadlineCall, 2)}
|
||||
ka := &keepAlive{
|
||||
s: s,
|
||||
Ctx: context.Background(),
|
||||
ExtensionTick: ticker,
|
||||
MaxExtension: time.Duration(0), // Should expire items at the first tick.
|
||||
}
|
||||
|
||||
ka.Start()
|
||||
ka.Add("a")
|
||||
ka.Add("b")
|
||||
|
||||
// Wait until the clock advances. Without this loop, this test fails on
|
||||
// Windows because the clock doesn't advance at all between ka.Add and the
|
||||
// expiration check after the tick is received.
|
||||
begin := time.Now()
|
||||
for time.Now().Equal(begin) {
|
||||
time.Sleep(time.Millisecond)
|
||||
}
|
||||
|
||||
// There should be no call to modifyAckDeadline since both items are expired.
|
||||
ticker <- time.Time{}
|
||||
|
||||
go func() {
|
||||
ka.Stop() // No live items; should not block.
|
||||
if len(s.modDeadlineCalled) > 0 {
|
||||
t.Errorf("unexpected extension to expired items")
|
||||
}
|
||||
close(stopped)
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-stopped:
|
||||
case <-time.After(time.Second):
|
||||
t.Errorf("timed out waiting for stop")
|
||||
}
|
||||
}
|
||||
|
||||
func TestKeepAliveBlocksUntilAllItemsRemoved(t *testing.T) {
|
||||
ticker := make(chan time.Time)
|
||||
eventc := make(chan string, 3)
|
||||
s := &testService{modDeadlineCalled: make(chan modDeadlineCall)}
|
||||
ka := &keepAlive{
|
||||
s: s,
|
||||
Ctx: context.Background(),
|
||||
ExtensionTick: ticker,
|
||||
MaxExtension: time.Hour, // Should not expire.
|
||||
}
|
||||
|
||||
ka.Start()
|
||||
ka.Add("a")
|
||||
ka.Add("b")
|
||||
|
||||
go func() {
|
||||
ticker <- time.Time{}
|
||||
|
||||
// We expect a call since both items should be extended.
|
||||
select {
|
||||
case args := <-s.modDeadlineCalled:
|
||||
sort.Strings(args.ackIDs)
|
||||
got := args.ackIDs
|
||||
want := []string{"a", "b"}
|
||||
if !reflect.DeepEqual(got, want) {
|
||||
t.Errorf("mismatching IDs:\ngot %v\nwant %v", got, want)
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
t.Errorf("timed out waiting for deadline extend call")
|
||||
}
|
||||
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
eventc <- "pre-remove-b"
|
||||
// Remove one item, Stop should still be waiting.
|
||||
ka.Remove("b")
|
||||
|
||||
ticker <- time.Time{}
|
||||
|
||||
// We expect a call since the item is still alive.
|
||||
select {
|
||||
case args := <-s.modDeadlineCalled:
|
||||
got := args.ackIDs
|
||||
want := []string{"a"}
|
||||
if !reflect.DeepEqual(got, want) {
|
||||
t.Errorf("mismatching IDs:\ngot %v\nwant %v", got, want)
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
t.Errorf("timed out waiting for deadline extend call")
|
||||
}
|
||||
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
|
||||
eventc <- "pre-remove-a"
|
||||
// Remove the last item so that Stop can proceed.
|
||||
ka.Remove("a")
|
||||
}()
|
||||
|
||||
go func() {
|
||||
ka.Stop() // Should block all item are removed.
|
||||
eventc <- "post-stop"
|
||||
}()
|
||||
|
||||
for i, want := range []string{"pre-remove-b", "pre-remove-a", "post-stop"} {
|
||||
select {
|
||||
case got := <-eventc:
|
||||
if got != want {
|
||||
t.Errorf("event #%d:\ngot %v\nwant %v", i, got, want)
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
t.Errorf("time out waiting for #%d event: want %v", i, want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// extendCallResult contains a list of ackIDs which are expected in an ackID
|
||||
// extension request, along with the result that should be returned.
|
||||
type extendCallResult struct {
|
||||
ackIDs []string
|
||||
err error
|
||||
}
|
||||
|
||||
// extendService implements modifyAckDeadline using a hard-coded list of extendCallResults.
|
||||
type extendService struct {
|
||||
service
|
||||
|
||||
calls []extendCallResult
|
||||
|
||||
t *testing.T // used for error logging.
|
||||
}
|
||||
|
||||
func (es *extendService) modifyAckDeadline(ctx context.Context, subName string, deadline time.Duration, ackIDs []string) error {
|
||||
if len(es.calls) == 0 {
|
||||
es.t.Fatalf("unexpected call to modifyAckDeadline: ackIDs: %v", ackIDs)
|
||||
}
|
||||
call := es.calls[0]
|
||||
es.calls = es.calls[1:]
|
||||
|
||||
if got, want := ackIDs, call.ackIDs; !reflect.DeepEqual(got, want) {
|
||||
es.t.Errorf("unexpected arguments to modifyAckDeadline: got: %v ; want: %v", got, want)
|
||||
}
|
||||
return call.err
|
||||
}
|
||||
|
||||
// Test implementation returns the first 2 elements as head, and the rest as tail.
|
||||
func (es *extendService) splitAckIDs(ids []string) ([]string, []string) {
|
||||
if len(ids) < 2 {
|
||||
return ids, nil
|
||||
}
|
||||
return ids[:2], ids[2:]
|
||||
}
|
||||
|
||||
func TestKeepAliveSplitsBatches(t *testing.T) {
|
||||
type testCase struct {
|
||||
calls []extendCallResult
|
||||
}
|
||||
for _, tc := range []testCase{
|
||||
{
|
||||
calls: []extendCallResult{
|
||||
{
|
||||
ackIDs: []string{"a", "b"},
|
||||
},
|
||||
{
|
||||
ackIDs: []string{"c", "d"},
|
||||
},
|
||||
{
|
||||
ackIDs: []string{"e", "f"},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
calls: []extendCallResult{
|
||||
{
|
||||
ackIDs: []string{"a", "b"},
|
||||
err: errors.New("bang"),
|
||||
},
|
||||
// On error we retry once.
|
||||
{
|
||||
ackIDs: []string{"a", "b"},
|
||||
err: errors.New("bang"),
|
||||
},
|
||||
// We give up after failing twice, so we move on to the next set, "c" and "d".
|
||||
{
|
||||
ackIDs: []string{"c", "d"},
|
||||
err: errors.New("bang"),
|
||||
},
|
||||
// Again, we retry once.
|
||||
{
|
||||
ackIDs: []string{"c", "d"},
|
||||
},
|
||||
{
|
||||
ackIDs: []string{"e", "f"},
|
||||
},
|
||||
},
|
||||
},
|
||||
} {
|
||||
s := &extendService{
|
||||
t: t,
|
||||
calls: tc.calls,
|
||||
}
|
||||
|
||||
ka := &keepAlive{
|
||||
s: s,
|
||||
Ctx: context.Background(),
|
||||
Sub: "subname",
|
||||
}
|
||||
|
||||
ka.extendDeadlines([]string{"a", "b", "c", "d", "e", "f"})
|
||||
|
||||
if len(s.calls) != 0 {
|
||||
t.Errorf("expected extend calls did not occur: %v", s.calls)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user