AdGuardHome/internal/aghtest/exchanger.go

package aghtest

import (
	"github.com/AdguardTeam/dnsproxy/upstream"
	"github.com/miekg/dns"
)

// Exchanger is a mock aghnet.Exchanger implementation for tests.
type Exchanger struct {
	Ups upstream.Upstream
}

// Exchange implements aghnet.Exchanger interface for *Exchanger.
func (e *Exchanger) Exchange(req *dns.Msg) (resp *dns.Msg, err error) {
	if e.Ups == nil {
		e.Ups = &TestErrUpstream{}
	}

	return e.Ups.Exchange(req)
}