observer_mock.go

package observer

import (
	"github.com/reactivex/rxgo/handlers"
	"github.com/stretchr/testify/mock"
)

func NewObserverMock() *ObserverMock {
	obMock := new(ObserverMock)
	obMock.On("OnDone").Return()
	obMock.On("OnNext", mock.Anything).Return()
	obMock.On("OnError", mock.Anything).Return()
	return obMock
}

type ObserverMock struct {
	mock.Mock
}

// OnDone provides a mock function with given fields:
func (m *ObserverMock) OnDone() {
	m.Called()
}

// OnError provides a mock function with given fields: err
func (m *ObserverMock) OnError(err error) {
	m.Called(err)
}

// OnNext provides a mock function with given fields: item
func (m *ObserverMock) OnNext(item interface{}) {
	m.Called(item)
}

func (m *ObserverMock) Capture() Observer {
	ob := New(
		handlers.NextFunc(func(el interface{}) {
			m.OnNext(el)
		}),
		handlers.ErrFunc(func(err error) {
			m.OnError(err)
		}),
		handlers.DoneFunc(func() {
			m.OnDone()
		}),
	)
	return ob
}