Add a new example test that will patch utils and os
functions so that they can be 'retargeted' to a temporary directory, which allows us the ability to run a full set of cloud-init stages. Neat things: 1. All cloud-init code is unchanged (as long as it goes through the utils functions for most functionality) 2. Allows for a natural way to setup a temporary directory then patch the new directory as the new 'root' and then run cloud-init stages and then check the contents of what was placed as desired.
This commit is contained in:
parent
db7c4050b8
commit
cc875f9963
3
tests/data/roots/simple_ubuntu/etc/networks/interfaces
Normal file
3
tests/data/roots/simple_ubuntu/etc/networks/interfaces
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
auto lo
|
||||||
|
iface lo inet loopback
|
||||||
|
|
@ -1,8 +1,42 @@
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
|
import mocker
|
||||||
|
|
||||||
from mocker import MockerTestCase
|
from mocker import MockerTestCase
|
||||||
|
|
||||||
from cloudinit import helpers as ch
|
from cloudinit import helpers as ch
|
||||||
|
from cloudinit import util
|
||||||
|
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
# Makes the old path start
|
||||||
|
# with new base instead of whatever
|
||||||
|
# it previously had
|
||||||
|
def rebase_path(old_path, new_base):
|
||||||
|
if old_path.startswith(new_base):
|
||||||
|
# Already handled...
|
||||||
|
return old_path
|
||||||
|
# Retarget the base of that path
|
||||||
|
# to the new base instead of the
|
||||||
|
# old one...
|
||||||
|
path = os.path.join(new_base, old_path.lstrip("/"))
|
||||||
|
path = os.path.abspath(path)
|
||||||
|
return path
|
||||||
|
|
||||||
|
|
||||||
|
# Can work on anything that takes a path
|
||||||
|
# as first argument
|
||||||
|
def retarget_many_wrapper(new_base, am, old_func):
|
||||||
|
def wrapper(*args, **kwds):
|
||||||
|
n_args = list(args)
|
||||||
|
nam = am
|
||||||
|
if am == -1:
|
||||||
|
nam = len(n_args)
|
||||||
|
for i in range(0, nam):
|
||||||
|
path = args[i]
|
||||||
|
n_args[i] = rebase_path(path, new_base)
|
||||||
|
return old_func(*n_args, **kwds)
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
class ResourceUsingTestCase(MockerTestCase):
|
class ResourceUsingTestCase(MockerTestCase):
|
||||||
@ -40,3 +74,75 @@ class ResourceUsingTestCase(MockerTestCase):
|
|||||||
'templates_dir': self.resourceLocation(),
|
'templates_dir': self.resourceLocation(),
|
||||||
})
|
})
|
||||||
return cp
|
return cp
|
||||||
|
|
||||||
|
|
||||||
|
class FilesystemMockingTestCase(ResourceUsingTestCase):
|
||||||
|
def __init__(self, methodName="runTest"):
|
||||||
|
ResourceUsingTestCase.__init__(self, methodName)
|
||||||
|
self.patched_funcs = []
|
||||||
|
|
||||||
|
def replicateTestRoot(self, example_root, target_root):
|
||||||
|
real_root = self.resourceLocation()
|
||||||
|
real_root = os.path.join(real_root, 'roots', example_root)
|
||||||
|
for (dir_path, dirnames, filenames) in os.walk(real_root):
|
||||||
|
real_path = dir_path
|
||||||
|
make_path = rebase_path(real_path[len(real_root):], target_root)
|
||||||
|
util.ensure_dir(make_path)
|
||||||
|
for f in filenames:
|
||||||
|
real_path = util.abs_join(real_path, f)
|
||||||
|
make_path = util.abs_join(make_path, f)
|
||||||
|
shutil.copy(real_path, make_path)
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
self.restore()
|
||||||
|
ResourceUsingTestCase.tearDown(self)
|
||||||
|
|
||||||
|
def restore(self):
|
||||||
|
for (mod, f, func) in self.patched_funcs:
|
||||||
|
setattr(mod, f, func)
|
||||||
|
self.patched_funcs = []
|
||||||
|
|
||||||
|
def patchUtils(self, new_root):
|
||||||
|
patch_funcs = {
|
||||||
|
util: [('write_file', 1),
|
||||||
|
('load_file', 1),
|
||||||
|
('ensure_dir', 1),
|
||||||
|
('chmod', 1),
|
||||||
|
('delete_dir_contents', 1),
|
||||||
|
('del_file', 1),
|
||||||
|
('sym_link', -1)],
|
||||||
|
}
|
||||||
|
for (mod, funcs) in patch_funcs.items():
|
||||||
|
for (f, am) in funcs:
|
||||||
|
func = getattr(mod, f)
|
||||||
|
trap_func = retarget_many_wrapper(new_root, am, func)
|
||||||
|
setattr(mod, f, trap_func)
|
||||||
|
self.patched_funcs.append((mod, f, func))
|
||||||
|
|
||||||
|
# Handle subprocess calls
|
||||||
|
func = getattr(util, 'subp')
|
||||||
|
|
||||||
|
def nsubp(*args, **kwargs):
|
||||||
|
return ('', '')
|
||||||
|
|
||||||
|
setattr(util, 'subp', nsubp)
|
||||||
|
self.patched_funcs.append((util, 'subp', func))
|
||||||
|
|
||||||
|
def null_func(*args, **kwargs):
|
||||||
|
return None
|
||||||
|
|
||||||
|
for f in ['chownbyid', 'chownbyname']:
|
||||||
|
func = getattr(util, f)
|
||||||
|
setattr(util, f, null_func)
|
||||||
|
self.patched_funcs.append((util, f, func))
|
||||||
|
|
||||||
|
def patchOS(self, new_root):
|
||||||
|
patch_funcs = {
|
||||||
|
os.path: ['isfile', 'exists', 'islink', 'isdir'],
|
||||||
|
}
|
||||||
|
for (mod, funcs) in patch_funcs.items():
|
||||||
|
for f in funcs:
|
||||||
|
func = getattr(mod, f)
|
||||||
|
trap_func = retarget_many_wrapper(new_root, 1, func)
|
||||||
|
setattr(mod, f, trap_func)
|
||||||
|
self.patched_funcs.append((mod, f, func))
|
||||||
|
89
tests/unittests/test_runs/test_simple_run.py
Normal file
89
tests/unittests/test_runs/test_simple_run.py
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
from mocker import MockerTestCase
|
||||||
|
|
||||||
|
import mocker
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
# Allow running this test individually
|
||||||
|
top_dir = os.path.join(os.path.dirname(__file__), os.pardir, "helpers.py")
|
||||||
|
top_dir = os.path.abspath(top_dir)
|
||||||
|
if os.path.exists(top_dir):
|
||||||
|
sys.path.insert(0, os.path.dirname(top_dir))
|
||||||
|
|
||||||
|
|
||||||
|
import helpers
|
||||||
|
|
||||||
|
from cloudinit import util
|
||||||
|
from cloudinit import stages
|
||||||
|
|
||||||
|
from cloudinit.settings import (PER_INSTANCE)
|
||||||
|
|
||||||
|
|
||||||
|
class TestSimpleRunDistro(helpers.FilesystemMockingTestCase):
|
||||||
|
def _patchIn(self, root):
|
||||||
|
self.restore()
|
||||||
|
self.patchOS(root)
|
||||||
|
self.patchUtils(root)
|
||||||
|
|
||||||
|
def _pp_root(self, root, repatch=True):
|
||||||
|
self.restore()
|
||||||
|
for (dirpath, dirnames, filenames) in os.walk(root):
|
||||||
|
print(dirpath)
|
||||||
|
for f in filenames:
|
||||||
|
joined = os.path.join(dirpath, f)
|
||||||
|
if os.path.islink(joined):
|
||||||
|
print("f %s - (symlink)" % (f))
|
||||||
|
else:
|
||||||
|
print("f %s" % (f))
|
||||||
|
for d in dirnames:
|
||||||
|
joined = os.path.join(dirpath, d)
|
||||||
|
if os.path.islink(joined):
|
||||||
|
print("d %s - (symlink)" % (d))
|
||||||
|
else:
|
||||||
|
print("d %s" % (d))
|
||||||
|
if repatch:
|
||||||
|
self._patchIn(root)
|
||||||
|
|
||||||
|
def test_none_ds(self):
|
||||||
|
new_root = self.makeDir()
|
||||||
|
self.replicateTestRoot('simple_ubuntu', new_root)
|
||||||
|
cfg = {
|
||||||
|
'datasource_list': ['None'],
|
||||||
|
'write_files': [{
|
||||||
|
'path': '/etc/blah.ini',
|
||||||
|
'content': 'blah',
|
||||||
|
'permissions': 0755,
|
||||||
|
}],
|
||||||
|
'cloud_init_modules': ['write-files'],
|
||||||
|
}
|
||||||
|
cloud_cfg = util.yaml_dumps(cfg)
|
||||||
|
util.ensure_dir(os.path.join(new_root, 'etc', 'cloud'))
|
||||||
|
util.write_file(os.path.join(new_root, 'etc',
|
||||||
|
'cloud', 'cloud.cfg'), cloud_cfg)
|
||||||
|
self._patchIn(new_root)
|
||||||
|
|
||||||
|
# Now start verifying whats created
|
||||||
|
initer = stages.Init()
|
||||||
|
initer.read_cfg()
|
||||||
|
initer.initialize()
|
||||||
|
self.assertTrue(os.path.exists("/var/lib/cloud"))
|
||||||
|
for d in ['scripts', 'seed', 'instances', 'handlers', 'sem', 'data']:
|
||||||
|
self.assertTrue(os.path.isdir(os.path.join("/var/lib/cloud", d)))
|
||||||
|
|
||||||
|
initer.fetch()
|
||||||
|
iid = initer.instancify()
|
||||||
|
self.assertEquals(iid, 'iid-datasource-none')
|
||||||
|
initer.update()
|
||||||
|
self.assertTrue(os.path.islink("var/lib/cloud/instance"))
|
||||||
|
|
||||||
|
(ran, results) = initer.cloudify().run('consume_userdata',
|
||||||
|
initer.consume_userdata,
|
||||||
|
args=[PER_INSTANCE],
|
||||||
|
freq=PER_INSTANCE)
|
||||||
|
|
||||||
|
mods = stages.Modules(initer)
|
||||||
|
(which_ran, failures) = mods.run_section('cloud_init_modules')
|
||||||
|
self.assertTrue(os.path.exists('/etc/blah.ini'))
|
||||||
|
self.assertIn('write-files', which_ran)
|
||||||
|
contents = util.load_file('/etc/blah.ini')
|
||||||
|
self.assertEquals(contents, 'blah')
|
Loading…
x
Reference in New Issue
Block a user