-
Notifications
You must be signed in to change notification settings - Fork 44
/
Copy pathworker.go
137 lines (116 loc) · 4.65 KB
/
worker.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
134
135
136
137
// Code generated by MockGen. DO NOT EDIT.
// Source: worker.go
// Package mock_dao is a generated GoMock package.
package mock_dao
import (
context "context"
reflect "reflect"
model "github.com/TUM-Dev/gocast/model"
gomock "github.com/golang/mock/gomock"
)
// MockWorkerDao is a mock of WorkerDao interface.
type MockWorkerDao struct {
ctrl *gomock.Controller
recorder *MockWorkerDaoMockRecorder
}
// MockWorkerDaoMockRecorder is the mock recorder for MockWorkerDao.
type MockWorkerDaoMockRecorder struct {
mock *MockWorkerDao
}
// NewMockWorkerDao creates a new mock instance.
func NewMockWorkerDao(ctrl *gomock.Controller) *MockWorkerDao {
mock := &MockWorkerDao{ctrl: ctrl}
mock.recorder = &MockWorkerDaoMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockWorkerDao) EXPECT() *MockWorkerDaoMockRecorder {
return m.recorder
}
// CreateWorker mocks base method.
func (m *MockWorkerDao) CreateWorker(worker *model.Worker) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateWorker", worker)
ret0, _ := ret[0].(error)
return ret0
}
// CreateWorker indicates an expected call of CreateWorker.
func (mr *MockWorkerDaoMockRecorder) CreateWorker(worker interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateWorker", reflect.TypeOf((*MockWorkerDao)(nil).CreateWorker), worker)
}
// DeleteWorker mocks base method.
func (m *MockWorkerDao) DeleteWorker(workerID string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteWorker", workerID)
ret0, _ := ret[0].(error)
return ret0
}
// DeleteWorker indicates an expected call of DeleteWorker.
func (mr *MockWorkerDaoMockRecorder) DeleteWorker(workerID interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorker", reflect.TypeOf((*MockWorkerDao)(nil).DeleteWorker), workerID)
}
// GetAliveWorkers mocks base method.
func (m *MockWorkerDao) GetAliveWorkers() []model.Worker {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetAliveWorkers")
ret0, _ := ret[0].([]model.Worker)
return ret0
}
// GetAliveWorkers indicates an expected call of GetAliveWorkers.
func (mr *MockWorkerDaoMockRecorder) GetAliveWorkers() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAliveWorkers", reflect.TypeOf((*MockWorkerDao)(nil).GetAliveWorkers))
}
// GetAllWorkers mocks base method.
func (m *MockWorkerDao) GetAllWorkers() ([]model.Worker, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetAllWorkers")
ret0, _ := ret[0].([]model.Worker)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetAllWorkers indicates an expected call of GetAllWorkers.
func (mr *MockWorkerDaoMockRecorder) GetAllWorkers() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllWorkers", reflect.TypeOf((*MockWorkerDao)(nil).GetAllWorkers))
}
// GetWorkerByHostname mocks base method.
func (m *MockWorkerDao) GetWorkerByHostname(ctx context.Context, hostname string) (model.Worker, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetWorkerByHostname", ctx, hostname)
ret0, _ := ret[0].(model.Worker)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetWorkerByHostname indicates an expected call of GetWorkerByHostname.
func (mr *MockWorkerDaoMockRecorder) GetWorkerByHostname(ctx, hostname interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerByHostname", reflect.TypeOf((*MockWorkerDao)(nil).GetWorkerByHostname), ctx, hostname)
}
// GetWorkerByID mocks base method.
func (m *MockWorkerDao) GetWorkerByID(ctx context.Context, workerID string) (model.Worker, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetWorkerByID", ctx, workerID)
ret0, _ := ret[0].(model.Worker)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetWorkerByID indicates an expected call of GetWorkerByID.
func (mr *MockWorkerDaoMockRecorder) GetWorkerByID(ctx, workerID interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerByID", reflect.TypeOf((*MockWorkerDao)(nil).GetWorkerByID), ctx, workerID)
}
// SaveWorker mocks base method.
func (m *MockWorkerDao) SaveWorker(worker model.Worker) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "SaveWorker", worker)
ret0, _ := ret[0].(error)
return ret0
}
// SaveWorker indicates an expected call of SaveWorker.
func (mr *MockWorkerDaoMockRecorder) SaveWorker(worker interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveWorker", reflect.TypeOf((*MockWorkerDao)(nil).SaveWorker), worker)
}