ce052b665e
This diff has been extracted from eb0bf38957.
See https://github.com/ooni/probe/issues/2096.
While there, skip the broken tests caused by issue
https://github.com/ooni/probe/issues/2098.
37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
package mocks
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/miekg/dns"
|
|
"github.com/ooni/probe-cli/v3/internal/model"
|
|
)
|
|
|
|
// DNSDecoder allows mocking dnsx.DNSDecoder.
|
|
type DNSDecoder struct {
|
|
MockDecodeLookupHost func(qtype uint16, reply []byte, queryID uint16) ([]string, error)
|
|
MockDecodeHTTPS func(reply []byte, queryID uint16) (*model.HTTPSSvc, error)
|
|
MockDecodeNS func(reply []byte, queryID uint16) ([]*net.NS, error)
|
|
MockDecodeReply func(reply []byte) (*dns.Msg, error)
|
|
}
|
|
|
|
// DecodeLookupHost calls MockDecodeLookupHost.
|
|
func (e *DNSDecoder) DecodeLookupHost(qtype uint16, reply []byte, queryID uint16) ([]string, error) {
|
|
return e.MockDecodeLookupHost(qtype, reply, queryID)
|
|
}
|
|
|
|
// DecodeHTTPS calls MockDecodeHTTPS.
|
|
func (e *DNSDecoder) DecodeHTTPS(reply []byte, queryID uint16) (*model.HTTPSSvc, error) {
|
|
return e.MockDecodeHTTPS(reply, queryID)
|
|
}
|
|
|
|
// DecodeNS calls MockDecodeNS.
|
|
func (e *DNSDecoder) DecodeNS(reply []byte, queryID uint16) ([]*net.NS, error) {
|
|
return e.MockDecodeNS(reply, queryID)
|
|
}
|
|
|
|
// DecodeReply calls MockDecodeReply.
|
|
func (e *DNSDecoder) DecodeReply(reply []byte) (*dns.Msg, error) {
|
|
return e.MockDecodeReply(reply)
|
|
}
|