mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-29 18:19:07 +02:00
Pascal Fischer
782c943410
* extract shared db conn + data repository * extract repository interface * protect against nested transactions * fix mysql and db conn creation * fix context management * remove query warpper * remove withContext and withLock wrapper * remove context from function call * remove in memory mode * fix nested transaction handling * use db directly * remove leftover test * remove pool close on error during conn creation
103 lines
3.4 KiB
Go
103 lines
3.4 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: ./repository.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package accesslogs -destination=repository_mock.go -source=./repository.go -build_flags=-mod=mod
|
|
//
|
|
|
|
// Package accesslogs is a generated GoMock package.
|
|
package accesslogs
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
db "github.com/netbirdio/netbird/management/internals/shared/db"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockRepository is a mock of Repository interface.
|
|
type MockRepository struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockRepositoryMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockRepositoryMockRecorder is the mock recorder for MockRepository.
|
|
type MockRepositoryMockRecorder struct {
|
|
mock *MockRepository
|
|
}
|
|
|
|
// NewMockRepository creates a new mock instance.
|
|
func NewMockRepository(ctrl *gomock.Controller) *MockRepository {
|
|
mock := &MockRepository{ctrl: ctrl}
|
|
mock.recorder = &MockRepositoryMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockRepository) EXPECT() *MockRepositoryMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Create mocks base method.
|
|
func (m *MockRepository) Create(ctx context.Context, entry *AccessLogEntry) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Create", ctx, entry)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Create indicates an expected call of Create.
|
|
func (mr *MockRepositoryMockRecorder) Create(ctx, entry any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockRepository)(nil).Create), ctx, entry)
|
|
}
|
|
|
|
// DeleteOlderThan mocks base method.
|
|
func (m *MockRepository) DeleteOlderThan(ctx context.Context, olderThan time.Time) (int64, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteOlderThan", ctx, olderThan)
|
|
ret0, _ := ret[0].(int64)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// DeleteOlderThan indicates an expected call of DeleteOlderThan.
|
|
func (mr *MockRepositoryMockRecorder) DeleteOlderThan(ctx, olderThan any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOlderThan", reflect.TypeOf((*MockRepository)(nil).DeleteOlderThan), ctx, olderThan)
|
|
}
|
|
|
|
// ListByAccount mocks base method.
|
|
func (m *MockRepository) ListByAccount(ctx context.Context, lockStrength db.LockingStrength, accountID string, filter AccessLogFilter) ([]*AccessLogEntry, int64, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ListByAccount", ctx, lockStrength, accountID, filter)
|
|
ret0, _ := ret[0].([]*AccessLogEntry)
|
|
ret1, _ := ret[1].(int64)
|
|
ret2, _ := ret[2].(error)
|
|
return ret0, ret1, ret2
|
|
}
|
|
|
|
// ListByAccount indicates an expected call of ListByAccount.
|
|
func (mr *MockRepositoryMockRecorder) ListByAccount(ctx, lockStrength, accountID, filter any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListByAccount", reflect.TypeOf((*MockRepository)(nil).ListByAccount), ctx, lockStrength, accountID, filter)
|
|
}
|
|
|
|
// WithTx mocks base method.
|
|
func (m *MockRepository) WithTx(tx *db.Tx) Repository {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "WithTx", tx)
|
|
ret0, _ := ret[0].(Repository)
|
|
return ret0
|
|
}
|
|
|
|
// WithTx indicates an expected call of WithTx.
|
|
func (mr *MockRepositoryMockRecorder) WithTx(tx any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTx", reflect.TypeOf((*MockRepository)(nil).WithTx), tx)
|
|
}
|