9d2301cae2
This diff has been extracted from c2f7ccab0e
See https://github.com/ooni/probe/issues/2096
While there, export DecodeReply to decode a raw reply without
interpreting the Rcode or parsing the results, which seems a
nice extra feature to have to more flexibly parse DNS replies
in other parts of the codebase.
31 lines
914 B
Go
31 lines
914 B
Go
package mocks
|
|
|
|
import (
|
|
"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)
|
|
|
|
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)
|
|
}
|
|
|
|
// DecodeReply calls MockDecodeReply.
|
|
func (e *DNSDecoder) DecodeReply(reply []byte) (*dns.Msg, error) {
|
|
return e.MockDecodeReply(reply)
|
|
}
|