Update dependencies

vendor/github.com/docker/spdystream/priority_test.go

@@ -0,0 +1,108 @@
+package spdystream
+
+import (
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/docker/spdystream/spdy"
+)
+
+func TestPriorityQueueOrdering(t *testing.T) {
+	queue := NewPriorityFrameQueue(150)
+	data1 := &spdy.DataFrame{}
+	data2 := &spdy.DataFrame{}
+	data3 := &spdy.DataFrame{}
+	data4 := &spdy.DataFrame{}
+	queue.Push(data1, 2)
+	queue.Push(data2, 1)
+	queue.Push(data3, 1)
+	queue.Push(data4, 0)
+
+	if queue.Pop() != data4 {
+		t.Fatalf("Wrong order, expected data4 first")
+	}
+	if queue.Pop() != data2 {
+		t.Fatalf("Wrong order, expected data2 second")
+	}
+	if queue.Pop() != data3 {
+		t.Fatalf("Wrong order, expected data3 third")
+	}
+	if queue.Pop() != data1 {
+		t.Fatalf("Wrong order, expected data1 fourth")
+	}
+
+	// Insert 50 Medium priority frames
+	for i := spdy.StreamId(50); i < 100; i++ {
+		queue.Push(&spdy.DataFrame{StreamId: i}, 1)
+	}
+	// Insert 50 low priority frames
+	for i := spdy.StreamId(100); i < 150; i++ {
+		queue.Push(&spdy.DataFrame{StreamId: i}, 2)
+	}
+	// Insert 50 high priority frames
+	for i := spdy.StreamId(0); i < 50; i++ {
+		queue.Push(&spdy.DataFrame{StreamId: i}, 0)
+	}
+
+	for i := spdy.StreamId(0); i < 150; i++ {
+		frame := queue.Pop()
+		if frame.(*spdy.DataFrame).StreamId != i {
+			t.Fatalf("Wrong frame\nActual: %d\nExpecting: %d", frame.(*spdy.DataFrame).StreamId, i)
+		}
+	}
+}
+
+func TestPriorityQueueSync(t *testing.T) {
+	queue := NewPriorityFrameQueue(150)
+	var wg sync.WaitGroup
+	insertRange := func(start, stop spdy.StreamId, priority uint8) {
+		for i := start; i < stop; i++ {
+			queue.Push(&spdy.DataFrame{StreamId: i}, priority)
+		}
+		wg.Done()
+	}
+	wg.Add(3)
+	go insertRange(spdy.StreamId(100), spdy.StreamId(150), 2)
+	go insertRange(spdy.StreamId(0), spdy.StreamId(50), 0)
+	go insertRange(spdy.StreamId(50), spdy.StreamId(100), 1)
+
+	wg.Wait()
+	for i := spdy.StreamId(0); i < 150; i++ {
+		frame := queue.Pop()
+		if frame.(*spdy.DataFrame).StreamId != i {
+			t.Fatalf("Wrong frame\nActual: %d\nExpecting: %d", frame.(*spdy.DataFrame).StreamId, i)
+		}
+	}
+}
+
+func TestPriorityQueueBlocking(t *testing.T) {
+	queue := NewPriorityFrameQueue(15)
+	for i := 0; i < 15; i++ {
+		queue.Push(&spdy.DataFrame{}, 2)
+	}
+	doneChan := make(chan bool)
+	go func() {
+		queue.Push(&spdy.DataFrame{}, 2)
+		close(doneChan)
+	}()
+	select {
+	case <-doneChan:
+		t.Fatalf("Push succeeded, expected to block")
+	case <-time.After(time.Millisecond):
+		break
+	}
+
+	queue.Pop()
+
+	select {
+	case <-doneChan:
+		break
+	case <-time.After(time.Millisecond):
+		t.Fatalf("Push should have succeeded, but timeout reached")
+	}
+
+	for i := 0; i < 15; i++ {
+		queue.Pop()
+	}
+}