forked from ohsu-comp-bio/funnel
/
Database_mock.go
116 lines (94 loc) · 2.84 KB
/
Database_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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package mocks
import context "golang.org/x/net/context"
import events "github.com/ohsu-comp-bio/funnel/events"
import mock "github.com/stretchr/testify/mock"
import scheduler "github.com/ohsu-comp-bio/funnel/proto/scheduler"
import tes "github.com/ohsu-comp-bio/funnel/proto/tes"
// Database is an autogenerated mock type for the Database type
type Database struct {
mock.Mock
}
// DeleteNode provides a mock function with given fields: _a0, _a1
func (_m *Database) DeleteNode(_a0 context.Context, _a1 *scheduler.Node) error {
ret := _m.Called(_a0, _a1)
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, *scheduler.Node) error); ok {
r0 = rf(_a0, _a1)
} else {
r0 = ret.Error(0)
}
return r0
}
// ListNodes provides a mock function with given fields: _a0, _a1
func (_m *Database) ListNodes(_a0 context.Context, _a1 *scheduler.ListNodesRequest) (*scheduler.ListNodesResponse, error) {
ret := _m.Called(_a0, _a1)
var r0 *scheduler.ListNodesResponse
if rf, ok := ret.Get(0).(func(context.Context, *scheduler.ListNodesRequest) *scheduler.ListNodesResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*scheduler.ListNodesResponse)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(context.Context, *scheduler.ListNodesRequest) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// PutNode provides a mock function with given fields: _a0, _a1
func (_m *Database) PutNode(_a0 context.Context, _a1 *scheduler.Node) (*scheduler.PutNodeResponse, error) {
ret := _m.Called(_a0, _a1)
var r0 *scheduler.PutNodeResponse
if rf, ok := ret.Get(0).(func(context.Context, *scheduler.Node) *scheduler.PutNodeResponse); ok {
r0 = rf(_a0, _a1)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*scheduler.PutNodeResponse)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(context.Context, *scheduler.Node) error); ok {
r1 = rf(_a0, _a1)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// QueueTask provides a mock function with given fields: _a0
func (_m *Database) QueueTask(_a0 *tes.Task) error {
ret := _m.Called(_a0)
var r0 error
if rf, ok := ret.Get(0).(func(*tes.Task) error); ok {
r0 = rf(_a0)
} else {
r0 = ret.Error(0)
}
return r0
}
// ReadQueue provides a mock function with given fields: _a0
func (_m *Database) ReadQueue(_a0 int) []*tes.Task {
ret := _m.Called(_a0)
var r0 []*tes.Task
if rf, ok := ret.Get(0).(func(int) []*tes.Task); ok {
r0 = rf(_a0)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*tes.Task)
}
}
return r0
}
// WriteContext provides a mock function with given fields: _a0, _a1
func (_m *Database) WriteContext(_a0 context.Context, _a1 *events.Event) error {
ret := _m.Called(_a0, _a1)
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, *events.Event) error); ok {
r0 = rf(_a0, _a1)
} else {
r0 = ret.Error(0)
}
return r0
}