mirror of
https://github.com/coredns/coredns.git
synced 2025-11-02 18:23:25 -05:00
test: add t.Helper() calls to test helper functions (#7351)
This commit is contained in:
@@ -62,6 +62,7 @@ func TestWalkNonExistent(t *testing.T) {
|
||||
}
|
||||
|
||||
func createFiles(t *testing.T) (string, error) {
|
||||
t.Helper()
|
||||
dir := t.TempDir()
|
||||
|
||||
for _, name := range dbFiles {
|
||||
|
||||
1
plugin/cache/freq/freq_test.go
vendored
1
plugin/cache/freq/freq_test.go
vendored
@@ -30,6 +30,7 @@ func TestReset(t *testing.T) {
|
||||
}
|
||||
|
||||
func hitsCheck(t *testing.T, f *Freq, expected int) {
|
||||
t.Helper()
|
||||
if x := f.Hits(); x != expected {
|
||||
t.Fatalf("Expected hits to be %d, got %d", expected, x)
|
||||
}
|
||||
|
||||
@@ -248,6 +248,7 @@ func testEmptyMsg() *dns.Msg {
|
||||
}
|
||||
|
||||
func newDnssec(t *testing.T, zones []string) (Dnssec, func(), func()) {
|
||||
t.Helper()
|
||||
k, rm1, rm2 := newKey(t)
|
||||
c := cache.New(defaultCap)
|
||||
d := New(zones, []*DNSKEY{k}, false, nil, c)
|
||||
@@ -255,6 +256,7 @@ func newDnssec(t *testing.T, zones []string) (Dnssec, func(), func()) {
|
||||
}
|
||||
|
||||
func newKey(t *testing.T) (*DNSKEY, func(), func()) {
|
||||
t.Helper()
|
||||
fPriv, rmPriv, _ := test.TempFile(".", privKey)
|
||||
fPub, rmPub, _ := test.TempFile(".", pubKey)
|
||||
|
||||
|
||||
@@ -14,6 +14,7 @@ import (
|
||||
)
|
||||
|
||||
func testCase(t *testing.T, tapq, tapr *tap.Dnstap, q, r *dns.Msg, extraFormat string) {
|
||||
t.Helper()
|
||||
w := writer{t: t}
|
||||
w.queue = append(w.queue, tapq, tapr)
|
||||
h := Dnstap{
|
||||
|
||||
@@ -18,6 +18,7 @@ var (
|
||||
)
|
||||
|
||||
func accept(t *testing.T, l net.Listener, count int) {
|
||||
t.Helper()
|
||||
server, err := l.Accept()
|
||||
if err != nil {
|
||||
t.Fatalf("Server accepted: %s", err)
|
||||
|
||||
@@ -170,6 +170,7 @@ func TestLookupSecureDelegation(t *testing.T) {
|
||||
}
|
||||
|
||||
func testDelegation(t *testing.T, z, origin string, testcases []test.Case) {
|
||||
t.Helper()
|
||||
zone, err := Parse(strings.NewReader(z), origin, "stdin", 0)
|
||||
if err != nil {
|
||||
t.Fatalf("Expect no error when reading zone, got %q", err)
|
||||
|
||||
@@ -71,6 +71,7 @@ func TestMetadata(t *testing.T) {
|
||||
}
|
||||
|
||||
func testMetadata(t *testing.T, state request.Request, geoIP *GeoIP, label, expectedValue string) {
|
||||
t.Helper()
|
||||
ctx := metadata.ContextWithMetadata(context.Background())
|
||||
rCtx := geoIP.Metadata(ctx, state)
|
||||
if fmt.Sprintf("%p", ctx) != fmt.Sprintf("%p", rCtx) {
|
||||
|
||||
@@ -120,6 +120,7 @@ func TestLookupStaticHost(t *testing.T) {
|
||||
}
|
||||
|
||||
func testStaticHost(t *testing.T, ent staticHostEntry, h *Hostsfile) {
|
||||
t.Helper()
|
||||
ins := []string{ent.in, plugin.Name(ent.in).Normalize(), strings.ToLower(ent.in), strings.ToUpper(ent.in)}
|
||||
for k, in := range ins {
|
||||
addrsV4 := h.LookupStaticHostV4(in)
|
||||
@@ -206,6 +207,7 @@ func TestLookupStaticAddr(t *testing.T) {
|
||||
}
|
||||
|
||||
func testStaticAddr(t *testing.T, ent staticIPEntry, h *Hostsfile) {
|
||||
t.Helper()
|
||||
hosts := h.LookupStaticAddr(ent.in)
|
||||
for i := range ent.out {
|
||||
ent.out[i] = plugin.Name(ent.out[i]).Normalize()
|
||||
|
||||
@@ -21,6 +21,7 @@ func TestDefaultProcessor(t *testing.T) {
|
||||
}
|
||||
|
||||
func testProcessor(t *testing.T, processor cache.ProcessFunc, idx cache.Indexer) {
|
||||
t.Helper()
|
||||
obj := &api.Service{
|
||||
ObjectMeta: metav1.ObjectMeta{Name: "service1", Namespace: "test1"},
|
||||
Spec: api.ServiceSpec{
|
||||
@@ -129,6 +130,7 @@ func TestDefaultProcessorWithPod(t *testing.T) {
|
||||
}
|
||||
|
||||
func testProcessorWithPod(t *testing.T, processor cache.ProcessFunc, idx cache.Indexer) {
|
||||
t.Helper()
|
||||
container := api.Container{}
|
||||
obj := &api.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{Name: "pod1", Namespace: "test1"},
|
||||
|
||||
@@ -105,6 +105,7 @@ func TestKubernetesIXFRCurrent(t *testing.T) {
|
||||
}
|
||||
|
||||
func validateAXFR(t *testing.T, ch <-chan []dns.RR, multicluster bool) {
|
||||
t.Helper()
|
||||
xfr := []dns.RR{}
|
||||
for rrs := range ch {
|
||||
xfr = append(xfr, rrs...)
|
||||
|
||||
@@ -136,6 +136,7 @@ func TestWeightFileUpdate(t *testing.T) {
|
||||
}
|
||||
|
||||
func checkDomainsWRR(t *testing.T, testIndex int, expectedDomains, domains map[string]weights) error {
|
||||
t.Helper()
|
||||
var ret error
|
||||
retError := errors.New("Check domains failed")
|
||||
for dname, expectedWeights := range expectedDomains {
|
||||
@@ -410,6 +411,7 @@ func TestLoadBalanceWRR(t *testing.T) {
|
||||
}
|
||||
|
||||
func checkTopIP(t *testing.T, i, j int, result []dns.RR, expectedTopIP string) {
|
||||
t.Helper()
|
||||
expected := net.ParseIP(expectedTopIP)
|
||||
for _, r := range result {
|
||||
switch r.Header().Rrtype {
|
||||
|
||||
@@ -43,6 +43,7 @@ func TestMultipleListenerMock(t *testing.T) {
|
||||
}
|
||||
|
||||
func testListenersCalled(t *testing.T, listenerNames []string, outputs []string) {
|
||||
t.Helper()
|
||||
for _, name := range listenerNames {
|
||||
err := RegisterListener(NewMockListener(name))
|
||||
if err != nil {
|
||||
|
||||
@@ -8,6 +8,7 @@ import (
|
||||
)
|
||||
|
||||
func getPEMFiles(t *testing.T) (cert, key, ca string) {
|
||||
t.Helper()
|
||||
tempDir, err := test.WritePEMFiles(t)
|
||||
if err != nil {
|
||||
t.Fatalf("Could not write PEM files: %s", err)
|
||||
|
||||
@@ -196,6 +196,7 @@ func TestQuicSetup(t *testing.T) {
|
||||
|
||||
// assertMaxStreamsValue compares the actual MaxQUICStreams value with the expected one
|
||||
func assertMaxStreamsValue(t *testing.T, testIndex int, testInput string, actual, expected *int) {
|
||||
t.Helper()
|
||||
if actual == nil && expected == nil {
|
||||
return
|
||||
}
|
||||
@@ -214,6 +215,7 @@ func assertMaxStreamsValue(t *testing.T, testIndex int, testInput string, actual
|
||||
|
||||
// assertWorkerPoolSizeValue compares the actual MaxQUICWorkerPoolSize value with the expected one
|
||||
func assertWorkerPoolSizeValue(t *testing.T, testIndex int, testInput string, actual, expected *int) {
|
||||
t.Helper()
|
||||
if actual == nil && expected == nil {
|
||||
return
|
||||
}
|
||||
|
||||
@@ -88,10 +88,11 @@ func TestCNameTargetRewrite(t *testing.T) {
|
||||
cnameTargetRule.Upstream = &MockedUpstream{}
|
||||
rules = append(rules, rule)
|
||||
}
|
||||
doTestCNameTargetTests(rules, t)
|
||||
doTestCNameTargetTests(t, rules)
|
||||
}
|
||||
|
||||
func doTestCNameTargetTests(rules []Rule, t *testing.T) {
|
||||
func doTestCNameTargetTests(t *testing.T, rules []Rule) {
|
||||
t.Helper()
|
||||
tests := []struct {
|
||||
from string
|
||||
fromType uint16
|
||||
|
||||
@@ -35,16 +35,17 @@ func TestResponseReverter(t *testing.T) {
|
||||
r, _ := newNameRule("stop", "regex", `(core)\.(dns)\.(rocks)`, "{2}.{1}.{3}", "answer", "name", `(dns)\.(core)\.(rocks)`, "{2}.{1}.{3}")
|
||||
rules = append(rules, r)
|
||||
|
||||
doReverterTests(rules, t)
|
||||
doReverterTests(t, rules)
|
||||
|
||||
rules = []Rule{}
|
||||
r, _ = newNameRule("continue", "regex", `(core)\.(dns)\.(rocks)`, "{2}.{1}.{3}", "answer", "name", `(dns)\.(core)\.(rocks)`, "{2}.{1}.{3}")
|
||||
rules = append(rules, r)
|
||||
|
||||
doReverterTests(rules, t)
|
||||
doReverterTests(t, rules)
|
||||
}
|
||||
|
||||
func doReverterTests(rules []Rule, t *testing.T) {
|
||||
func doReverterTests(t *testing.T, rules []Rule) {
|
||||
t.Helper()
|
||||
ctx := context.TODO()
|
||||
for i, tc := range tests {
|
||||
m := new(dns.Msg)
|
||||
@@ -105,7 +106,7 @@ func TestValueResponseReverter(t *testing.T) {
|
||||
}
|
||||
rules = append(rules, r)
|
||||
|
||||
doValueReverterTests("stop", rules, t)
|
||||
doValueReverterTests(t, "stop", rules)
|
||||
|
||||
rules = []Rule{}
|
||||
r, err = newNameRule("continue", "regex", `(.*)\.domain\.uk`, "{1}.cluster.local", "answer", "name", `(.*)\.cluster\.local`, "{1}.domain.uk", "answer", "value", `(.*)\.cluster\.local`, "{1}.domain.uk")
|
||||
@@ -115,7 +116,7 @@ func TestValueResponseReverter(t *testing.T) {
|
||||
}
|
||||
rules = append(rules, r)
|
||||
|
||||
doValueReverterTests("continue", rules, t)
|
||||
doValueReverterTests(t, "continue", rules)
|
||||
|
||||
rules = []Rule{}
|
||||
r, err = newNameRule("stop", "suffix", `.domain.uk`, ".cluster.local", "answer", "auto", "answer", "value", `(.*)\.cluster\.local`, "{1}.domain.uk")
|
||||
@@ -125,7 +126,7 @@ func TestValueResponseReverter(t *testing.T) {
|
||||
}
|
||||
rules = append(rules, r)
|
||||
|
||||
doValueReverterTests("suffix", rules, t)
|
||||
doValueReverterTests(t, "suffix", rules)
|
||||
|
||||
// multiple rules
|
||||
rules = []Rule{}
|
||||
@@ -143,10 +144,11 @@ func TestValueResponseReverter(t *testing.T) {
|
||||
}
|
||||
rules = append(rules, r)
|
||||
|
||||
doValueReverterTests("suffix_multiple", rules, t)
|
||||
doValueReverterTests(t, "suffix_multiple", rules)
|
||||
}
|
||||
|
||||
func doValueReverterTests(name string, rules []Rule, t *testing.T) {
|
||||
func doValueReverterTests(t *testing.T, name string, rules []Rule) {
|
||||
t.Helper()
|
||||
ctx := context.TODO()
|
||||
for i, tc := range valueTests {
|
||||
m := new(dns.Msg)
|
||||
|
||||
@@ -99,10 +99,11 @@ func TestTtlRewrite(t *testing.T) {
|
||||
}
|
||||
rules = append(rules, rule)
|
||||
}
|
||||
doTTLTests(rules, t)
|
||||
doTTLTests(t, rules)
|
||||
}
|
||||
|
||||
func doTTLTests(rules []Rule, t *testing.T) {
|
||||
func doTTLTests(t *testing.T, rules []Rule) {
|
||||
t.Helper()
|
||||
tests := []struct {
|
||||
from string
|
||||
fromType uint16
|
||||
|
||||
@@ -21,6 +21,7 @@ func TempFile(dir, content string) (string, func(), error) {
|
||||
|
||||
// WritePEMFiles creates a tmp dir with ca.pem, cert.pem, and key.pem
|
||||
func WritePEMFiles(t *testing.T) (string, error) {
|
||||
t.Helper()
|
||||
tempDir := t.TempDir()
|
||||
|
||||
data := `-----BEGIN CERTIFICATE-----
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
package test
|
||||
|
||||
import "testing"
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestTempFile(t *testing.T) {
|
||||
_, f, e := TempFile(".", "test")
|
||||
|
||||
@@ -218,6 +218,7 @@ func TestTransferIXFRFallback(t *testing.T) {
|
||||
}
|
||||
|
||||
func validateAXFRResponse(t *testing.T, w *dnstest.MultiRecorder) {
|
||||
t.Helper()
|
||||
if len(w.Msgs) == 0 {
|
||||
t.Fatal("Did not get back a zone response")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user