From 1951061b8673147630a9cf2cb7960fdc1d884569 Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Fri, 29 Jun 2012 11:42:03 -0700
Subject: Add a test that ensures the upstart handler does not output files
 when in non-per-instance mode.

---
 tests/unittests/test_builtin_handlers.py | 54 ++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 tests/unittests/test_builtin_handlers.py

(limited to 'tests')

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)))
-- 
cgit v1.2.3