-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
113 lines (98 loc) · 3.33 KB
/
db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Code generated by MockGen. DO NOT EDIT.
// Source: db.go
//
// Generated by this command:
//
// mockgen -source db.go -destination ../../mocks/db.go -package mocks -mock_names Database=Database
//
// Package mocks is a generated GoMock package.
package mocks
import (
context "context"
reflect "reflect"
pgx "github.com/jackc/pgx/v5"
gomock "go.uber.org/mock/gomock"
)
// Database is a mock of Database interface.
type Database struct {
ctrl *gomock.Controller
recorder *DatabaseMockRecorder
}
// DatabaseMockRecorder is the mock recorder for Database.
type DatabaseMockRecorder struct {
mock *Database
}
// NewDatabase creates a new mock instance.
func NewDatabase(ctrl *gomock.Controller) *Database {
mock := &Database{ctrl: ctrl}
mock.recorder = &DatabaseMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *Database) EXPECT() *DatabaseMockRecorder {
return m.recorder
}
// Create mocks base method.
func (m *Database) Create(arg0 context.Context, arg1 string, arg2 ...any) (uint64, error) {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "Create", varargs...)
ret0, _ := ret[0].(uint64)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Create indicates an expected call of Create.
func (mr *DatabaseMockRecorder) Create(arg0, arg1 any, arg2 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*Database)(nil).Create), varargs...)
}
// DoAtomic mocks base method.
func (m *Database) DoAtomic(arg0 context.Context, arg1 func(context.Context) error) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DoAtomic", arg0, arg1)
ret0, _ := ret[0].(error)
return ret0
}
// DoAtomic indicates an expected call of DoAtomic.
func (mr *DatabaseMockRecorder) DoAtomic(arg0, arg1 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DoAtomic", reflect.TypeOf((*Database)(nil).DoAtomic), arg0, arg1)
}
// Get mocks base method.
func (m *Database) Get(arg0 context.Context, arg1 func(pgx.Row) error, arg2 string, arg3 ...any) error {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1, arg2}
for _, a := range arg3 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "Get", varargs...)
ret0, _ := ret[0].(error)
return ret0
}
// Get indicates an expected call of Get.
func (mr *DatabaseMockRecorder) Get(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1, arg2}, arg3...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*Database)(nil).Get), varargs...)
}
// One mocks base method.
func (m *Database) One(arg0 context.Context, arg1 string, arg2 ...any) error {
m.ctrl.T.Helper()
varargs := []any{arg0, arg1}
for _, a := range arg2 {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "One", varargs...)
ret0, _ := ret[0].(error)
return ret0
}
// One indicates an expected call of One.
func (mr *DatabaseMockRecorder) One(arg0, arg1 any, arg2 ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{arg0, arg1}, arg2...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "One", reflect.TypeOf((*Database)(nil).One), varargs...)
}