Source file
src/net/dnsconfig_unix_test.go
Documentation: net
1
2
3
4
5
6
7 package net
8
9 import (
10 "errors"
11 "os"
12 "reflect"
13 "strings"
14 "testing"
15 "time"
16 )
17
18 var dnsReadConfigTests = []struct {
19 name string
20 want *dnsConfig
21 }{
22 {
23 name: "testdata/resolv.conf",
24 want: &dnsConfig{
25 servers: []string{"8.8.8.8:53", "[2001:4860:4860::8888]:53", "[fe80::1%lo0]:53"},
26 search: []string{"localdomain."},
27 ndots: 5,
28 timeout: 10 * time.Second,
29 attempts: 3,
30 rotate: true,
31 unknownOpt: true,
32 },
33 },
34 {
35 name: "testdata/domain-resolv.conf",
36 want: &dnsConfig{
37 servers: []string{"8.8.8.8:53"},
38 search: []string{"localdomain."},
39 ndots: 1,
40 timeout: 5 * time.Second,
41 attempts: 2,
42 },
43 },
44 {
45 name: "testdata/search-resolv.conf",
46 want: &dnsConfig{
47 servers: []string{"8.8.8.8:53"},
48 search: []string{"test.", "invalid."},
49 ndots: 1,
50 timeout: 5 * time.Second,
51 attempts: 2,
52 },
53 },
54 {
55 name: "testdata/empty-resolv.conf",
56 want: &dnsConfig{
57 servers: defaultNS,
58 ndots: 1,
59 timeout: 5 * time.Second,
60 attempts: 2,
61 search: []string{"domain.local."},
62 },
63 },
64 {
65 name: "testdata/invalid-ndots-resolv.conf",
66 want: &dnsConfig{
67 servers: defaultNS,
68 ndots: 0,
69 timeout: 5 * time.Second,
70 attempts: 2,
71 search: []string{"domain.local."},
72 },
73 },
74 {
75 name: "testdata/large-ndots-resolv.conf",
76 want: &dnsConfig{
77 servers: defaultNS,
78 ndots: 15,
79 timeout: 5 * time.Second,
80 attempts: 2,
81 search: []string{"domain.local."},
82 },
83 },
84 {
85 name: "testdata/negative-ndots-resolv.conf",
86 want: &dnsConfig{
87 servers: defaultNS,
88 ndots: 0,
89 timeout: 5 * time.Second,
90 attempts: 2,
91 search: []string{"domain.local."},
92 },
93 },
94 {
95 name: "testdata/openbsd-resolv.conf",
96 want: &dnsConfig{
97 ndots: 1,
98 timeout: 5 * time.Second,
99 attempts: 2,
100 lookup: []string{"file", "bind"},
101 servers: []string{"169.254.169.254:53", "10.240.0.1:53"},
102 search: []string{"c.symbolic-datum-552.internal."},
103 },
104 },
105 }
106
107 func TestDNSReadConfig(t *testing.T) {
108 origGetHostname := getHostname
109 defer func() { getHostname = origGetHostname }()
110 getHostname = func() (string, error) { return "host.domain.local", nil }
111
112 for _, tt := range dnsReadConfigTests {
113 conf := dnsReadConfig(tt.name)
114 if conf.err != nil {
115 t.Fatal(conf.err)
116 }
117 conf.mtime = time.Time{}
118 if !reflect.DeepEqual(conf, tt.want) {
119 t.Errorf("%s:\ngot: %+v\nwant: %+v", tt.name, conf, tt.want)
120 }
121 }
122 }
123
124 func TestDNSReadMissingFile(t *testing.T) {
125 origGetHostname := getHostname
126 defer func() { getHostname = origGetHostname }()
127 getHostname = func() (string, error) { return "host.domain.local", nil }
128
129 conf := dnsReadConfig("a-nonexistent-file")
130 if !os.IsNotExist(conf.err) {
131 t.Errorf("missing resolv.conf:\ngot: %v\nwant: %v", conf.err, os.ErrNotExist)
132 }
133 conf.err = nil
134 want := &dnsConfig{
135 servers: defaultNS,
136 ndots: 1,
137 timeout: 5 * time.Second,
138 attempts: 2,
139 search: []string{"domain.local."},
140 }
141 if !reflect.DeepEqual(conf, want) {
142 t.Errorf("missing resolv.conf:\ngot: %+v\nwant: %+v", conf, want)
143 }
144 }
145
146 var dnsDefaultSearchTests = []struct {
147 name string
148 err error
149 want []string
150 }{
151 {
152 name: "host.long.domain.local",
153 want: []string{"long.domain.local."},
154 },
155 {
156 name: "host.local",
157 want: []string{"local."},
158 },
159 {
160 name: "host",
161 want: nil,
162 },
163 {
164 name: "host.domain.local",
165 err: errors.New("errored"),
166 want: nil,
167 },
168 {
169
170
171 name: "foo.",
172 want: nil,
173 },
174 }
175
176 func TestDNSDefaultSearch(t *testing.T) {
177 origGetHostname := getHostname
178 defer func() { getHostname = origGetHostname }()
179
180 for _, tt := range dnsDefaultSearchTests {
181 getHostname = func() (string, error) { return tt.name, tt.err }
182 got := dnsDefaultSearch()
183 if !reflect.DeepEqual(got, tt.want) {
184 t.Errorf("dnsDefaultSearch with hostname %q and error %+v = %q, wanted %q", tt.name, tt.err, got, tt.want)
185 }
186 }
187 }
188
189 func TestDNSNameLength(t *testing.T) {
190 origGetHostname := getHostname
191 defer func() { getHostname = origGetHostname }()
192 getHostname = func() (string, error) { return "host.domain.local", nil }
193
194 var char63 = ""
195 for i := 0; i < 63; i++ {
196 char63 += "a"
197 }
198 longDomain := strings.Repeat(char63+".", 5) + "example"
199
200 for _, tt := range dnsReadConfigTests {
201 conf := dnsReadConfig(tt.name)
202 if conf.err != nil {
203 t.Fatal(conf.err)
204 }
205
206 var shortestSuffix int
207 for _, suffix := range tt.want.search {
208 if shortestSuffix == 0 || len(suffix) < shortestSuffix {
209 shortestSuffix = len(suffix)
210 }
211 }
212
213
214
215 longName := longDomain[len(longDomain)-254+1+shortestSuffix:]
216 if longName[0] == '.' || longName[1] == '.' {
217 longName = "aa." + longName[3:]
218 }
219 for _, fqdn := range conf.nameList(longName) {
220 if len(fqdn) > 254 {
221 t.Errorf("got %d; want less than or equal to 254", len(fqdn))
222 }
223 }
224
225
226 unsuffixable := "a." + longName[1:]
227 unsuffixableResults := conf.nameList(unsuffixable)
228 if len(unsuffixableResults) != 1 {
229 t.Errorf("suffixed names %v; want []", unsuffixableResults[1:])
230 }
231
232
233 tooLong := "a." + longDomain
234 tooLongResults := conf.nameList(tooLong)
235 if tooLongResults != nil {
236 t.Errorf("suffixed names %v; want nil", tooLongResults)
237 }
238 }
239 }
240
View as plain text