21 lines
437 B
Go
21 lines
437 B
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 (lr *Exchanger) Exchange(req *dns.Msg) (resp *dns.Msg, err error) {
|
||
|
|
if lr.Ups == nil {
|
||
|
|
lr.Ups = &TestErrUpstream{}
|
||
|
|
}
|
||
|
|
|
||
|
|
return lr.Ups.Exchange(req)
|
||
|
|
}
|