dbd1c047cb
* Run gostaticheck Run gostaticcheck on the codebase and fix almost all flagged items. Only keep * coremain/run.go:192:2: var appVersion is unused (U1000) * plugin/chaos/setup.go:54:3: the surrounding loop is unconditionally terminated (SA4004) * plugin/etcd/setup.go:103:3: the surrounding loop is unconditionally terminated (SA4004) * plugin/pkg/replacer/replacer.go:274:13: argument should be pointer-like to avoid allocations (SA6002) * plugin/route53/setup.go:124:28: session.New is deprecated: Use NewSession functions to create sessions instead. NewSession has the same functionality as New except an error can be returned when the func is called instead of waiting to receive an error until a request is made. (SA1019) * test/grpc_test.go:25:69: grpc.WithTimeout is deprecated: use DialContext and context.WithTimeout instead. Will be supported throughout 1.x. (SA1019) The first one isn't true, as this is set via ldflags. The rest is minor. The deprecation should be fixed at some point; I'll file some issues. Signed-off-by: Miek Gieben <miek@miek.nl> * Make sure to plug in the plugins import the plugins, that file that did this was removed, put it in the reload test as this requires an almost complete coredns server. Signed-off-by: Miek Gieben <miek@miek.nl>
77 lines
2 KiB
Go
77 lines
2 KiB
Go
package test
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
// Hook in CoreDNS.
|
|
_ "github.com/coredns/coredns/core"
|
|
// Load all managed plugins in github.com/coredns/coredns
|
|
_ "github.com/coredns/coredns/core/plugin"
|
|
|
|
"github.com/caddyserver/caddy"
|
|
)
|
|
|
|
var mu sync.Mutex
|
|
|
|
// CoreDNSServer returns a CoreDNS test server. It just takes a normal Corefile as input.
|
|
func CoreDNSServer(corefile string) (*caddy.Instance, error) {
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
caddy.Quiet = true
|
|
dnsserver.Quiet = true
|
|
|
|
return caddy.Start(NewInput(corefile))
|
|
}
|
|
|
|
// CoreDNSServerStop stops a server.
|
|
func CoreDNSServerStop(i *caddy.Instance) { i.Stop() }
|
|
|
|
// CoreDNSServerPorts returns the ports the instance is listening on. The integer k indicates
|
|
// which ServerListener you want.
|
|
func CoreDNSServerPorts(i *caddy.Instance, k int) (udp, tcp string) {
|
|
srvs := i.Servers()
|
|
if len(srvs) < k+1 {
|
|
return "", ""
|
|
}
|
|
u := srvs[k].LocalAddr()
|
|
t := srvs[k].Addr()
|
|
|
|
if u != nil {
|
|
udp = u.String()
|
|
}
|
|
if t != nil {
|
|
tcp = t.String()
|
|
}
|
|
return
|
|
}
|
|
|
|
// CoreDNSServerAndPorts combines CoreDNSServer and CoreDNSServerPorts to start a CoreDNS
|
|
// server and returns the udp and tcp ports of the first instance.
|
|
func CoreDNSServerAndPorts(corefile string) (i *caddy.Instance, udp, tcp string, err error) {
|
|
i, err = CoreDNSServer(corefile)
|
|
if err != nil {
|
|
return nil, "", "", err
|
|
}
|
|
udp, tcp = CoreDNSServerPorts(i, 0)
|
|
return i, udp, tcp, nil
|
|
}
|
|
|
|
// Input implements the caddy.Input interface and acts as an easy way to use a string as a Corefile.
|
|
type Input struct {
|
|
corefile []byte
|
|
}
|
|
|
|
// NewInput returns a pointer to Input, containing the corefile string as input.
|
|
func NewInput(corefile string) *Input {
|
|
return &Input{corefile: []byte(corefile)}
|
|
}
|
|
|
|
// Body implements the Input interface.
|
|
func (i *Input) Body() []byte { return i.corefile }
|
|
|
|
// Path implements the Input interface.
|
|
func (i *Input) Path() string { return "Corefile" }
|
|
|
|
// ServerType implements the Input interface.
|
|
func (i *Input) ServerType() string { return "dns" }
|