1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
package test
import (
"bufio"
"io/ioutil"
"log"
"os"
"path/filepath"
"strconv"
"testing"
"github.com/coredns/coredns/core/dnsserver"
"github.com/mholt/caddy"
)
// Pasrse all README.md's of the middleware and check if every example Corefile
// actually works. Each corefile is only used if:
//
// ~~~ corefile
// . {
// # check-this-please
// }
// ~~~
func TestReadme(t *testing.T) {
port := 30053
caddy.Quiet = true
dnsserver.Quiet = true
log.SetOutput(ioutil.Discard)
middle := filepath.Join("..", "middleware")
dirs, err := ioutil.ReadDir(middle)
if err != nil {
t.Fatalf("Could not read %s: %q", middle, err)
}
for _, d := range dirs {
if !d.IsDir() {
continue
}
readme := filepath.Join(middle, d.Name())
readme = filepath.Join(readme, "README.md")
inputs, err := corefileFromReadme(readme)
if err != nil {
continue
}
// Test each snippet.
for _, in := range inputs {
t.Logf("Testing %s, with %d byte snippet", readme, len(in.Body()))
dnsserver.Port = strconv.Itoa(port)
server, err := caddy.Start(in)
if err != nil {
t.Errorf("Failed to start server for input %q:\n%s", err, in.Body())
}
server.Stop()
port++
}
}
}
// corefileFromReadme parses a readme and returns all fragments that
// have ~~~ corefile (or ``` corefile).
func corefileFromReadme(readme string) ([]*Input, error) {
f, err := os.Open(readme)
if err != nil {
return nil, err
}
defer f.Close()
s := bufio.NewScanner(f)
input := []*Input{}
corefile := false
temp := ""
for s.Scan() {
line := s.Text()
if line == "~~~ corefile" || line == "``` corefile" {
corefile = true
continue
}
if corefile && (line == "~~~" || line == "```") {
// last line
input = append(input, NewInput(temp))
temp = ""
corefile = false
continue
}
if corefile {
temp += line + "\n" // readd newline stripped by s.Text()
}
}
if err := s.Err(); err != nil {
return nil, err
}
return input, nil
}
|
