mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-14 18:59:08 +02:00
* extract peer update loop into a dedicated struct and wrap it in tests Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> * make linter happy Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io> --------- Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io>
112 lines
3.7 KiB
Go
112 lines
3.7 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: ./token_mgr.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -source=./token_mgr.go -destination=./token_mgr_mock.go -package=grpc
|
|
//
|
|
|
|
// Package grpc is a generated GoMock package.
|
|
package grpc
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
gomock "go.uber.org/mock/gomock"
|
|
wgtypes "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
)
|
|
|
|
// MockSecretsManager is a mock of SecretsManager interface.
|
|
type MockSecretsManager struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockSecretsManagerMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockSecretsManagerMockRecorder is the mock recorder for MockSecretsManager.
|
|
type MockSecretsManagerMockRecorder struct {
|
|
mock *MockSecretsManager
|
|
}
|
|
|
|
// NewMockSecretsManager creates a new mock instance.
|
|
func NewMockSecretsManager(ctrl *gomock.Controller) *MockSecretsManager {
|
|
mock := &MockSecretsManager{ctrl: ctrl}
|
|
mock.recorder = &MockSecretsManagerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockSecretsManager) EXPECT() *MockSecretsManagerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CancelRefresh mocks base method.
|
|
func (m *MockSecretsManager) CancelRefresh(peerKey string) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "CancelRefresh", peerKey)
|
|
}
|
|
|
|
// CancelRefresh indicates an expected call of CancelRefresh.
|
|
func (mr *MockSecretsManagerMockRecorder) CancelRefresh(peerKey any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelRefresh", reflect.TypeOf((*MockSecretsManager)(nil).CancelRefresh), peerKey)
|
|
}
|
|
|
|
// GenerateRelayToken mocks base method.
|
|
func (m *MockSecretsManager) GenerateRelayToken() (*Token, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GenerateRelayToken")
|
|
ret0, _ := ret[0].(*Token)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GenerateRelayToken indicates an expected call of GenerateRelayToken.
|
|
func (mr *MockSecretsManagerMockRecorder) GenerateRelayToken() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GenerateRelayToken", reflect.TypeOf((*MockSecretsManager)(nil).GenerateRelayToken))
|
|
}
|
|
|
|
// GenerateTurnToken mocks base method.
|
|
func (m *MockSecretsManager) GenerateTurnToken() (*Token, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GenerateTurnToken")
|
|
ret0, _ := ret[0].(*Token)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GenerateTurnToken indicates an expected call of GenerateTurnToken.
|
|
func (mr *MockSecretsManagerMockRecorder) GenerateTurnToken() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GenerateTurnToken", reflect.TypeOf((*MockSecretsManager)(nil).GenerateTurnToken))
|
|
}
|
|
|
|
// GetWGKey mocks base method.
|
|
func (m *MockSecretsManager) GetWGKey() (wgtypes.Key, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetWGKey")
|
|
ret0, _ := ret[0].(wgtypes.Key)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetWGKey indicates an expected call of GetWGKey.
|
|
func (mr *MockSecretsManagerMockRecorder) GetWGKey() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWGKey", reflect.TypeOf((*MockSecretsManager)(nil).GetWGKey))
|
|
}
|
|
|
|
// SetupRefresh mocks base method.
|
|
func (m *MockSecretsManager) SetupRefresh(ctx context.Context, accountID, peerKey string) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "SetupRefresh", ctx, accountID, peerKey)
|
|
}
|
|
|
|
// SetupRefresh indicates an expected call of SetupRefresh.
|
|
func (mr *MockSecretsManagerMockRecorder) SetupRefresh(ctx, accountID, peerKey any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetupRefresh", reflect.TypeOf((*MockSecretsManager)(nil).SetupRefresh), ctx, accountID, peerKey)
|
|
}
|