gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

remove mock TokenStore implementation (#4597)

Browse Source 
No functional change expected.
This commit is contained in:
Marten Seemann
2024-07-24 17:28:01 -06:00
committed by GitHub
parent 8451b0afd7
commit 9b70412b0f
3 changed files with 4 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
connection_test.go
View File

@@ -2718,19 +2718,17 @@ var _ = Describe("Client Connection", func() {
})
Context("handling tokens", func() {
var mockTokenStore *MockTokenStore
var tokenStore TokenStore
BeforeEach(func() {
mockTokenStore = NewMockTokenStore(mockCtrl)
tlsConf = &tls.Config{ServerName: "server"}
quicConf.TokenStore = mockTokenStore
mockTokenStore.EXPECT().Pop(gomock.Any())
quicConf.TokenStore = mockTokenStore
tokenStore = NewLRUTokenStore(10, 10)
quicConf.TokenStore = tokenStore
})
It("handles NEW_TOKEN frames", func() {
mockTokenStore.EXPECT().Put("server", &ClientToken{data: []byte("foobar")})
Expect(conn.handleNewTokenFrame(&wire.NewTokenFrame{Token: []byte("foobar")})).To(Succeed())
Expect(tokenStore.Pop("server")).To(Equal(&ClientToken{data: []byte("foobar")}))
})
})

113
mock_token_store_test.go
View File

@@ -1,113 +0,0 @@
// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/quic-go/quic-go (interfaces: TokenStore)
//
// Generated by this command:
//
// mockgen -typed -package quic -self_package github.com/quic-go/quic-go -self_package github.com/quic-go/quic-go -destination mock_token_store_test.go github.com/quic-go/quic-go TokenStore
//
// Package quic is a generated GoMock package.
package quic
import (
reflect "reflect"
gomock "go.uber.org/mock/gomock"
)
// MockTokenStore is a mock of TokenStore interface.
type MockTokenStore struct {
ctrl *gomock.Controller
recorder *MockTokenStoreMockRecorder
}
// MockTokenStoreMockRecorder is the mock recorder for MockTokenStore.
type MockTokenStoreMockRecorder struct {
mock *MockTokenStore
}
// NewMockTokenStore creates a new mock instance.
func NewMockTokenStore(ctrl *gomock.Controller) *MockTokenStore {
mock := &MockTokenStore{ctrl: ctrl}
mock.recorder = &MockTokenStoreMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockTokenStore) EXPECT() *MockTokenStoreMockRecorder {
return m.recorder
}
// Pop mocks base method.
func (m *MockTokenStore) Pop(arg0 string) *ClientToken {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Pop", arg0)
ret0, _ := ret[0].(*ClientToken)
return ret0
}
// Pop indicates an expected call of Pop.
func (mr *MockTokenStoreMockRecorder) Pop(arg0 any) *MockTokenStorePopCall {
mr.mock.ctrl.T.Helper()
call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pop", reflect.TypeOf((*MockTokenStore)(nil).Pop), arg0)
return &MockTokenStorePopCall{Call: call}
}
// MockTokenStorePopCall wrap *gomock.Call
type MockTokenStorePopCall struct {
*gomock.Call
}
// Return rewrite *gomock.Call.Return
func (c *MockTokenStorePopCall) Return(arg0 *ClientToken) *MockTokenStorePopCall {
c.Call = c.Call.Return(arg0)
return c
}
// Do rewrite *gomock.Call.Do
func (c *MockTokenStorePopCall) Do(f func(string) *ClientToken) *MockTokenStorePopCall {
c.Call = c.Call.Do(f)
return c
}
// DoAndReturn rewrite *gomock.Call.DoAndReturn
func (c *MockTokenStorePopCall) DoAndReturn(f func(string) *ClientToken) *MockTokenStorePopCall {
c.Call = c.Call.DoAndReturn(f)
return c
}
// Put mocks base method.
func (m *MockTokenStore) Put(arg0 string, arg1 *ClientToken) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Put", arg0, arg1)
}
// Put indicates an expected call of Put.
func (mr *MockTokenStoreMockRecorder) Put(arg0, arg1 any) *MockTokenStorePutCall {
mr.mock.ctrl.T.Helper()
call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockTokenStore)(nil).Put), arg0, arg1)
return &MockTokenStorePutCall{Call: call}
}
// MockTokenStorePutCall wrap *gomock.Call
type MockTokenStorePutCall struct {
*gomock.Call
}
// Return rewrite *gomock.Call.Return
func (c *MockTokenStorePutCall) Return() *MockTokenStorePutCall {
c.Call = c.Call.Return()
return c
}
// Do rewrite *gomock.Call.Do
func (c *MockTokenStorePutCall) Do(f func(string, *ClientToken)) *MockTokenStorePutCall {
c.Call = c.Call.Do(f)
return c
}
// DoAndReturn rewrite *gomock.Call.DoAndReturn
func (c *MockTokenStorePutCall) DoAndReturn(f func(string, *ClientToken)) *MockTokenStorePutCall {
c.Call = c.Call.DoAndReturn(f)
return c
}

1
mockgen.go
View File

@@ -69,5 +69,4 @@ type PacketHandlerManager = packetHandlerManager
//
//go:generate sh -c "go run go.uber.org/mock/mockgen -typed -package quic -self_package github.com/quic-go/quic-go -source sys_conn_oob.go -destination mock_batch_conn_test.go -mock_names batchConn=MockBatchConn"
//go:generate sh -c "go run go.uber.org/mock/mockgen -typed -package quic -self_package github.com/quic-go/quic-go -self_package github.com/quic-go/quic-go -destination mock_token_store_test.go github.com/quic-go/quic-go TokenStore"
//go:generate sh -c "go run go.uber.org/mock/mockgen -typed -package quic -self_package github.com/quic-go/quic-go -self_package github.com/quic-go/quic-go -destination mock_packetconn_test.go net PacketConn"