mirror of
https://github.com/cloudflare/cloudflared.git
synced 2025-07-27 14:39:58 +00:00
TUN-528: Move cloudflared into a separate repo
This commit is contained in:
31
h2mux/idletimer_test.go
Normal file
31
h2mux/idletimer_test.go
Normal file
@@ -0,0 +1,31 @@
|
||||
package h2mux
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestRetry(t *testing.T) {
|
||||
timer := NewIdleTimer(time.Second, 2)
|
||||
assert.Equal(t, uint64(0), timer.RetryCount())
|
||||
ok := timer.Retry()
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(1), timer.RetryCount())
|
||||
ok = timer.Retry()
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(2), timer.RetryCount())
|
||||
ok = timer.Retry()
|
||||
assert.False(t, ok)
|
||||
}
|
||||
|
||||
func TestMarkActive(t *testing.T) {
|
||||
timer := NewIdleTimer(time.Second, 2)
|
||||
assert.Equal(t, uint64(0), timer.RetryCount())
|
||||
ok := timer.Retry()
|
||||
assert.True(t, ok)
|
||||
assert.Equal(t, uint64(1), timer.RetryCount())
|
||||
timer.MarkActive()
|
||||
assert.Equal(t, uint64(0), timer.RetryCount())
|
||||
}
|
Reference in New Issue
Block a user