forked from argoproj/argo-workflows
-
Notifications
You must be signed in to change notification settings - Fork 0
/
OffloadNodeStatusRepo.go
133 lines (107 loc) · 3.09 KB
/
OffloadNodeStatusRepo.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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
// Code generated by mockery v1.0.0. DO NOT EDIT.
package mocks
import (
sqldb "github.com/argoproj/argo/persist/sqldb"
mock "github.com/stretchr/testify/mock"
v1alpha1 "github.com/argoproj/argo/pkg/apis/workflow/v1alpha1"
)
// OffloadNodeStatusRepo is an autogenerated mock type for the OffloadNodeStatusRepo type
type OffloadNodeStatusRepo struct {
mock.Mock
}
// Delete provides a mock function with given fields: uid, version
func (_m *OffloadNodeStatusRepo) Delete(uid string, version string) error {
ret := _m.Called(uid, version)
var r0 error
if rf, ok := ret.Get(0).(func(string, string) error); ok {
r0 = rf(uid, version)
} else {
r0 = ret.Error(0)
}
return r0
}
// Get provides a mock function with given fields: uid, version
func (_m *OffloadNodeStatusRepo) Get(uid string, version string) (v1alpha1.Nodes, error) {
ret := _m.Called(uid, version)
var r0 v1alpha1.Nodes
if rf, ok := ret.Get(0).(func(string, string) v1alpha1.Nodes); ok {
r0 = rf(uid, version)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(v1alpha1.Nodes)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(string, string) error); ok {
r1 = rf(uid, version)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// IsEnabled provides a mock function with given fields:
func (_m *OffloadNodeStatusRepo) IsEnabled() bool {
ret := _m.Called()
var r0 bool
if rf, ok := ret.Get(0).(func() bool); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// List provides a mock function with given fields: namespace
func (_m *OffloadNodeStatusRepo) List(namespace string) (map[sqldb.UUIDVersion]v1alpha1.Nodes, error) {
ret := _m.Called(namespace)
var r0 map[sqldb.UUIDVersion]v1alpha1.Nodes
if rf, ok := ret.Get(0).(func(string) map[sqldb.UUIDVersion]v1alpha1.Nodes); ok {
r0 = rf(namespace)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(map[sqldb.UUIDVersion]v1alpha1.Nodes)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(string) error); ok {
r1 = rf(namespace)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// ListOldOffloads provides a mock function with given fields: namespace
func (_m *OffloadNodeStatusRepo) ListOldOffloads(namespace string) ([]sqldb.UUIDVersion, error) {
ret := _m.Called(namespace)
var r0 []sqldb.UUIDVersion
if rf, ok := ret.Get(0).(func(string) []sqldb.UUIDVersion); ok {
r0 = rf(namespace)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]sqldb.UUIDVersion)
}
}
var r1 error
if rf, ok := ret.Get(1).(func(string) error); ok {
r1 = rf(namespace)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// Save provides a mock function with given fields: uid, namespace, nodes
func (_m *OffloadNodeStatusRepo) Save(uid string, namespace string, nodes v1alpha1.Nodes) (string, error) {
ret := _m.Called(uid, namespace, nodes)
var r0 string
if rf, ok := ret.Get(0).(func(string, string, v1alpha1.Nodes) string); ok {
r0 = rf(uid, namespace, nodes)
} else {
r0 = ret.Get(0).(string)
}
var r1 error
if rf, ok := ret.Get(1).(func(string, string, v1alpha1.Nodes) error); ok {
r1 = rf(uid, namespace, nodes)
} else {
r1 = ret.Error(1)
}
return r0, r1
}