go/sser/mocks/Server.go

// Code generated by mockery v0.0.0-dev. DO NOT EDIT.

package mocks

import (
	context "context"
	http "net/http"

	mock "github.com/stretchr/testify/mock"

	testing "testing"
)

// Server is an autogenerated mock type for the Server type
type Server struct {
	mock.Mock
}

// ClientConnectionHandler provides a mock function with given fields: ctx
func (_m *Server) ClientConnectionHandler(ctx context.Context) http.HandlerFunc {
	ret := _m.Called(ctx)

	var r0 http.HandlerFunc
	if rf, ok := ret.Get(0).(func(context.Context) http.HandlerFunc); ok {
		r0 = rf(ctx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(http.HandlerFunc)
		}
	}

	return r0
}

// Send provides a mock function with given fields: ctx, stream, msg
func (_m *Server) Send(ctx context.Context, stream string, msg string) error {
	ret := _m.Called(ctx, stream, msg)

	var r0 error
	if rf, ok := ret.Get(0).(func(context.Context, string, string) error); ok {
		r0 = rf(ctx, stream, msg)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}

// Start provides a mock function with given fields: ctx
func (_m *Server) Start(ctx context.Context) error {
	ret := _m.Called(ctx)

	var r0 error
	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
		r0 = rf(ctx)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}

// NewServer creates a new instance of Server. It also registers a cleanup function to assert the mocks expectations.
func NewServer(t testing.TB) *Server {
	mock := &Server{}

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}