Speed up testing (#4239)

* Speed up testing

* make notification run in the background, this recudes the test_readme
time from 18s to 0.10s
* reduce time for zone reload

* TestServeDNSConcurrent remove entirely. This took a whopping 58s for
  ... ? A few minutes staring didn't reveal wth it is actually testing.
  Making values smaller revealed race conditions in the tests. Remove
  entirely.

* Move many interval values to variables so we can reset them to short
  values for the tests.

* test_large_axfr: make the zone smaller. The number used 64K has no
  rational, make it 64/10 to speed up.
* TestProxyThreeWay: use client with shorter timeout

A few random tidbits in other tests.

Total time saved: 177s (almost 3m) - which makes it worthwhile again to
run the test locally:

this branch:

~~~
ok  	github.com/coredns/coredns/test	10.437s
cd plugin; time go t ./...
5,51s user 7,51s system 11,15s elapsed 744%CPU (
~~~

master:

~~~
ok  	github.com/coredns/coredns/test	35.252s
cd plugin; time go t ./...
157,64s user 15,39s system 50,05s elapsed 345%CPU ()
~~~
tests/ -25s
plugins/ -40s

This brings the total on 20s, and another 10s can be saved by fixing
dnstapio. Moving this to 5s would be even better, but 10s is also nice.

Signed-off-by: Miek Gieben <miek@miek.nl>

* Also 0.01

Signed-off-by: Miek Gieben <miek@miek.nl>
This commit is contained in:
Miek Gieben 2020-10-30 10:27:04 +01:00 committed by GitHub
parent bc0115d71f
commit c840caf1ef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
15 changed files with 85 additions and 285 deletions

View file

@ -6,11 +6,8 @@ import (
"errors"
"fmt"
golog "log"
"math/rand"
"regexp"
"strconv"
"strings"
"sync"
"sync/atomic"
"testing"
"time"
@ -196,190 +193,6 @@ func TestStop(t *testing.T) {
}
}
const (
pattern0 = "failed"
pattern1 = "down$"
)
func TestServeDNSConcurrent(t *testing.T) {
buf := bytes.NewBuffer(nil)
golog.SetOutput(buf)
eg := newErrorGenerator()
h := &errorHandler{
patterns: []*pattern{
{
period: 3 * time.Nanosecond,
pattern: regexp.MustCompile(pattern0),
},
{
period: 2 * time.Nanosecond,
pattern: regexp.MustCompile(pattern1),
},
},
Next: eg,
}
ctx := context.TODO()
r := new(dns.Msg)
w := &test.ResponseWriter{}
var wg sync.WaitGroup
runnersCnt := 9
runner := func() {
defer wg.Done()
for eg.progress() < 100 {
h.ServeDNS(ctx, w, r)
}
}
wg.Add(runnersCnt)
for ; runnersCnt > 0; runnersCnt-- {
go runner()
}
stopCalled := false
for eg.progress() < 100 {
if !stopCalled && eg.progress() > 50 {
h.stop()
stopCalled = true
}
h.ServeDNS(ctx, w, r)
}
if !stopCalled {
h.stop()
}
wg.Wait()
time.Sleep(time.Millisecond)
rep := makeReport(t, buf)
if rep.c[Err0] == 0 {
t.Errorf("Err0 was never consolidated")
}
if rep.c[Err1] == 0 {
t.Errorf("Err1 was never consolidated")
}
if rep.c[Err0]+rep.r[Err0] != ErrCnt0 {
t.Errorf("Unexpected Err0 count, expected %d, consolidated %d and reported %d",
ErrCnt0, rep.c[Err0], rep.r[Err0])
}
if rep.c[Err1]+rep.r[Err1] != ErrCnt1 {
t.Errorf("Unexpected Err1 count, expected %d, consolidated %d and reported %d",
ErrCnt1, rep.c[Err1], rep.r[Err1])
}
if rep.r[Err2] != ErrCnt2 {
t.Errorf("Unexpected Err2 count, expected %d, reported %d",
ErrCnt2, rep.r[Err2])
}
if rep.r[Err3] != ErrCnt3 {
t.Errorf("Unexpected Err3 count, expected %d, reported %d",
ErrCnt3, rep.r[Err3])
}
}
// error generator
const (
Err0 = iota
Err1
Err2
Err3
ErrStr0 = "failed to connect"
ErrStr1 = "upstream is down"
ErrStr2 = "access denied"
ErrStr3 = "yet another error"
ErrCnt0 = 120000
ErrCnt1 = 130000
ErrCnt2 = 150000
ErrCnt3 = 100000
)
type errorBunch struct {
cnt int32
err error
}
type errorGenerator struct {
errors [4]errorBunch
totalCnt int32
totalLim int32
}
func newErrorGenerator() *errorGenerator {
rand.Seed(time.Now().UnixNano())
return &errorGenerator{
errors: [4]errorBunch{
{ErrCnt0, fmt.Errorf(ErrStr0)},
{ErrCnt1, fmt.Errorf(ErrStr1)},
{ErrCnt2, fmt.Errorf(ErrStr2)},
{ErrCnt3, fmt.Errorf(ErrStr3)},
},
totalLim: ErrCnt0 + ErrCnt1 + ErrCnt2 + ErrCnt3,
}
}
// returns progress as a value from [0..100] range
func (sh *errorGenerator) progress() int32 {
return atomic.LoadInt32(&sh.totalCnt) * 100 / sh.totalLim
}
func (sh *errorGenerator) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
i := rand.Int()
for c := 0; c < 4; c++ {
errInd := (i + c) & 3
if atomic.AddInt32(&sh.errors[errInd].cnt, -1) >= 0 {
atomic.AddInt32(&sh.totalCnt, 1)
return 0, sh.errors[errInd].err
}
atomic.AddInt32(&sh.errors[errInd].cnt, 1)
}
return 0, nil
}
func (sh *errorGenerator) Name() string { return "errorGenerator" }
// error report
type errorReport struct {
c [4]uint32
r [4]uint32
}
func makeReport(t *testing.T, b *bytes.Buffer) errorReport {
logDump := b.String()
r := errorReport{}
re := regexp.MustCompile(`(\d+) errors like '(.*)' occurred`)
m := re.FindAllStringSubmatch(logDump, -1)
for _, sub := range m {
ind := 0
switch sub[2] {
case pattern0:
ind = Err0
case pattern1:
ind = Err1
default:
t.Fatalf("Unknown pattern found in log: %s", sub[0])
}
n, err := strconv.ParseUint(sub[1], 10, 32)
if err != nil {
t.Fatalf("Invalid number found in log: %s", sub[0])
}
r.c[ind] += uint32(n)
}
t.Logf("%d consolidated messages found", len(m))
r.r[Err0] = uint32(strings.Count(logDump, ErrStr0))
r.r[Err1] = uint32(strings.Count(logDump, ErrStr1))
r.r[Err2] = uint32(strings.Count(logDump, ErrStr2))
r.r[Err3] = uint32(strings.Count(logDump, ErrStr3))
return r
}
func genErrorHandler(rcode int, err error) plugin.Handler {
return plugin.HandlerFunc(func(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
return rcode, err