fix: mock DNS in resolver tests for hermetic, fast unit tests
- Extract DNSClient interface from resolver to allow dependency injection - Convert all resolver methods from package-level to receiver methods using the injectable DNS client - Rewrite resolver_test.go with a mock DNS client that simulates the full delegation chain (root → TLD → authoritative) in-process - Move 2 integration tests (real DNS) behind //go:build integration tag - Add NewFromLoggerWithClient constructor for test injection - Add LookupAllRecords implementation (was returning ErrNotImplemented) All unit tests are hermetic (no network) and complete in <1s. Total make check passes in ~5s. Closes #12
This commit is contained in:
48
internal/resolver/dns_client.go
Normal file
48
internal/resolver/dns_client.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package resolver
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
// DNSClient abstracts DNS wire-protocol exchanges so the resolver
|
||||
// can be tested without hitting real nameservers.
|
||||
type DNSClient interface {
|
||||
ExchangeContext(
|
||||
ctx context.Context,
|
||||
msg *dns.Msg,
|
||||
addr string,
|
||||
) (*dns.Msg, time.Duration, error)
|
||||
}
|
||||
|
||||
// udpClient wraps a real dns.Client for production use.
|
||||
type udpClient struct {
|
||||
timeout time.Duration
|
||||
}
|
||||
|
||||
func (c *udpClient) ExchangeContext(
|
||||
ctx context.Context,
|
||||
msg *dns.Msg,
|
||||
addr string,
|
||||
) (*dns.Msg, time.Duration, error) {
|
||||
cl := &dns.Client{Timeout: c.timeout}
|
||||
|
||||
return cl.ExchangeContext(ctx, msg, addr)
|
||||
}
|
||||
|
||||
// tcpClient wraps a real dns.Client using TCP.
|
||||
type tcpClient struct {
|
||||
timeout time.Duration
|
||||
}
|
||||
|
||||
func (c *tcpClient) ExchangeContext(
|
||||
ctx context.Context,
|
||||
msg *dns.Msg,
|
||||
addr string,
|
||||
) (*dns.Msg, time.Duration, error) {
|
||||
cl := &dns.Client{Net: "tcp", Timeout: c.timeout}
|
||||
|
||||
return cl.ExchangeContext(ctx, msg, addr)
|
||||
}
|
||||
Reference in New Issue
Block a user