summaryrefslogtreecommitdiff
path: root/tests/unittests/test_distros/test_resolv.py
diff options
context:
space:
mode:
authorJoshua Harlow <harlowja@yahoo-inc.com>2012-10-10 16:21:22 -0700
committerJoshua Harlow <harlowja@yahoo-inc.com>2012-10-10 16:21:22 -0700
commit80eb53deb9f80694c5fd0e0190197de1ff496716 (patch)
treed67bcef6beca34fe371be003c23007bbd19f7f92 /tests/unittests/test_distros/test_resolv.py
parentf8b71af537aab3aef04a1e5ceba19e90a1445948 (diff)
downloadvyos-cloud-init-80eb53deb9f80694c5fd0e0190197de1ff496716.tar.gz
vyos-cloud-init-80eb53deb9f80694c5fd0e0190197de1ff496716.zip
System config niceness!
1. Move out the old helpers that provided oop access/reading/writing to various standard conf files and place those in parsers instead. 2. Unify the 'update_hostname' which varied very little between distros and make it generic so that subclasses can only provide a couple of functions to obtain the hostname updating functionality 3. Implement that new set of functions in rhel/debian 4. Use the new parsers chop_comment function for similar use cases as well as add a new utils make header function that can be used for configuration files that are newly generated to use (less duplication here of this same thing being done in multiple places. 5. Add in a distro '_apply_hostname' which calls out to the 'hostname' program to set the system hostname (more duplication elimination). 6. Make the 'constant' filenames being written to for configuration by the various distros be instance members instead of string constants 'sprinkled' throughout the code
Diffstat (limited to 'tests/unittests/test_distros/test_resolv.py')
-rw-r--r--tests/unittests/test_distros/test_resolv.py14
1 files changed, 8 insertions, 6 deletions
diff --git a/tests/unittests/test_distros/test_resolv.py b/tests/unittests/test_distros/test_resolv.py
index 5f122833..d947dda0 100644
--- a/tests/unittests/test_distros/test_resolv.py
+++ b/tests/unittests/test_distros/test_resolv.py
@@ -1,6 +1,8 @@
from mocker import MockerTestCase
-from cloudinit.distros import helpers
+from cloudinit.distros.parsers import resolv_conf
+
+import re
BASE_RESOLVE = '''
@@ -14,12 +16,12 @@ BASE_RESOLVE = BASE_RESOLVE.strip()
class TestResolvHelper(MockerTestCase):
def test_parse_same(self):
- rp = helpers.ResolvConf(BASE_RESOLVE)
+ rp = resolv_conf.ResolvConf(BASE_RESOLVE)
rp_r = str(rp).strip()
self.assertEquals(BASE_RESOLVE, rp_r)
def test_local_domain(self):
- rp = helpers.ResolvConf(BASE_RESOLVE)
+ rp = resolv_conf.ResolvConf(BASE_RESOLVE)
self.assertEquals(None, rp.local_domain)
rp.local_domain = "bob"
@@ -27,7 +29,7 @@ class TestResolvHelper(MockerTestCase):
self.assertIn('domain bob', str(rp))
def test_nameservers(self):
- rp = helpers.ResolvConf(BASE_RESOLVE)
+ rp = resolv_conf.ResolvConf(BASE_RESOLVE)
self.assertIn('10.15.44.14', rp.nameservers)
self.assertIn('10.15.30.92', rp.nameservers)
rp.add_nameserver('10.2')
@@ -41,12 +43,12 @@ class TestResolvHelper(MockerTestCase):
self.assertNotIn('10.3', rp.nameservers)
def test_search_domains(self):
- rp = helpers.ResolvConf(BASE_RESOLVE)
+ rp = resolv_conf.ResolvConf(BASE_RESOLVE)
self.assertIn('yahoo.com', rp.search_domains)
self.assertIn('blah.yahoo.com', rp.search_domains)
rp.add_search_domain('bbb.y.com')
self.assertIn('bbb.y.com', rp.search_domains)
- self.assertRegexpMatches(str(rp), r'search(.*)bbb.y.com(.*)')
+ self.assertTrue(re.search(r'search(.*)bbb.y.com(.*)', str(rp)))
self.assertIn('bbb.y.com', rp.search_domains)
rp.add_search_domain('bbb.y.com')
self.assertEquals(len(rp.search_domains), 3)