github.com/yankunsam/loki/v2@v2.6.3-0.20220817130409-389df5235c27/pkg/ruler/base/pusher_mock_test.go (about)

     1  package base
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/stretchr/testify/mock"
     7  
     8  	"github.com/grafana/loki/pkg/logproto"
     9  )
    10  
    11  type pusherMock struct {
    12  	mock.Mock
    13  }
    14  
    15  func newPusherMock() *pusherMock {
    16  	return &pusherMock{}
    17  }
    18  
    19  func (m *pusherMock) Push(ctx context.Context, req *logproto.WriteRequest) (*logproto.WriteResponse, error) {
    20  	args := m.Called(ctx, req)
    21  	return args.Get(0).(*logproto.WriteResponse), args.Error(1)
    22  }
    23  
    24  func (m *pusherMock) MockPush(res *logproto.WriteResponse, err error) {
    25  	m.On("Push", mock.Anything, mock.Anything).Return(res, err)
    26  }