-
Notifications
You must be signed in to change notification settings - Fork 2
/
database_mongo_mock.go
100 lines (80 loc) · 3.41 KB
/
database_mongo_mock.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
package core
import (
"github.com/stretchr/testify/mock"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
type MockMongoDB struct {
mock.Mock
}
func NewMockMongoDB() *MockMongoDB {
return &MockMongoDB{}
}
func (m *MockMongoDB) Close() {
m.Called()
}
func (m *MockMongoDB) FindPagination(dest interface{}, coll string, filter interface{}, pageOptions *PageOptions, opts ...*options.FindOptions) (*PageResponse, error) {
args := m.Called(dest, coll, filter, pageOptions, opts)
return args.Get(0).(*PageResponse), args.Error(1)
}
func (m *MockMongoDB) Count(coll string, filter interface{}, opts ...*options.CountOptions) (int64, error) {
args := m.Called(coll, filter, opts)
return args.Get(0).(int64), args.Error(1)
}
func (m *MockMongoDB) FindAggregate(dest interface{}, coll string, pipeline interface{}, opts ...*options.AggregateOptions) error {
args := m.Called(dest, coll, pipeline, opts)
return args.Error(0)
}
func (m *MockMongoDB) FindAggregatePagination(dest interface{}, coll string, pipeline interface{}, pageOptions *PageOptions, opts ...*options.AggregateOptions) (*PageResponse, error) {
args := m.Called(dest, coll, pipeline, pageOptions, opts)
return args.Get(0).(*PageResponse), args.Error(1)
}
func (m *MockMongoDB) FindAggregateOne(dest interface{}, coll string, pipeline interface{}, opts ...*options.AggregateOptions) error {
args := m.Called(dest, coll, pipeline, opts)
return args.Error(0)
}
func (m *MockMongoDB) Find(dest interface{}, coll string, filter interface{}, opts ...*options.FindOptions) error {
args := m.Called(dest, coll, filter, opts)
return args.Error(0)
}
func (m *MockMongoDB) UpdateOne(coll string, filter interface{}, update interface{},
opts ...*options.UpdateOptions) (*mongo.UpdateResult, error) {
args := m.Called(coll, filter, update, opts)
return args.Get(0).(*mongo.UpdateResult), args.Error(1)
}
func (m *MockMongoDB) FindOneAndUpdate(dest interface{}, coll string, filter interface{}, update interface{},
opts ...*options.FindOneAndUpdateOptions) error {
args := m.Called(dest, coll, filter, update, opts)
return args.Error(0)
}
func (m *MockMongoDB) Drop(coll string) error {
args := m.Called(coll)
return args.Error(0)
}
func (m *MockMongoDB) DeleteOne(coll string, filter interface{}, opts ...*options.DeleteOptions) (*mongo.DeleteResult, error) {
args := m.Called(coll, filter, opts)
return args.Get(0).(*mongo.DeleteResult), args.Error(1)
}
func (m *MockMongoDB) FindOneAndDelete(coll string, filter interface{}, opts ...*options.FindOneAndDeleteOptions) error {
args := m.Called(coll, filter, opts)
return args.Error(0)
}
func (m *MockMongoDB) DeleteMany(coll string, filter interface{}, opts ...*options.DeleteOptions) (*mongo.DeleteResult, error) {
args := m.Called(coll, filter, opts)
return args.Get(0).(*mongo.DeleteResult), args.Error(1)
}
func (m *MockMongoDB) FindOne(dest interface{}, coll string, filter interface{}, opts ...*options.FindOneOptions) error {
args := m.Called(dest, coll, filter, opts)
return args.Error(0)
}
func (m *MockMongoDB) Create(coll string, document interface{}, opts ...*options.InsertOneOptions) (*mongo.InsertOneResult, error) {
args := m.Called(coll, document, opts)
return args.Get(0).(*mongo.InsertOneResult), args.Error(1)
}
func (m *MockMongoDB) DB() *mongo.Database {
args := m.Called()
return args.Get(0).(*mongo.Database)
}
func (m *MockMongoDB) Helper() IMongoDBHelper {
return NewMongoHelper()
}