Live: RunStream improvements – backoff, better layout, tests (#33029)

pkg/services/live/runstream/mock.go

@@ -0,0 +1,125 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: github.com/grafana/grafana/pkg/services/live/runstream (interfaces: StreamPacketSender,PresenceGetter,StreamRunner)
+
+// Package runstream is a generated GoMock package.
+package runstream
+
+import (
+	context "context"
+	reflect "reflect"
+
+	gomock "github.com/golang/mock/gomock"
+	backend "github.com/grafana/grafana-plugin-sdk-go/backend"
+)
+
+// MockStreamPacketSender is a mock of StreamPacketSender interface.
+type MockStreamPacketSender struct {
+	ctrl     *gomock.Controller
+	recorder *MockStreamPacketSenderMockRecorder
+}
+
+// MockStreamPacketSenderMockRecorder is the mock recorder for MockStreamPacketSender.
+type MockStreamPacketSenderMockRecorder struct {
+	mock *MockStreamPacketSender
+}
+
+// NewMockStreamPacketSender creates a new mock instance.
+func NewMockStreamPacketSender(ctrl *gomock.Controller) *MockStreamPacketSender {
+	mock := &MockStreamPacketSender{ctrl: ctrl}
+	mock.recorder = &MockStreamPacketSenderMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockStreamPacketSender) EXPECT() *MockStreamPacketSenderMockRecorder {
+	return m.recorder
+}
+
+// Send mocks base method.
+func (m *MockStreamPacketSender) Send(arg0 string, arg1 *backend.StreamPacket) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Send", arg0, arg1)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Send indicates an expected call of Send.
+func (mr *MockStreamPacketSenderMockRecorder) Send(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockStreamPacketSender)(nil).Send), arg0, arg1)
+}
+
+// MockPresenceGetter is a mock of PresenceGetter interface.
+type MockPresenceGetter struct {
+	ctrl     *gomock.Controller
+	recorder *MockPresenceGetterMockRecorder
+}
+
+// MockPresenceGetterMockRecorder is the mock recorder for MockPresenceGetter.
+type MockPresenceGetterMockRecorder struct {
+	mock *MockPresenceGetter
+}
+
+// NewMockPresenceGetter creates a new mock instance.
+func NewMockPresenceGetter(ctrl *gomock.Controller) *MockPresenceGetter {
+	mock := &MockPresenceGetter{ctrl: ctrl}
+	mock.recorder = &MockPresenceGetterMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockPresenceGetter) EXPECT() *MockPresenceGetterMockRecorder {
+	return m.recorder
+}
+
+// GetNumSubscribers mocks base method.
+func (m *MockPresenceGetter) GetNumSubscribers(arg0 string) (int, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetNumSubscribers", arg0)
+	ret0, _ := ret[0].(int)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetNumSubscribers indicates an expected call of GetNumSubscribers.
+func (mr *MockPresenceGetterMockRecorder) GetNumSubscribers(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNumSubscribers", reflect.TypeOf((*MockPresenceGetter)(nil).GetNumSubscribers), arg0)
+}
+
+// MockStreamRunner is a mock of StreamRunner interface.
+type MockStreamRunner struct {
+	ctrl     *gomock.Controller
+	recorder *MockStreamRunnerMockRecorder
+}
+
+// MockStreamRunnerMockRecorder is the mock recorder for MockStreamRunner.
+type MockStreamRunnerMockRecorder struct {
+	mock *MockStreamRunner
+}
+
+// NewMockStreamRunner creates a new mock instance.
+func NewMockStreamRunner(ctrl *gomock.Controller) *MockStreamRunner {
+	mock := &MockStreamRunner{ctrl: ctrl}
+	mock.recorder = &MockStreamRunnerMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockStreamRunner) EXPECT() *MockStreamRunnerMockRecorder {
+	return m.recorder
+}
+
+// RunStream mocks base method.
+func (m *MockStreamRunner) RunStream(arg0 context.Context, arg1 *backend.RunStreamRequest, arg2 backend.StreamPacketSender) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "RunStream", arg0, arg1, arg2)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// RunStream indicates an expected call of RunStream.
+func (mr *MockStreamRunnerMockRecorder) RunStream(arg0, arg1, arg2 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunStream", reflect.TypeOf((*MockStreamRunner)(nil).RunStream), arg0, arg1, arg2)
+}