forked from quic-go/quic-go
8ef0a71581
* remove RTT from session ticket * save RTT in token * use mus as rtt unit in token * move RTT from session state to token on client side * Rename the rtt variables Co-authored-by: Marten Seemann <martenseemann@gmail.com> * Update internal/handshake/token_generator.go Co-authored-by: Marten Seemann <martenseemann@gmail.com> * Update token_generator.go * Update connection.go * Update token_generator.go * correct slice access in fuzz.go * rearrange connection constructor parameters --------- Co-authored-by: Marten Seemann <martenseemann@gmail.com>
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func mockToken(num int) *ClientToken {
|
|
return &ClientToken{data: []byte(fmt.Sprintf("%d", num)), rtt: 1337 * time.Millisecond}
|
|
}
|
|
|
|
func TestTokenStoreSingleOrigin(t *testing.T) {
|
|
const origin = "localhost"
|
|
|
|
s := NewLRUTokenStore(1, 3)
|
|
s.Put(origin, mockToken(1))
|
|
s.Put(origin, mockToken(2))
|
|
require.Equal(t, mockToken(2), s.Pop(origin))
|
|
require.Equal(t, mockToken(1), s.Pop(origin))
|
|
require.Nil(t, s.Pop(origin))
|
|
|
|
// now add more tokens than the cache size
|
|
s.Put(origin, mockToken(1))
|
|
s.Put(origin, mockToken(2))
|
|
s.Put(origin, mockToken(3))
|
|
require.Equal(t, mockToken(3), s.Pop(origin))
|
|
s.Put(origin, mockToken(4))
|
|
s.Put(origin, mockToken(5))
|
|
require.Equal(t, mockToken(5), s.Pop(origin))
|
|
require.Equal(t, mockToken(4), s.Pop(origin))
|
|
require.Equal(t, mockToken(2), s.Pop(origin))
|
|
require.Nil(t, s.Pop(origin))
|
|
}
|
|
|
|
func TestTokenStoreMultipleOrigins(t *testing.T) {
|
|
s := NewLRUTokenStore(3, 4)
|
|
|
|
s.Put("host1", mockToken(1))
|
|
s.Put("host2", mockToken(2))
|
|
s.Put("host3", mockToken(3))
|
|
s.Put("host4", mockToken(4))
|
|
require.Nil(t, s.Pop("host1"))
|
|
require.Equal(t, mockToken(2), s.Pop("host2"))
|
|
require.Equal(t, mockToken(3), s.Pop("host3"))
|
|
require.Equal(t, mockToken(4), s.Pop("host4"))
|
|
}
|
|
|
|
func TestTokenStoreUpdates(t *testing.T) {
|
|
s := NewLRUTokenStore(3, 4)
|
|
s.Put("host1", mockToken(1))
|
|
s.Put("host2", mockToken(2))
|
|
s.Put("host3", mockToken(3))
|
|
s.Put("host1", mockToken(11))
|
|
// make sure one is evicted
|
|
s.Put("host4", mockToken(4))
|
|
require.Nil(t, s.Pop("host2"))
|
|
require.Equal(t, mockToken(11), s.Pop("host1"))
|
|
require.Equal(t, mockToken(1), s.Pop("host1"))
|
|
require.Equal(t, mockToken(3), s.Pop("host3"))
|
|
require.Equal(t, mockToken(4), s.Pop("host4"))
|
|
}
|
|
|
|
func TestTokenStoreEviction(t *testing.T) {
|
|
s := NewLRUTokenStore(3, 4)
|
|
|
|
s.Put("host1", mockToken(1))
|
|
s.Put("host2", mockToken(2))
|
|
s.Put("host3", mockToken(3))
|
|
require.Equal(t, mockToken(2), s.Pop("host2"))
|
|
require.Nil(t, s.Pop("host2"))
|
|
// host2 is now empty and should have been deleted, making space for host4
|
|
s.Put("host4", mockToken(4))
|
|
require.Equal(t, mockToken(1), s.Pop("host1"))
|
|
require.Equal(t, mockToken(3), s.Pop("host3"))
|
|
require.Equal(t, mockToken(4), s.Pop("host4"))
|
|
}
|