consul/agent/dns/mock_DNSRouter.go

// Code generated by mockery v2.37.1. DO NOT EDIT.

package dns

import (
	config "github.com/hashicorp/consul/agent/config"
	discovery "github.com/hashicorp/consul/agent/discovery"

	miekgdns "github.com/miekg/dns"

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

	net "net"
)

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

// HandleRequest provides a mock function with given fields: req, reqCtx, remoteAddress
func (_m *MockDNSRouter) HandleRequest(req *miekgdns.Msg, reqCtx discovery.Context, remoteAddress net.Addr) *miekgdns.Msg {
	ret := _m.Called(req, reqCtx, remoteAddress)

	var r0 *miekgdns.Msg
	if rf, ok := ret.Get(0).(func(*miekgdns.Msg, discovery.Context, net.Addr) *miekgdns.Msg); ok {
		r0 = rf(req, reqCtx, remoteAddress)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*miekgdns.Msg)
		}
	}

	return r0
}

// ReloadConfig provides a mock function with given fields: newCfg
func (_m *MockDNSRouter) ReloadConfig(newCfg *config.RuntimeConfig) error {
	ret := _m.Called(newCfg)

	var r0 error
	if rf, ok := ret.Get(0).(func(*config.RuntimeConfig) error); ok {
		r0 = rf(newCfg)
	} else {
		r0 = ret.Error(0)
	}

	return r0
}

// ServeDNS provides a mock function with given fields: w, req
func (_m *MockDNSRouter) ServeDNS(w miekgdns.ResponseWriter, req *miekgdns.Msg) {
	_m.Called(w, req)
}

// NewMockDNSRouter creates a new instance of MockDNSRouter. 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 NewMockDNSRouter(t interface {
	mock.TestingT
	Cleanup(func())
}) *MockDNSRouter {
	mock := &MockDNSRouter{}
	mock.Mock.Test(t)

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

	return mock
}