fluencelabs/go-libp2p-kad-dht
1
0
Fork 0
mirror of https://github.com/fluencelabs/go-libp2p-kad-dht synced 2025-07-01 13:41:38 +00:00
Code Issues Projects Releases Wiki Activity

park waiters in slice; revise closure logic.

Browse Source
This commit is contained in:
Raúl Kripalani
2019-01-30 23:02:35 +00:00
parent bf4b91ce4b
commit 74d22f3f5e
3 changed files with 49 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
dial_queue_test.go
View File

@ -16,31 +16,6 @@ func init() {
DialQueueScalingMutePeriod = 0
}
func TestDialQueueErrorsWithTooManyConsumers(t *testing.T) {
var calls int
defer func() {
if e := recover(); e == nil {
t.Error("expected a panic, got none")
} else if calls != 4 {
t.Errorf("expected a panic on the 4th call to Consume(); got it on call number %d", calls)
}
}()
in := queue.NewChanQueue(context.Background(), queue.NewXORDistancePQ("test"))
hang := make(chan struct{})
dialFn := func(ctx context.Context, p peer.ID) error {
<-hang
return nil
}
dq := newDialQueue(context.Background(), "test", in, dialFn, 3)
for ; calls < 3; calls++ {
dq.Consume()
}
calls++
dq.Consume()
}
func TestDialQueueGrowsOnSlowDials(t *testing.T) {
DialQueueMaxIdle = 10 * time.Minute
@ -60,7 +35,7 @@ func TestDialQueueGrowsOnSlowDials(t *testing.T) {
}
// remove the mute period to grow faster.
dq := newDialQueue(context.Background(), "test", in, dialFn, 4)
dq := newDialQueue(context.Background(), "test", in, dialFn)
for i := 0; i < 4; i++ {
_ = dq.Consume()
@ -93,7 +68,7 @@ func TestDialQueueShrinksWithNoConsumers(t *testing.T) {
return nil
}
dq := newDialQueue(context.Background(), "test", in, dialFn, 3)
dq := newDialQueue(context.Background(), "test", in, dialFn)
defer func() {
recover()
@ -160,7 +135,7 @@ func TestDialQueueShrinksWithWhenIdle(t *testing.T) {
in.EnqChan <- peer.ID(i)
}
dq := newDialQueue(context.Background(), "test", in, dialFn, 3)
dq := newDialQueue(context.Background(), "test", in, dialFn)
// keep up to speed with backlog by releasing the dial function every time we acquire a channel.
for i := 0; i < 13; i++ {
@ -203,7 +178,7 @@ func TestDialQueueMutePeriodHonored(t *testing.T) {
in.EnqChan <- peer.ID(i)
}
dq := newDialQueue(context.Background(), "test", in, dialFn, 3)
dq := newDialQueue(context.Background(), "test", in, dialFn)
// pick up three consumers.
for i := 0; i < 3; i++ {