github.com/google/syzkaller@v0.0.0-20251211124644-a066d2bc4b02/pkg/rpcserver/mocks/Manager.go

// Code generated by mockery; DO NOT EDIT.
// github.com/vektra/mockery
// template: testify

package mocks

import (
	"github.com/google/syzkaller/pkg/flatrpc"
	"github.com/google/syzkaller/pkg/fuzzer/queue"
	"github.com/google/syzkaller/pkg/signal"
	"github.com/google/syzkaller/pkg/vminfo"
	"github.com/google/syzkaller/prog"
	mock "github.com/stretchr/testify/mock"
)

// NewManager creates a new instance of Manager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewManager(t interface {
	mock.TestingT
	Cleanup(func())
}) *Manager {
	mock := &Manager{}
	mock.Mock.Test(t)

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

	return mock
}

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

type Manager_Expecter struct {
	mock *mock.Mock
}

func (_m *Manager) EXPECT() *Manager_Expecter {
	return &Manager_Expecter{mock: &_m.Mock}
}

// BugFrames provides a mock function for the type Manager
func (_mock *Manager) BugFrames() ([]string, []string) {
	ret := _mock.Called()

	if len(ret) == 0 {
		panic("no return value specified for BugFrames")
	}

	var r0 []string
	var r1 []string
	if returnFunc, ok := ret.Get(0).(func() ([]string, []string)); ok {
		return returnFunc()
	}
	if returnFunc, ok := ret.Get(0).(func() []string); ok {
		r0 = returnFunc()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([]string)
		}
	}
	if returnFunc, ok := ret.Get(1).(func() []string); ok {
		r1 = returnFunc()
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).([]string)
		}
	}
	return r0, r1
}

// Manager_BugFrames_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BugFrames'
type Manager_BugFrames_Call struct {
	*mock.Call
}

// BugFrames is a helper method to define mock.On call
func (_e *Manager_Expecter) BugFrames() *Manager_BugFrames_Call {
	return &Manager_BugFrames_Call{Call: _e.mock.On("BugFrames")}
}

func (_c *Manager_BugFrames_Call) Run(run func()) *Manager_BugFrames_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Manager_BugFrames_Call) Return(leaks []string, races []string) *Manager_BugFrames_Call {
	_c.Call.Return(leaks, races)
	return _c
}

func (_c *Manager_BugFrames_Call) RunAndReturn(run func() ([]string, []string)) *Manager_BugFrames_Call {
	_c.Call.Return(run)
	return _c
}

// CoverageFilter provides a mock function for the type Manager
func (_mock *Manager) CoverageFilter(modules []*vminfo.KernelModule) ([]uint64, error) {
	ret := _mock.Called(modules)

	if len(ret) == 0 {
		panic("no return value specified for CoverageFilter")
	}

	var r0 []uint64
	var r1 error
	if returnFunc, ok := ret.Get(0).(func([]*vminfo.KernelModule) ([]uint64, error)); ok {
		return returnFunc(modules)
	}
	if returnFunc, ok := ret.Get(0).(func([]*vminfo.KernelModule) []uint64); ok {
		r0 = returnFunc(modules)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([]uint64)
		}
	}
	if returnFunc, ok := ret.Get(1).(func([]*vminfo.KernelModule) error); ok {
		r1 = returnFunc(modules)
	} else {
		r1 = ret.Error(1)
	}
	return r0, r1
}

// Manager_CoverageFilter_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CoverageFilter'
type Manager_CoverageFilter_Call struct {
	*mock.Call
}

// CoverageFilter is a helper method to define mock.On call
//   - modules []*vminfo.KernelModule
func (_e *Manager_Expecter) CoverageFilter(modules interface{}) *Manager_CoverageFilter_Call {
	return &Manager_CoverageFilter_Call{Call: _e.mock.On("CoverageFilter", modules)}
}

func (_c *Manager_CoverageFilter_Call) Run(run func(modules []*vminfo.KernelModule)) *Manager_CoverageFilter_Call {
	_c.Call.Run(func(args mock.Arguments) {
		var arg0 []*vminfo.KernelModule
		if args[0] != nil {
			arg0 = args[0].([]*vminfo.KernelModule)
		}
		run(
			arg0,
		)
	})
	return _c
}

func (_c *Manager_CoverageFilter_Call) Return(uint64s []uint64, err error) *Manager_CoverageFilter_Call {
	_c.Call.Return(uint64s, err)
	return _c
}

func (_c *Manager_CoverageFilter_Call) RunAndReturn(run func(modules []*vminfo.KernelModule) ([]uint64, error)) *Manager_CoverageFilter_Call {
	_c.Call.Return(run)
	return _c
}

// MachineChecked provides a mock function for the type Manager
func (_mock *Manager) MachineChecked(features flatrpc.Feature, syscalls map[*prog.Syscall]bool) (queue.Source, error) {
	ret := _mock.Called(features, syscalls)

	if len(ret) == 0 {
		panic("no return value specified for MachineChecked")
	}

	var r0 queue.Source
	var r1 error
	if returnFunc, ok := ret.Get(0).(func(flatrpc.Feature, map[*prog.Syscall]bool) (queue.Source, error)); ok {
		return returnFunc(features, syscalls)
	}
	if returnFunc, ok := ret.Get(0).(func(flatrpc.Feature, map[*prog.Syscall]bool) queue.Source); ok {
		r0 = returnFunc(features, syscalls)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(queue.Source)
		}
	}
	if returnFunc, ok := ret.Get(1).(func(flatrpc.Feature, map[*prog.Syscall]bool) error); ok {
		r1 = returnFunc(features, syscalls)
	} else {
		r1 = ret.Error(1)
	}
	return r0, r1
}

// Manager_MachineChecked_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MachineChecked'
type Manager_MachineChecked_Call struct {
	*mock.Call
}

// MachineChecked is a helper method to define mock.On call
//   - features flatrpc.Feature
//   - syscalls map[*prog.Syscall]bool
func (_e *Manager_Expecter) MachineChecked(features interface{}, syscalls interface{}) *Manager_MachineChecked_Call {
	return &Manager_MachineChecked_Call{Call: _e.mock.On("MachineChecked", features, syscalls)}
}

func (_c *Manager_MachineChecked_Call) Run(run func(features flatrpc.Feature, syscalls map[*prog.Syscall]bool)) *Manager_MachineChecked_Call {
	_c.Call.Run(func(args mock.Arguments) {
		var arg0 flatrpc.Feature
		if args[0] != nil {
			arg0 = args[0].(flatrpc.Feature)
		}
		var arg1 map[*prog.Syscall]bool
		if args[1] != nil {
			arg1 = args[1].(map[*prog.Syscall]bool)
		}
		run(
			arg0,
			arg1,
		)
	})
	return _c
}

func (_c *Manager_MachineChecked_Call) Return(source queue.Source, err error) *Manager_MachineChecked_Call {
	_c.Call.Return(source, err)
	return _c
}

func (_c *Manager_MachineChecked_Call) RunAndReturn(run func(features flatrpc.Feature, syscalls map[*prog.Syscall]bool) (queue.Source, error)) *Manager_MachineChecked_Call {
	_c.Call.Return(run)
	return _c
}

// MaxSignal provides a mock function for the type Manager
func (_mock *Manager) MaxSignal() signal.Signal {
	ret := _mock.Called()

	if len(ret) == 0 {
		panic("no return value specified for MaxSignal")
	}

	var r0 signal.Signal
	if returnFunc, ok := ret.Get(0).(func() signal.Signal); ok {
		r0 = returnFunc()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(signal.Signal)
		}
	}
	return r0
}

// Manager_MaxSignal_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MaxSignal'
type Manager_MaxSignal_Call struct {
	*mock.Call
}

// MaxSignal is a helper method to define mock.On call
func (_e *Manager_Expecter) MaxSignal() *Manager_MaxSignal_Call {
	return &Manager_MaxSignal_Call{Call: _e.mock.On("MaxSignal")}
}

func (_c *Manager_MaxSignal_Call) Run(run func()) *Manager_MaxSignal_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Manager_MaxSignal_Call) Return(signal1 signal.Signal) *Manager_MaxSignal_Call {
	_c.Call.Return(signal1)
	return _c
}

func (_c *Manager_MaxSignal_Call) RunAndReturn(run func() signal.Signal) *Manager_MaxSignal_Call {
	_c.Call.Return(run)
	return _c
}