~ubuntu-branches/ubuntu/vivid/dnspython/vivid-proposed : revision 12

1

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

16

import unittest

17

18

import cStringIO

19

import socket

20

21

import dns.name

22

import dns.reversename

23

import dns.e164

24

25

class NameTestCase(unittest.TestCase):

26

def setUp(self):

27

self.origin = dns.name.from_text('example.')

28

29

def testFromTextRel1(self):

30

n = dns.name.from_text('foo.bar')

31

self.failUnless(n.labels == ('foo', 'bar', ''))

32

33

def testFromTextRel2(self):

34

n = dns.name.from_text('foo.bar', origin=self.origin)

35

self.failUnless(n.labels == ('foo', 'bar', 'example', ''))

36

37

def testFromTextRel3(self):

38

n = dns.name.from_text('foo.bar', origin=None)

39

self.failUnless(n.labels == ('foo', 'bar'))

40

41

def testFromTextRel4(self):

42

n = dns.name.from_text('@', origin=None)

43

self.failUnless(n == dns.name.empty)

44

45

def testFromTextRel5(self):

46

n = dns.name.from_text('@', origin=self.origin)

47

self.failUnless(n == self.origin)

48

49

def testFromTextAbs1(self):

50

n = dns.name.from_text('foo.bar.')

51

self.failUnless(n.labels == ('foo', 'bar', ''))

52

53

def testTortureFromText(self):

54

good = [

55

r'.',

56

r'a',

57

r'a.',

58

r'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',

59

r'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',

60

r'\000.\008.\010.\032.\046.\092.\099.\255',

61

r'\\',

62

r'\..\.',

63

r'\\.\\',

64

r'!"#%&/()=+-',

65

r'\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255',

66

]

67

bad = [

68

r'..',

69

r'.a',

70

r'\\..',

71

'\\', # yes, we don't want the 'r' prefix!

72

r'\0',

73

r'\00',

74

r'\00Z',

75

r'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',

76

r'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',

77

r'\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255.\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255\255',

78

]

79

for t in good:

80

try:

81

n = dns.name.from_text(t)

82

except:

83

self.fail("good test '%s' raised an exception" % t)

84

for t in bad:

85

caught = False

86

try:

87

n = dns.name.from_text(t)

88

except:

89

caught = True

90

if not caught:

91

self.fail("bad test '%s' did not raise an exception" % t)

92

93

def testImmutable1(self):

94

def bad():

95

self.origin.labels = ()

96

self.failUnlessRaises(TypeError, bad)

97

98

def testImmutable2(self):

99

def bad():

100

self.origin.labels[0] = 'foo'

101

self.failUnlessRaises(TypeError, bad)

102

103

def testAbs1(self):

104

self.failUnless(dns.name.root.is_absolute())

105

106

def testAbs2(self):

107

self.failUnless(not dns.name.empty.is_absolute())

108

109

def testAbs3(self):

110

self.failUnless(self.origin.is_absolute())

111

112

def testAbs3(self):

113

n = dns.name.from_text('foo', origin=None)

114

self.failUnless(not n.is_absolute())

115

116

def testWild1(self):

117

n = dns.name.from_text('*.foo', origin=None)

118

self.failUnless(n.is_wild())

119

120

def testWild2(self):

121

n = dns.name.from_text('*a.foo', origin=None)

122

self.failUnless(not n.is_wild())

123

124

def testWild3(self):

125

n = dns.name.from_text('a.*.foo', origin=None)

126

self.failUnless(not n.is_wild())

127

128

def testWild4(self):

129

self.failUnless(not dns.name.root.is_wild())

130

131

def testWild5(self):

132

self.failUnless(not dns.name.empty.is_wild())

133

134

def testHash1(self):

135

n1 = dns.name.from_text('fOo.COM')

136

n2 = dns.name.from_text('foo.com')

137

self.failUnless(hash(n1) == hash(n2))

138

139

def testCompare1(self):

140

n1 = dns.name.from_text('a')

141

n2 = dns.name.from_text('b')

142

self.failUnless(n1 < n2)

143

self.failUnless(n2 > n1)

144

145

def testCompare2(self):

146

n1 = dns.name.from_text('')

147

n2 = dns.name.from_text('b')

148

self.failUnless(n1 < n2)

149

self.failUnless(n2 > n1)

150

151

def testCompare3(self):

152

self.failUnless(dns.name.empty < dns.name.root)

153

self.failUnless(dns.name.root > dns.name.empty)

154

155

def testCompare4(self):

156

self.failUnless(dns.name.root != 1)

157

158

def testCompare5(self):

159

self.failUnless(dns.name.root < 1 or dns.name.root > 1)

160

161

def testSubdomain1(self):

162

self.failUnless(not dns.name.empty.is_subdomain(dns.name.root))

163

164

def testSubdomain2(self):

165

self.failUnless(not dns.name.root.is_subdomain(dns.name.empty))

166

167

def testSubdomain3(self):

168

n = dns.name.from_text('foo', origin=self.origin)

169

self.failUnless(n.is_subdomain(self.origin))

170

171

def testSubdomain4(self):

172

n = dns.name.from_text('foo', origin=self.origin)

173

self.failUnless(n.is_subdomain(dns.name.root))

174

175

def testSubdomain5(self):

176

n = dns.name.from_text('foo', origin=self.origin)

177

self.failUnless(n.is_subdomain(n))

178

179

def testSuperdomain1(self):

180

self.failUnless(not dns.name.empty.is_superdomain(dns.name.root))

181

182

def testSuperdomain2(self):

183

self.failUnless(not dns.name.root.is_superdomain(dns.name.empty))

184

185

def testSuperdomain3(self):

186

n = dns.name.from_text('foo', origin=self.origin)

187

self.failUnless(self.origin.is_superdomain(n))

188

189

def testSuperdomain4(self):

190

n = dns.name.from_text('foo', origin=self.origin)

191

self.failUnless(dns.name.root.is_superdomain(n))

192

193

def testSuperdomain5(self):

194

n = dns.name.from_text('foo', origin=self.origin)

195

self.failUnless(n.is_superdomain(n))

196

197

def testCanonicalize1(self):

198

n = dns.name.from_text('FOO.bar', origin=self.origin)

199

c = n.canonicalize()

200

self.failUnless(c.labels == ('foo', 'bar', 'example', ''))

201

202

def testToText1(self):

203

n = dns.name.from_text('FOO.bar', origin=self.origin)

204

t = n.to_text()

205

self.failUnless(t == 'FOO.bar.example.')

206

207

def testToText2(self):

208

n = dns.name.from_text('FOO.bar', origin=self.origin)

209

t = n.to_text(True)

210

self.failUnless(t == 'FOO.bar.example')

211

212

def testToText3(self):

213

n = dns.name.from_text('FOO.bar', origin=None)

214

t = n.to_text()

215

self.failUnless(t == 'FOO.bar')

216

217

def testToText4(self):

218

t = dns.name.empty.to_text()

219

self.failUnless(t == '@')

220

221

def testToText5(self):

222

t = dns.name.root.to_text()

223

self.failUnless(t == '.')

224

225

def testToText6(self):

226

n = dns.name.from_text('FOO bar', origin=None)

227

t = n.to_text()

228

self.failUnless(t == r'FOO\032bar')

229

230

def testToText7(self):

231

n = dns.name.from_text(r'FOO\.bar', origin=None)

232

t = n.to_text()

233

self.failUnless(t == r'FOO\.bar')

234

235

def testToText8(self):

236

n = dns.name.from_text(r'\070OO\.bar', origin=None)

237

t = n.to_text()

238

self.failUnless(t == r'FOO\.bar')

239

240

def testSlice1(self):

241

n = dns.name.from_text(r'a.b.c.', origin=None)

242

s = n[:]

243

self.failUnless(s == ('a', 'b', 'c', ''))

244

245

def testSlice2(self):

246

n = dns.name.from_text(r'a.b.c.', origin=None)

247

s = n[:2]

248

self.failUnless(s == ('a', 'b'))

249

250

def testSlice3(self):

251

n = dns.name.from_text(r'a.b.c.', origin=None)

252

s = n[2:]

253

self.failUnless(s == ('c', ''))

254

255

def testEmptyLabel1(self):

256

def bad():

257

n = dns.name.Name(['a', '', 'b'])

258

self.failUnlessRaises(dns.name.EmptyLabel, bad)

259

260

def testEmptyLabel2(self):

261

def bad():

262

n = dns.name.Name(['', 'b'])

263

self.failUnlessRaises(dns.name.EmptyLabel, bad)

264

265

def testEmptyLabel3(self):

266

n = dns.name.Name(['b', ''])

267

self.failUnless(n)

268

269

def testLongLabel(self):

270

n = dns.name.Name(['a' * 63])

271

self.failUnless(n)

272

273

def testLabelTooLong(self):

274

def bad():

275

n = dns.name.Name(['a' * 64, 'b'])

276

self.failUnlessRaises(dns.name.LabelTooLong, bad)

277

278

def testLongName(self):

279

n = dns.name.Name(['a' * 63, 'a' * 63, 'a' * 63, 'a' * 62])

280

self.failUnless(n)

281

282

def testNameTooLong(self):

283

def bad():

284

n = dns.name.Name(['a' * 63, 'a' * 63, 'a' * 63, 'a' * 63])

285

self.failUnlessRaises(dns.name.NameTooLong, bad)

286

287

def testConcat1(self):

288

n1 = dns.name.Name(['a', 'b'])

289

n2 = dns.name.Name(['c', 'd'])

290

e = dns.name.Name(['a', 'b', 'c', 'd'])

291

r = n1 + n2

292

self.failUnless(r == e)

293

294

def testConcat2(self):

295

n1 = dns.name.Name(['a', 'b'])

296

n2 = dns.name.Name([])

297

e = dns.name.Name(['a', 'b'])

298

r = n1 + n2

299

self.failUnless(r == e)

300

301

def testConcat2(self):

302

n1 = dns.name.Name([])

303

n2 = dns.name.Name(['a', 'b'])

304

e = dns.name.Name(['a', 'b'])

305

r = n1 + n2

306

self.failUnless(r == e)

307

308

def testConcat3(self):

309

n1 = dns.name.Name(['a', 'b', ''])

310

n2 = dns.name.Name([])

311

e = dns.name.Name(['a', 'b', ''])

312

r = n1 + n2

313

self.failUnless(r == e)

314

315

def testConcat4(self):

316

n1 = dns.name.Name(['a', 'b'])

317

n2 = dns.name.Name(['c', ''])

318

e = dns.name.Name(['a', 'b', 'c', ''])

319

r = n1 + n2

320

self.failUnless(r == e)

321

322

def testConcat5(self):

323

def bad():

324

n1 = dns.name.Name(['a', 'b', ''])

325

n2 = dns.name.Name(['c'])

326

r = n1 + n2

327

self.failUnlessRaises(dns.name.AbsoluteConcatenation, bad)

328

329

def testBadEscape(self):

330

def bad():

331

n = dns.name.from_text(r'a.b\0q1.c.')

332

print n

333

self.failUnlessRaises(dns.name.BadEscape, bad)

334

335

def testDigestable1(self):

336

n = dns.name.from_text('FOO.bar')

337

d = n.to_digestable()

338

self.failUnless(d == '\x03foo\x03bar\x00')

339

340

def testDigestable2(self):

341

n1 = dns.name.from_text('FOO.bar')

342

n2 = dns.name.from_text('foo.BAR.')

343

d1 = n1.to_digestable()

344

d2 = n2.to_digestable()

345

self.failUnless(d1 == d2)

346

347

def testDigestable3(self):

348

d = dns.name.root.to_digestable()

349

self.failUnless(d == '\x00')

350

351

def testDigestable4(self):

352

n = dns.name.from_text('FOO.bar', None)

353

d = n.to_digestable(dns.name.root)

354

self.failUnless(d == '\x03foo\x03bar\x00')

355

356

def testBadDigestable(self):

357

def bad():

358

n = dns.name.from_text('FOO.bar', None)

359

d = n.to_digestable()

360

self.failUnlessRaises(dns.name.NeedAbsoluteNameOrOrigin, bad)

361

362

def testToWire1(self):

363

n = dns.name.from_text('FOO.bar')

364

f = cStringIO.StringIO()

365

compress = {}

366

n.to_wire(f, compress)

367

self.failUnless(f.getvalue() == '\x03FOO\x03bar\x00')

368

369

def testToWire2(self):

370

n = dns.name.from_text('FOO.bar')

371

f = cStringIO.StringIO()

372

compress = {}

373

n.to_wire(f, compress)

374

n.to_wire(f, compress)

375

self.failUnless(f.getvalue() == '\x03FOO\x03bar\x00\xc0\x00')

376

377

def testToWire3(self):

378

n1 = dns.name.from_text('FOO.bar')

379

n2 = dns.name.from_text('foo.bar')

380

f = cStringIO.StringIO()

381

compress = {}

382

n1.to_wire(f, compress)

383

n2.to_wire(f, compress)

384

self.failUnless(f.getvalue() == '\x03FOO\x03bar\x00\xc0\x00')

385

386

def testToWire4(self):

387

n1 = dns.name.from_text('FOO.bar')

388

n2 = dns.name.from_text('a.foo.bar')

389

f = cStringIO.StringIO()

390

compress = {}

391

n1.to_wire(f, compress)

392

n2.to_wire(f, compress)

393

self.failUnless(f.getvalue() == '\x03FOO\x03bar\x00\x01\x61\xc0\x00')

394

395

def testToWire5(self):

396

n1 = dns.name.from_text('FOO.bar')

397

n2 = dns.name.from_text('a.foo.bar')

398

f = cStringIO.StringIO()

399

compress = {}

400

n1.to_wire(f, compress)

401

n2.to_wire(f, None)

402

self.failUnless(f.getvalue() == \

403

'\x03FOO\x03bar\x00\x01\x61\x03foo\x03bar\x00')

404

405

def testToWire6(self):

406

n = dns.name.from_text('FOO.bar')

407

v = n.to_wire()

408

self.failUnless(v == '\x03FOO\x03bar\x00')

409

410

def testBadToWire(self):

411

def bad():

412

n = dns.name.from_text('FOO.bar', None)

413

f = cStringIO.StringIO()

414

compress = {}

415

n.to_wire(f, compress)

416

self.failUnlessRaises(dns.name.NeedAbsoluteNameOrOrigin, bad)

417

418

def testSplit1(self):

419

n = dns.name.from_text('foo.bar.')

420

(prefix, suffix) = n.split(2)

421

ep = dns.name.from_text('foo', None)

422

es = dns.name.from_text('bar.', None)

423

self.failUnless(prefix == ep and suffix == es)

424

425

def testSplit2(self):

426

n = dns.name.from_text('foo.bar.')

427

(prefix, suffix) = n.split(1)

428

ep = dns.name.from_text('foo.bar', None)

429

es = dns.name.from_text('.', None)

430

self.failUnless(prefix == ep and suffix == es)

431

432

def testSplit3(self):

433

n = dns.name.from_text('foo.bar.')

434

(prefix, suffix) = n.split(0)

435

ep = dns.name.from_text('foo.bar.', None)

436

es = dns.name.from_text('', None)

437

self.failUnless(prefix == ep and suffix == es)

438

439

def testSplit4(self):

440

n = dns.name.from_text('foo.bar.')

441

(prefix, suffix) = n.split(3)

442

ep = dns.name.from_text('', None)

443

es = dns.name.from_text('foo.bar.', None)

444

self.failUnless(prefix == ep and suffix == es)

445

446

def testBadSplit1(self):

447

def bad():

448

n = dns.name.from_text('foo.bar.')

449

(prefix, suffix) = n.split(-1)

450

self.failUnlessRaises(ValueError, bad)

451

452

def testBadSplit2(self):

453

def bad():

454

n = dns.name.from_text('foo.bar.')

455

(prefix, suffix) = n.split(4)

456

self.failUnlessRaises(ValueError, bad)

457

458

def testRelativize1(self):

459

n = dns.name.from_text('a.foo.bar.', None)

460

o = dns.name.from_text('bar.', None)

461

e = dns.name.from_text('a.foo', None)

462

self.failUnless(n.relativize(o) == e)

463

464

def testRelativize2(self):

465

n = dns.name.from_text('a.foo.bar.', None)

466

o = n

467

e = dns.name.empty

468

self.failUnless(n.relativize(o) == e)

469

470

def testRelativize3(self):

471

n = dns.name.from_text('a.foo.bar.', None)

472

o = dns.name.from_text('blaz.', None)

473

e = n

474

self.failUnless(n.relativize(o) == e)

475

476

def testRelativize4(self):

477

n = dns.name.from_text('a.foo', None)

478

o = dns.name.root

479

e = n

480

self.failUnless(n.relativize(o) == e)

481

482

def testDerelativize1(self):

483

n = dns.name.from_text('a.foo', None)

484

o = dns.name.from_text('bar.', None)

485

e = dns.name.from_text('a.foo.bar.', None)

486

self.failUnless(n.derelativize(o) == e)

487

488

def testDerelativize2(self):

489

n = dns.name.empty

490

o = dns.name.from_text('a.foo.bar.', None)

491

e = o

492

self.failUnless(n.derelativize(o) == e)

493

494

def testDerelativize3(self):

495

n = dns.name.from_text('a.foo.bar.', None)

496

o = dns.name.from_text('blaz.', None)

497

e = n

498

self.failUnless(n.derelativize(o) == e)

499

500

def testChooseRelativity1(self):

501

n = dns.name.from_text('a.foo.bar.', None)

502

o = dns.name.from_text('bar.', None)

503

e = dns.name.from_text('a.foo', None)

504

self.failUnless(n.choose_relativity(o, True) == e)

505

506

def testChooseRelativity2(self):

507

n = dns.name.from_text('a.foo.bar.', None)

508

o = dns.name.from_text('bar.', None)

509

e = n

510

self.failUnless(n.choose_relativity(o, False) == e)

511

512

def testChooseRelativity3(self):

513

n = dns.name.from_text('a.foo', None)

514

o = dns.name.from_text('bar.', None)

515

e = dns.name.from_text('a.foo.bar.', None)

516

self.failUnless(n.choose_relativity(o, False) == e)

517

518

def testChooseRelativity4(self):

519

n = dns.name.from_text('a.foo', None)

520

o = None

521

e = n

522

self.failUnless(n.choose_relativity(o, True) == e)

523

524

def testChooseRelativity5(self):

525

n = dns.name.from_text('a.foo', None)

526

o = None

527

e = n

528

self.failUnless(n.choose_relativity(o, False) == e)

529

530

def testChooseRelativity6(self):

531

n = dns.name.from_text('a.foo.', None)

532

o = None

533

e = n

534

self.failUnless(n.choose_relativity(o, True) == e)

535

536

def testChooseRelativity7(self):

537

n = dns.name.from_text('a.foo.', None)

538

o = None

539

e = n

540

self.failUnless(n.choose_relativity(o, False) == e)

541

542

def testFromWire1(self):

543

w = '\x03foo\x00\xc0\x00'

544

(n1, cused1) = dns.name.from_wire(w, 0)

545

(n2, cused2) = dns.name.from_wire(w, cused1)

546

en1 = dns.name.from_text('foo.')

547

en2 = en1

548

ecused1 = 5

549

ecused2 = 2

550

self.failUnless(n1 == en1 and cused1 == ecused1 and \

551

n2 == en2 and cused2 == ecused2)

552

553

def testFromWire1(self):

554

w = '\x03foo\x00\x01a\xc0\x00\x01b\xc0\x05'

555

current = 0

556

(n1, cused1) = dns.name.from_wire(w, current)

557

current += cused1

558

(n2, cused2) = dns.name.from_wire(w, current)

559

current += cused2

560

(n3, cused3) = dns.name.from_wire(w, current)

561

en1 = dns.name.from_text('foo.')

562

en2 = dns.name.from_text('a.foo.')

563

en3 = dns.name.from_text('b.a.foo.')

564

ecused1 = 5

565

ecused2 = 4

566

ecused3 = 4

567

self.failUnless(n1 == en1 and cused1 == ecused1 and \

568

n2 == en2 and cused2 == ecused2 and \

569

n3 == en3 and cused3 == ecused3)

570

571

def testBadFromWire1(self):

572

def bad():

573

w = '\x03foo\xc0\x04'

574

(n, cused) = dns.name.from_wire(w, 0)

575

self.failUnlessRaises(dns.name.BadPointer, bad)

576

577

def testBadFromWire2(self):

578

def bad():

579

w = '\x03foo\xc0\x05'

580

(n, cused) = dns.name.from_wire(w, 0)

581

self.failUnlessRaises(dns.name.BadPointer, bad)

582

583

def testBadFromWire3(self):

584

def bad():

585

w = '\xbffoo'

586

(n, cused) = dns.name.from_wire(w, 0)

587

self.failUnlessRaises(dns.name.BadLabelType, bad)

588

589

def testBadFromWire4(self):

590

def bad():

591

w = '\x41foo'

592

(n, cused) = dns.name.from_wire(w, 0)

593

self.failUnlessRaises(dns.name.BadLabelType, bad)

594

595

def testParent1(self):

596

n = dns.name.from_text('foo.bar.')

597

self.failUnless(n.parent() == dns.name.from_text('bar.'))

598

self.failUnless(n.parent().parent() == dns.name.root)

599

600

def testParent2(self):

601

n = dns.name.from_text('foo.bar', None)

602

self.failUnless(n.parent() == dns.name.from_text('bar', None))

603

self.failUnless(n.parent().parent() == dns.name.empty)

604

605

def testParent3(self):

606

def bad():

607

n = dns.name.root

608

n.parent()

609

self.failUnlessRaises(dns.name.NoParent, bad)

610

611

def testParent4(self):

612

def bad():

613

n = dns.name.empty

614

n.parent()

615

self.failUnlessRaises(dns.name.NoParent, bad)

616

617

def testFromUnicode1(self):

618

n = dns.name.from_text(u'foo.bar')

619

self.failUnless(n.labels == ('foo', 'bar', ''))

620

621

def testFromUnicode2(self):

622

n = dns.name.from_text(u'foo\u1234bar.bar')

623

self.failUnless(n.labels == ('xn--foobar-r5z', 'bar', ''))

624

625

def testFromUnicodeAlternateDot1(self):

626

n = dns.name.from_text(u'foo\u3002bar')

627

self.failUnless(n.labels == ('foo', 'bar', ''))

628

629

def testFromUnicodeAlternateDot2(self):

630

n = dns.name.from_text(u'foo\uff0ebar')

631

self.failUnless(n.labels == ('foo', 'bar', ''))

632

633

def testFromUnicodeAlternateDot3(self):

634

n = dns.name.from_text(u'foo\uff61bar')

635

self.failUnless(n.labels == ('foo', 'bar', ''))

636

637

def testToUnicode1(self):

638

n = dns.name.from_text(u'foo.bar')

639

s = n.to_unicode()

640

self.failUnless(s == u'foo.bar.')

641

642

def testToUnicode2(self):

643

n = dns.name.from_text(u'foo\u1234bar.bar')

644

s = n.to_unicode()

645

self.failUnless(s == u'foo\u1234bar.bar.')

646

647

def testToUnicode3(self):

648

n = dns.name.from_text('foo.bar')

649

s = n.to_unicode()

650

self.failUnless(s == u'foo.bar.')

651

652

def testReverseIPv4(self):

653

e = dns.name.from_text('1.0.0.127.in-addr.arpa.')

654

n = dns.reversename.from_address('127.0.0.1')

655

self.failUnless(e == n)

656

657

def testReverseIPv6(self):

658

e = dns.name.from_text('1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.')

659

n = dns.reversename.from_address('::1')

660

self.failUnless(e == n)

661

662

def testReverseIPv6MappedIpv4(self):

663

e = dns.name.from_text('1.0.0.127.in-addr.arpa.')

664

n = dns.reversename.from_address('::ffff:127.0.0.1')

665

self.failUnless(e == n)

666

667

def testBadReverseIPv4(self):

668

def bad():

669

n = dns.reversename.from_address('127.0.foo.1')

670

self.failUnlessRaises(dns.exception.SyntaxError, bad)

671

672

def testBadReverseIPv6(self):

673

def bad():

674

n = dns.reversename.from_address('::1::1')

675

self.failUnlessRaises(dns.exception.SyntaxError, bad)

676

677

def testForwardIPv4(self):

678

n = dns.name.from_text('1.0.0.127.in-addr.arpa.')

679

e = '127.0.0.1'

680

text = dns.reversename.to_address(n)

681

self.failUnless(text == e)

682

683

def testForwardIPv6(self):

684

n = dns.name.from_text('1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.')

685

e = '::1'

686

text = dns.reversename.to_address(n)

687

self.failUnless(text == e)

688

689

def testE164ToEnum(self):

690

text = '+1 650 555 1212'

691

e = dns.name.from_text('2.1.2.1.5.5.5.0.5.6.1.e164.arpa.')

692

n = dns.e164.from_e164(text)

693

self.failUnless(n == e)

694

695

def testEnumToE164(self):

696

n = dns.name.from_text('2.1.2.1.5.5.5.0.5.6.1.e164.arpa.')

697

e = '+16505551212'

698

text = dns.e164.to_e164(n)

699

self.failUnless(text == e)

700

701

if __name__ == '__main__':

702

unittest.main()