source: trunk/server/lib/dnspython/tests/zone.py

Last change on this file was 745, checked in by Silvan Scherrer, 13 years ago

Samba Server: updated trunk to 3.6.0
File size: 14.1 KB
Line 
1# Copyright (C) 2003-2007, 2009, 2010 Nominum, Inc.
2#
3# Permission to use, copy, modify, and distribute this software and its
4# documentation for any purpose with or without fee is hereby granted,
5# provided that the above copyright notice and this permission notice
6# appear in all copies.
7#
8# THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
9# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
11# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
14# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15
16import cStringIO
17import filecmp
18import os
19import unittest
20
21import dns.exception
22import dns.rdata
23import dns.rdataclass
24import dns.rdatatype
25import dns.rrset
26import dns.zone
27
28example_text = """$TTL 3600
29$ORIGIN example.
30@ soa foo bar 1 2 3 4 5
31@ ns ns1
32@ ns ns2
33ns1 a 10.0.0.1
34ns2 a 10.0.0.2
35$TTL 300
36$ORIGIN foo.example.
37bar mx 0 blaz
38"""
39
40example_text_output = """@ 3600 IN SOA foo bar 1 2 3 4 5
41@ 3600 IN NS ns1
42@ 3600 IN NS ns2
43bar.foo 300 IN MX 0 blaz.foo
44ns1 3600 IN A 10.0.0.1
45ns2 3600 IN A 10.0.0.2
46"""
47
48something_quite_similar = """@ 3600 IN SOA foo bar 1 2 3 4 5
49@ 3600 IN NS ns1
50@ 3600 IN NS ns2
51bar.foo 300 IN MX 0 blaz.foo
52ns1 3600 IN A 10.0.0.1
53ns2 3600 IN A 10.0.0.3
54"""
55
56something_different = """@ 3600 IN SOA fooa bar 1 2 3 4 5
57@ 3600 IN NS ns11
58@ 3600 IN NS ns21
59bar.fooa 300 IN MX 0 blaz.fooa
60ns11 3600 IN A 10.0.0.11
61ns21 3600 IN A 10.0.0.21
62"""
63
64ttl_example_text = """$TTL 1h
65$ORIGIN example.
66@ soa foo bar 1 2 3 4 5
67@ ns ns1
68@ ns ns2
69ns1 1d1s a 10.0.0.1
70ns2 1w1D1h1m1S a 10.0.0.2
71"""
72
73no_soa_text = """$TTL 1h
74$ORIGIN example.
75@ ns ns1
76@ ns ns2
77ns1 1d1s a 10.0.0.1
78ns2 1w1D1h1m1S a 10.0.0.2
79"""
80
81no_ns_text = """$TTL 1h
82$ORIGIN example.
83@ soa foo bar 1 2 3 4 5
84"""
85
86include_text = """$INCLUDE "example"
87"""
88
89bad_directive_text = """$FOO bar
90$ORIGIN example.
91@ soa foo bar 1 2 3 4 5
92@ ns ns1
93@ ns ns2
94ns1 1d1s a 10.0.0.1
95ns2 1w1D1h1m1S a 10.0.0.2
96"""
97
98_keep_output = False
99
100class ZoneTestCase(unittest.TestCase):
101
102 def testFromFile1(self):
103 z = dns.zone.from_file('example', 'example')
104 ok = False
105 try:
106 z.to_file('example1.out', nl='\x0a')
107 ok = filecmp.cmp('example1.out', 'example1.good')
108 finally:
109 if not _keep_output:
110 os.unlink('example1.out')
111 self.failUnless(ok)
112
113 def testFromFile2(self):
114 z = dns.zone.from_file('example', 'example', relativize=False)
115 ok = False
116 try:
117 z.to_file('example2.out', relativize=False, nl='\x0a')
118 ok = filecmp.cmp('example2.out', 'example2.good')
119 finally:
120 if not _keep_output:
121 os.unlink('example2.out')
122 self.failUnless(ok)
123
124 def testFromText(self):
125 z = dns.zone.from_text(example_text, 'example.', relativize=True)
126 f = cStringIO.StringIO()
127 names = z.nodes.keys()
128 names.sort()
129 for n in names:
130 print >> f, z[n].to_text(n)
131 self.failUnless(f.getvalue() == example_text_output)
132
133 def testTorture1(self):
134 #
135 # Read a zone containing all our supported RR types, and
136 # for each RR in the zone, convert the rdata into wire format
137 # and then back out, and see if we get equal rdatas.
138 #
139 f = cStringIO.StringIO()
140 o = dns.name.from_text('example.')
141 z = dns.zone.from_file('example', o)
142 for (name, node) in z.iteritems():
143 for rds in node:
144 for rd in rds:
145 f.seek(0)
146 f.truncate()
147 rd.to_wire(f, origin=o)
148 wire = f.getvalue()
149 rd2 = dns.rdata.from_wire(rds.rdclass, rds.rdtype,
150 wire, 0, len(wire),
151 origin = o)
152 self.failUnless(rd == rd2)
153
154 def testEqual(self):
155 z1 = dns.zone.from_text(example_text, 'example.', relativize=True)
156 z2 = dns.zone.from_text(example_text_output, 'example.',
157 relativize=True)
158 self.failUnless(z1 == z2)
159
160 def testNotEqual1(self):
161 z1 = dns.zone.from_text(example_text, 'example.', relativize=True)
162 z2 = dns.zone.from_text(something_quite_similar, 'example.',
163 relativize=True)
164 self.failUnless(z1 != z2)
165
166 def testNotEqual2(self):
167 z1 = dns.zone.from_text(example_text, 'example.', relativize=True)
168 z2 = dns.zone.from_text(something_different, 'example.',
169 relativize=True)
170 self.failUnless(z1 != z2)
171
172 def testNotEqual3(self):
173 z1 = dns.zone.from_text(example_text, 'example.', relativize=True)
174 z2 = dns.zone.from_text(something_different, 'example2.',
175 relativize=True)
176 self.failUnless(z1 != z2)
177
178 def testFindRdataset1(self):
179 z = dns.zone.from_text(example_text, 'example.', relativize=True)
180 rds = z.find_rdataset('@', 'soa')
181 exrds = dns.rdataset.from_text('IN', 'SOA', 300, 'foo bar 1 2 3 4 5')
182 self.failUnless(rds == exrds)
183
184 def testFindRdataset2(self):
185 def bad():
186 z = dns.zone.from_text(example_text, 'example.', relativize=True)
187 rds = z.find_rdataset('@', 'loc')
188 self.failUnlessRaises(KeyError, bad)
189
190 def testFindRRset1(self):
191 z = dns.zone.from_text(example_text, 'example.', relativize=True)
192 rrs = z.find_rrset('@', 'soa')
193 exrrs = dns.rrset.from_text('@', 300, 'IN', 'SOA', 'foo bar 1 2 3 4 5')
194 self.failUnless(rrs == exrrs)
195
196 def testFindRRset2(self):
197 def bad():
198 z = dns.zone.from_text(example_text, 'example.', relativize=True)
199 rrs = z.find_rrset('@', 'loc')
200 self.failUnlessRaises(KeyError, bad)
201
202 def testGetRdataset1(self):
203 z = dns.zone.from_text(example_text, 'example.', relativize=True)
204 rds = z.get_rdataset('@', 'soa')
205 exrds = dns.rdataset.from_text('IN', 'SOA', 300, 'foo bar 1 2 3 4 5')
206 self.failUnless(rds == exrds)
207
208 def testGetRdataset2(self):
209 z = dns.zone.from_text(example_text, 'example.', relativize=True)
210 rds = z.get_rdataset('@', 'loc')
211 self.failUnless(rds == None)
212
213 def testGetRRset1(self):
214 z = dns.zone.from_text(example_text, 'example.', relativize=True)
215 rrs = z.get_rrset('@', 'soa')
216 exrrs = dns.rrset.from_text('@', 300, 'IN', 'SOA', 'foo bar 1 2 3 4 5')
217 self.failUnless(rrs == exrrs)
218
219 def testGetRRset2(self):
220 z = dns.zone.from_text(example_text, 'example.', relativize=True)
221 rrs = z.get_rrset('@', 'loc')
222 self.failUnless(rrs == None)
223
224 def testReplaceRdataset1(self):
225 z = dns.zone.from_text(example_text, 'example.', relativize=True)
226 rdataset = dns.rdataset.from_text('in', 'ns', 300, 'ns3', 'ns4')
227 z.replace_rdataset('@', rdataset)
228 rds = z.get_rdataset('@', 'ns')
229 self.failUnless(rds is rdataset)
230
231 def testReplaceRdataset2(self):
232 z = dns.zone.from_text(example_text, 'example.', relativize=True)
233 rdataset = dns.rdataset.from_text('in', 'txt', 300, '"foo"')
234 z.replace_rdataset('@', rdataset)
235 rds = z.get_rdataset('@', 'txt')
236 self.failUnless(rds is rdataset)
237
238 def testDeleteRdataset1(self):
239 z = dns.zone.from_text(example_text, 'example.', relativize=True)
240 z.delete_rdataset('@', 'ns')
241 rds = z.get_rdataset('@', 'ns')
242 self.failUnless(rds is None)
243
244 def testDeleteRdataset2(self):
245 z = dns.zone.from_text(example_text, 'example.', relativize=True)
246 z.delete_rdataset('ns1', 'a')
247 node = z.get_node('ns1')
248 self.failUnless(node is None)
249
250 def testNodeFindRdataset1(self):
251 z = dns.zone.from_text(example_text, 'example.', relativize=True)
252 node = z['@']
253 rds = node.find_rdataset(dns.rdataclass.IN, dns.rdatatype.SOA)
254 exrds = dns.rdataset.from_text('IN', 'SOA', 300, 'foo bar 1 2 3 4 5')
255 self.failUnless(rds == exrds)
256
257 def testNodeFindRdataset2(self):
258 def bad():
259 z = dns.zone.from_text(example_text, 'example.', relativize=True)
260 node = z['@']
261 rds = node.find_rdataset(dns.rdataclass.IN, dns.rdatatype.LOC)
262 self.failUnlessRaises(KeyError, bad)
263
264 def testNodeGetRdataset1(self):
265 z = dns.zone.from_text(example_text, 'example.', relativize=True)
266 node = z['@']
267 rds = node.get_rdataset(dns.rdataclass.IN, dns.rdatatype.SOA)
268 exrds = dns.rdataset.from_text('IN', 'SOA', 300, 'foo bar 1 2 3 4 5')
269 self.failUnless(rds == exrds)
270
271 def testNodeGetRdataset2(self):
272 z = dns.zone.from_text(example_text, 'example.', relativize=True)
273 node = z['@']
274 rds = node.get_rdataset(dns.rdataclass.IN, dns.rdatatype.LOC)
275 self.failUnless(rds == None)
276
277 def testNodeDeleteRdataset1(self):
278 z = dns.zone.from_text(example_text, 'example.', relativize=True)
279 node = z['@']
280 rds = node.delete_rdataset(dns.rdataclass.IN, dns.rdatatype.SOA)
281 rds = node.get_rdataset(dns.rdataclass.IN, dns.rdatatype.SOA)
282 self.failUnless(rds == None)
283
284 def testNodeDeleteRdataset2(self):
285 z = dns.zone.from_text(example_text, 'example.', relativize=True)
286 node = z['@']
287 rds = node.delete_rdataset(dns.rdataclass.IN, dns.rdatatype.LOC)
288 rds = node.get_rdataset(dns.rdataclass.IN, dns.rdatatype.LOC)
289 self.failUnless(rds == None)
290
291 def testIterateRdatasets(self):
292 z = dns.zone.from_text(example_text, 'example.', relativize=True)
293 ns = [n for n, r in z.iterate_rdatasets('A')]
294 ns.sort()
295 self.failUnless(ns == [dns.name.from_text('ns1', None),
296 dns.name.from_text('ns2', None)])
297
298 def testIterateAllRdatasets(self):
299 z = dns.zone.from_text(example_text, 'example.', relativize=True)
300 ns = [n for n, r in z.iterate_rdatasets()]
301 ns.sort()
302 self.failUnless(ns == [dns.name.from_text('@', None),
303 dns.name.from_text('@', None),
304 dns.name.from_text('bar.foo', None),
305 dns.name.from_text('ns1', None),
306 dns.name.from_text('ns2', None)])
307
308 def testIterateRdatas(self):
309 z = dns.zone.from_text(example_text, 'example.', relativize=True)
310 l = list(z.iterate_rdatas('A'))
311 l.sort()
312 exl = [(dns.name.from_text('ns1', None),
313 3600,
314 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A,
315 '10.0.0.1')),
316 (dns.name.from_text('ns2', None),
317 3600,
318 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A,
319 '10.0.0.2'))]
320 self.failUnless(l == exl)
321
322 def testIterateAllRdatas(self):
323 z = dns.zone.from_text(example_text, 'example.', relativize=True)
324 l = list(z.iterate_rdatas())
325 l.sort()
326 exl = [(dns.name.from_text('@', None),
327 3600,
328 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.NS,
329 'ns1')),
330 (dns.name.from_text('@', None),
331 3600,
332 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.NS,
333 'ns2')),
334 (dns.name.from_text('@', None),
335 3600,
336 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.SOA,
337 'foo bar 1 2 3 4 5')),
338 (dns.name.from_text('bar.foo', None),
339 300,
340 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.MX,
341 '0 blaz.foo')),
342 (dns.name.from_text('ns1', None),
343 3600,
344 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A,
345 '10.0.0.1')),
346 (dns.name.from_text('ns2', None),
347 3600,
348 dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A,
349 '10.0.0.2'))]
350 self.failUnless(l == exl)
351
352 def testTTLs(self):
353 z = dns.zone.from_text(ttl_example_text, 'example.', relativize=True)
354 n = z['@']
355 rds = n.get_rdataset(dns.rdataclass.IN, dns.rdatatype.SOA)
356 self.failUnless(rds.ttl == 3600)
357 n = z['ns1']
358 rds = n.get_rdataset(dns.rdataclass.IN, dns.rdatatype.A)
359 self.failUnless(rds.ttl == 86401)
360 n = z['ns2']
361 rds = n.get_rdataset(dns.rdataclass.IN, dns.rdatatype.A)
362 self.failUnless(rds.ttl == 694861)
363
364 def testNoSOA(self):
365 def bad():
366 z = dns.zone.from_text(no_soa_text, 'example.',
367 relativize=True)
368 self.failUnlessRaises(dns.zone.NoSOA, bad)
369
370 def testNoNS(self):
371 def bad():
372 z = dns.zone.from_text(no_ns_text, 'example.',
373 relativize=True)
374 self.failUnlessRaises(dns.zone.NoNS, bad)
375
376 def testInclude(self):
377 z1 = dns.zone.from_text(include_text, 'example.', relativize=True,
378 allow_include=True)
379 z2 = dns.zone.from_file('example', 'example.', relativize=True)
380 self.failUnless(z1 == z2)
381
382 def testBadDirective(self):
383 def bad():
384 z = dns.zone.from_text(bad_directive_text, 'example.',
385 relativize=True)
386 self.failUnlessRaises(dns.exception.SyntaxError, bad)
387
388if __name__ == '__main__':
389 unittest.main()
Note: See TracBrowser for help on using the repository browser.