database.go RawHistoryFolder

1// Code generated by mockery; DO NOT EDIT.
2// github.com/vektra/mockery
3// template: testify
4
5package mocks
6
7import (
8	"context"
9	"database/sql"
10
11	mock "github.com/stretchr/testify/mock"
12)
13
14// NewMockDatabaseProvider creates a new instance of MockDatabaseProvider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
15// The first argument is typically a *testing.T value.
16func NewMockDatabaseProvider(t interface {
17	mock.TestingT
18	Cleanup(func())
19}) *MockDatabaseProvider {
20	mock := &MockDatabaseProvider{}
21	mock.Mock.Test(t)
22
23	t.Cleanup(func() { mock.AssertExpectations(t) })
24
25	return mock
26}
27
28// MockDatabaseProvider is an autogenerated mock type for the DatabaseProvider type
29type MockDatabaseProvider struct {
30	mock.Mock
31}
32
33type MockDatabaseProvider_Expecter struct {
34	mock *mock.Mock
35}
36
37func (_m *MockDatabaseProvider) EXPECT() *MockDatabaseProvider_Expecter {
38	return &MockDatabaseProvider_Expecter{mock: &_m.Mock}
39}
40
41// Close provides a mock function for the type MockDatabaseProvider
42func (_mock *MockDatabaseProvider) Close() error {
43	ret := _mock.Called()
44
45	if len(ret) == 0 {
46		panic("no return value specified for Close")
47	}
48
49	var r0 error
50	if returnFunc, ok := ret.Get(0).(func() error); ok {
51		r0 = returnFunc()
52	} else {
53		r0 = ret.Error(0)
54	}
55	return r0
56}
57
58// MockDatabaseProvider_Close_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Close'
59type MockDatabaseProvider_Close_Call struct {
60	*mock.Call
61}
62
63// Close is a helper method to define mock.On call
64func (_e *MockDatabaseProvider_Expecter) Close() *MockDatabaseProvider_Close_Call {
65	return &MockDatabaseProvider_Close_Call{Call: _e.mock.On("Close")}
66}
67
68func (_c *MockDatabaseProvider_Close_Call) Run(run func()) *MockDatabaseProvider_Close_Call {
69	_c.Call.Run(func(args mock.Arguments) {
70		run()
71	})
72	return _c
73}
74
75func (_c *MockDatabaseProvider_Close_Call) Return(err error) *MockDatabaseProvider_Close_Call {
76	_c.Call.Return(err)
77	return _c
78}
79
80func (_c *MockDatabaseProvider_Close_Call) RunAndReturn(run func() error) *MockDatabaseProvider_Close_Call {
81	_c.Call.Return(run)
82	return _c
83}
84
85// DB provides a mock function for the type MockDatabaseProvider
86func (_mock *MockDatabaseProvider) DB(ctx context.Context) (*sql.DB, error) {
87	ret := _mock.Called(ctx)
88
89	if len(ret) == 0 {
90		panic("no return value specified for DB")
91	}
92
93	var r0 *sql.DB
94	var r1 error
95	if returnFunc, ok := ret.Get(0).(func(context.Context) (*sql.DB, error)); ok {
96		return returnFunc(ctx)
97	}
98	if returnFunc, ok := ret.Get(0).(func(context.Context) *sql.DB); ok {
99		r0 = returnFunc(ctx)
100	} else {
101		if ret.Get(0) != nil {
102			r0 = ret.Get(0).(*sql.DB)
103		}
104	}
105	if returnFunc, ok := ret.Get(1).(func(context.Context) error); ok {
106		r1 = returnFunc(ctx)
107	} else {
108		r1 = ret.Error(1)
109	}
110	return r0, r1
111}
112
113// MockDatabaseProvider_DB_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DB'
114type MockDatabaseProvider_DB_Call struct {
115	*mock.Call
116}
117
118// DB is a helper method to define mock.On call
119//   - ctx context.Context
120func (_e *MockDatabaseProvider_Expecter) DB(ctx any) *MockDatabaseProvider_DB_Call {
121	return &MockDatabaseProvider_DB_Call{Call: _e.mock.On("DB", ctx)}
122}
123
124func (_c *MockDatabaseProvider_DB_Call) Run(run func(ctx context.Context)) *MockDatabaseProvider_DB_Call {
125	_c.Call.Run(func(args mock.Arguments) {
126		var arg0 context.Context
127		if args[0] != nil {
128			arg0 = args[0].(context.Context)
129		}
130		run(
131			arg0,
132		)
133	})
134	return _c
135}
136
137func (_c *MockDatabaseProvider_DB_Call) Return(dB *sql.DB, err error) *MockDatabaseProvider_DB_Call {
138	_c.Call.Return(dB, err)
139	return _c
140}
141
142func (_c *MockDatabaseProvider_DB_Call) RunAndReturn(run func(ctx context.Context) (*sql.DB, error)) *MockDatabaseProvider_DB_Call {
143	_c.Call.Return(run)
144	return _c
145}
146
147// IsInitialized provides a mock function for the type MockDatabaseProvider
148func (_mock *MockDatabaseProvider) IsInitialized() bool {
149	ret := _mock.Called()
150
151	if len(ret) == 0 {
152		panic("no return value specified for IsInitialized")
153	}
154
155	var r0 bool
156	if returnFunc, ok := ret.Get(0).(func() bool); ok {
157		r0 = returnFunc()
158	} else {
159		r0 = ret.Get(0).(bool)
160	}
161	return r0
162}
163
164// MockDatabaseProvider_IsInitialized_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IsInitialized'
165type MockDatabaseProvider_IsInitialized_Call struct {
166	*mock.Call
167}
168
169// IsInitialized is a helper method to define mock.On call
170func (_e *MockDatabaseProvider_Expecter) IsInitialized() *MockDatabaseProvider_IsInitialized_Call {
171	return &MockDatabaseProvider_IsInitialized_Call{Call: _e.mock.On("IsInitialized")}
172}
173
174func (_c *MockDatabaseProvider_IsInitialized_Call) Run(run func()) *MockDatabaseProvider_IsInitialized_Call {
175	_c.Call.Run(func(args mock.Arguments) {
176		run()
177	})
178	return _c
179}
180
181func (_c *MockDatabaseProvider_IsInitialized_Call) Return(b bool) *MockDatabaseProvider_IsInitialized_Call {
182	_c.Call.Return(b)
183	return _c
184}
185
186func (_c *MockDatabaseProvider_IsInitialized_Call) RunAndReturn(run func() bool) *MockDatabaseProvider_IsInitialized_Call {
187	_c.Call.Return(run)
188	return _c
189}