forked from eremetic-framework/eremetic
/
mock.go
100 lines (80 loc) · 2.45 KB
/
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 mock
import (
"errors"
"github.com/klarna/eremetic"
)
// Scheduler mocks the eremetic scheduler.
type Scheduler struct {
ScheduleTaskFn func(req eremetic.Request) (string, error)
ScheduleTaskInvoked bool
KillFn func(id string) error
KillInvoked bool
}
// ScheduleTask invokes the ScheduleTaskFn function.
func (s *Scheduler) ScheduleTask(req eremetic.Request) (string, error) {
s.ScheduleTaskInvoked = true
return s.ScheduleTaskFn(req)
}
func (s *Scheduler) Kill(id string) error {
s.KillInvoked = true
return s.KillFn(id)
}
// TaskDB mocks the eremetic task database.
type TaskDB struct {
CleanFn func() error
CloseFn func()
PutTaskFn func(*eremetic.Task) error
ReadTaskFn func(string) (eremetic.Task, error)
ReadUnmaskedTaskFn func(string) (eremetic.Task, error)
DeleteTaskFn func(string) error
ListNonTerminalTasksFn func() ([]*eremetic.Task, error)
}
// Clean invokes the CleanFn function.
func (db *TaskDB) Clean() error {
return db.CleanFn()
}
// Close invokes the CloseFn function.
func (db *TaskDB) Close() {
db.CloseFn()
}
// PutTask invokes the PutTaskFn function.
func (db *TaskDB) PutTask(task *eremetic.Task) error {
return db.PutTaskFn(task)
}
// ReadTask invokes the ReadTaskFn function.
func (db *TaskDB) ReadTask(id string) (eremetic.Task, error) {
return db.ReadTaskFn(id)
}
// ReadUnmaskedTask invokes the ReadUnmaskedTaskFn function.
func (db *TaskDB) ReadUnmaskedTask(id string) (eremetic.Task, error) {
return db.ReadUnmaskedTaskFn(id)
}
// ReadUnmaskedTask invokes the ReadUnmaskedTaskFn function.
func (db *TaskDB) DeleteTask(id string) error {
return db.DeleteTaskFn(id)
}
// ListNonTerminalTasks invokes the ListNonTerminalTasksFn function.
func (db *TaskDB) ListNonTerminalTasks() ([]*eremetic.Task, error) {
return db.ListNonTerminalTasksFn()
}
// ErrScheduler mocks the eremetic scheduler.
type ErrScheduler struct {
NextError *error
}
// ScheduleTask records any scheduling errors.
func (s *ErrScheduler) ScheduleTask(request eremetic.Request) (string, error) {
if err := s.NextError; err != nil {
s.NextError = nil
return "", *err
}
return "eremetic-task.mock", nil
}
func (s *ErrScheduler) Kill(_id string) error {
return nil
}
// ErrorReader simulates a failure to read stream.
type ErrorReader struct{}
// Read always returns an error.
func (r *ErrorReader) Read(p []byte) (int, error) {
return 0, errors.New("oh no")
}