moira

1
// Code generated by MockGen. DO NOT EDIT.
2
// Source: github.com/moira-alert/moira/notifier (interfaces: Scheduler)
3

4
// Package mock_moira_alert is a generated GoMock package.
5
package mock_scheduler
6

7
import (
8
	gomock "github.com/golang/mock/gomock"
9
	moira "go.avito.ru/DO/moira"
10
	reflect "reflect"
11
	time "time"
12
)
13

14
// MockScheduler is a mock of Scheduler interface
15
type MockScheduler struct {
16
	ctrl     *gomock.Controller
17
	recorder *MockSchedulerMockRecorder
18
}
19

20
// MockSchedulerMockRecorder is the mock recorder for MockScheduler
21
type MockSchedulerMockRecorder struct {
22
	mock *MockScheduler
23
}
24

25
// NewMockScheduler creates a new mock instance
26
func NewMockScheduler(ctrl *gomock.Controller) *MockScheduler {
27
	mock := &MockScheduler{ctrl: ctrl}
28
	mock.recorder = &MockSchedulerMockRecorder{mock}
29
	return mock
30
}
31

32
// EXPECT returns an object that allows the caller to indicate expected use
33
func (m *MockScheduler) EXPECT() *MockSchedulerMockRecorder {
34
	return m.recorder
35
}
36

37
// CalculateBackoff mocks base method
38
func (m *MockScheduler) CalculateBackoff(arg0 int) time.Duration {
39
	ret := m.ctrl.Call(m, "CalculateBackoff", arg0)
40
	ret0, _ := ret[0].(time.Duration)
41
	return ret0
42
}
43

44
// CalculateBackoff indicates an expected call of CalculateBackoff
45
func (mr *MockSchedulerMockRecorder) CalculateBackoff(arg0 interface{}) *gomock.Call {
46
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CalculateBackoff", reflect.TypeOf((*MockScheduler)(nil).CalculateBackoff), arg0)
47
}
48

49
// GetDeliveryInfo mocks base method
50
func (m *MockScheduler) GetDeliveryInfo(arg0 time.Time, arg1 moira.NotificationEvent, arg2 bool, arg3 int) (time.Time, bool) {
51
	ret := m.ctrl.Call(m, "GetDeliveryInfo", arg0, arg1, arg2, arg3)
52
	ret0, _ := ret[0].(time.Time)
53
	ret1, _ := ret[1].(bool)
54
	return ret0, ret1
55
}
56

57
// GetDeliveryInfo indicates an expected call of GetDeliveryInfo
58
func (mr *MockSchedulerMockRecorder) GetDeliveryInfo(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
59
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeliveryInfo", reflect.TypeOf((*MockScheduler)(nil).GetDeliveryInfo), arg0, arg1, arg2, arg3)
60
}
61

62
// ScheduleNotification mocks base method
63
func (m *MockScheduler) ScheduleNotification(arg0 time.Time, arg1 bool, arg2 moira.NotificationEvent, arg3 moira.TriggerData, arg4 moira.ContactData, arg5 int, arg6 bool) *moira.ScheduledNotification {
64
	ret := m.ctrl.Call(m, "ScheduleNotification", arg0, arg1, arg2, arg3, arg4, arg5, arg6)
65
	ret0, _ := ret[0].(*moira.ScheduledNotification)
66
	return ret0
67
}
68

69
// ScheduleNotification indicates an expected call of ScheduleNotification
70
func (mr *MockSchedulerMockRecorder) ScheduleNotification(arg0, arg1, arg2, arg3, arg4, arg5, arg6 interface{}) *gomock.Call {
71
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleNotification", reflect.TypeOf((*MockScheduler)(nil).ScheduleNotification), arg0, arg1, arg2, arg3, arg4, arg5, arg6)
72
}
73