summaryrefslogtreecommitdiff
path: root/tests/unittests/test_builtin_handlers.py
diff options
context:
space:
mode:
authorJoshua Harlow <harlowja@yahoo-inc.com>2012-06-29 11:42:03 -0700
committerJoshua Harlow <harlowja@yahoo-inc.com>2012-06-29 11:42:03 -0700
commit1951061b8673147630a9cf2cb7960fdc1d884569 (patch)
tree171007abf3e0896ed8ff592724c6644da96980b1 /tests/unittests/test_builtin_handlers.py
parentbb75af01d9bf802985e21d0b8d42f0855e57953f (diff)
downloadvyos-cloud-init-1951061b8673147630a9cf2cb7960fdc1d884569.tar.gz
vyos-cloud-init-1951061b8673147630a9cf2cb7960fdc1d884569.zip
Add a test that ensures the upstart handler does not output files when in non-per-instance mode.
Diffstat (limited to 'tests/unittests/test_builtin_handlers.py')
-rw-r--r--tests/unittests/test_builtin_handlers.py54
1 files changed, 54 insertions, 0 deletions
diff --git a/tests/unittests/test_builtin_handlers.py b/tests/unittests/test_builtin_handlers.py
new file mode 100644
index 00000000..84d85d4d
--- /dev/null
+++ b/tests/unittests/test_builtin_handlers.py
@@ -0,0 +1,54 @@
+"""Tests of the built-in user data handlers"""
+
+import os
+
+from mocker import MockerTestCase
+
+from cloudinit import handlers
+from cloudinit import helpers
+from cloudinit import util
+
+from cloudinit.handlers import upstart_job
+
+from cloudinit.settings import (PER_ALWAYS, PER_INSTANCE)
+
+
+class TestBuiltins(MockerTestCase):
+
+ def test_upstart_frequency_no_out(self):
+ c_root = self.makeDir()
+ up_root = self.makeDir()
+ paths = helpers.Paths({
+ 'cloud_dir': c_root,
+ 'upstart_dir': up_root,
+ })
+ freq = PER_ALWAYS
+ h = upstart_job.UpstartJobPartHandler(paths)
+ # No files should be written out when
+ # the frequency is ! per-instance
+ h.handle_part('', handlers.CONTENT_START,
+ None, None, None)
+ h.handle_part('blah', 'text/upstart-job',
+ 'test.conf', 'blah', freq)
+ h.handle_part('', handlers.CONTENT_END,
+ None, None, None)
+ self.assertEquals(0, len(os.listdir(up_root)))
+
+ def test_upstart_frequency_single(self):
+ c_root = self.makeDir()
+ up_root = self.makeDir()
+ paths = helpers.Paths({
+ 'cloud_dir': c_root,
+ 'upstart_dir': up_root,
+ })
+ freq = PER_INSTANCE
+ h = upstart_job.UpstartJobPartHandler(paths)
+ # No files should be written out when
+ # the frequency is ! per-instance
+ h.handle_part('', handlers.CONTENT_START,
+ None, None, None)
+ h.handle_part('blah', 'text/upstart-job',
+ 'test.conf', 'blah', freq)
+ h.handle_part('', handlers.CONTENT_END,
+ None, None, None)
+ self.assertEquals(1, len(os.listdir(up_root)))