1
from mocker import MockerTestCase
3
from cloudinit.distros.parsers import resolv_conf
9
; generated by /sbin/dhclient-script
10
search blah.yahoo.com yahoo.com
11
nameserver 10.15.44.14
12
nameserver 10.15.30.92
14
BASE_RESOLVE = BASE_RESOLVE.strip()
17
class TestResolvHelper(MockerTestCase):
18
def test_parse_same(self):
19
rp = resolv_conf.ResolvConf(BASE_RESOLVE)
20
rp_r = str(rp).strip()
21
self.assertEquals(BASE_RESOLVE, rp_r)
23
def test_local_domain(self):
24
rp = resolv_conf.ResolvConf(BASE_RESOLVE)
25
self.assertEquals(None, rp.local_domain)
27
rp.local_domain = "bob"
28
self.assertEquals('bob', rp.local_domain)
29
self.assertIn('domain bob', str(rp))
31
def test_nameservers(self):
32
rp = resolv_conf.ResolvConf(BASE_RESOLVE)
33
self.assertIn('10.15.44.14', rp.nameservers)
34
self.assertIn('10.15.30.92', rp.nameservers)
35
rp.add_nameserver('10.2')
36
self.assertIn('10.2', rp.nameservers)
37
self.assertIn('nameserver 10.2', str(rp))
38
self.assertNotIn('10.3', rp.nameservers)
39
self.assertEquals(len(rp.nameservers), 3)
40
rp.add_nameserver('10.2')
41
self.assertRaises(ValueError, rp.add_nameserver, '10.3')
42
self.assertNotIn('10.3', rp.nameservers)
44
def test_search_domains(self):
45
rp = resolv_conf.ResolvConf(BASE_RESOLVE)
46
self.assertIn('yahoo.com', rp.search_domains)
47
self.assertIn('blah.yahoo.com', rp.search_domains)
48
rp.add_search_domain('bbb.y.com')
49
self.assertIn('bbb.y.com', rp.search_domains)
50
self.assertTrue(re.search(r'search(.*)bbb.y.com(.*)', str(rp)))
51
self.assertIn('bbb.y.com', rp.search_domains)
52
rp.add_search_domain('bbb.y.com')
53
self.assertEquals(len(rp.search_domains), 3)
54
rp.add_search_domain('bbb2.y.com')
55
self.assertEquals(len(rp.search_domains), 4)
56
rp.add_search_domain('bbb3.y.com')
57
self.assertEquals(len(rp.search_domains), 5)
58
rp.add_search_domain('bbb4.y.com')
59
self.assertEquals(len(rp.search_domains), 6)
60
self.assertRaises(ValueError, rp.add_search_domain, 'bbb5.y.com')
61
self.assertEquals(len(rp.search_domains), 6)