summaryrefslogtreecommitdiff
path: root/tests/unittests
diff options
context:
space:
mode:
authorMartin Packman <martin.packman@canonical.com>2012-03-08 17:56:10 +0000
committerMartin Packman <martin.packman@canonical.com>2012-03-08 17:56:10 +0000
commit4877a389190f6598e13a0704e77d1c1586d25fb4 (patch)
tree1601bf7de62f7368ee0ebe4c9b926eac832ec0a2 /tests/unittests
parent54b593a166c4ac4043615dddc94a941ebc712300 (diff)
downloadvyos-cloud-init-4877a389190f6598e13a0704e77d1c1586d25fb4.tar.gz
vyos-cloud-init-4877a389190f6598e13a0704e77d1c1586d25fb4.zip
Add tests for writing of userdata parts to the filesystem
Diffstat (limited to 'tests/unittests')
-rw-r--r--tests/unittests/test_userdata.py60
1 files changed, 60 insertions, 0 deletions
diff --git a/tests/unittests/test_userdata.py b/tests/unittests/test_userdata.py
new file mode 100644
index 00000000..949f36dc
--- /dev/null
+++ b/tests/unittests/test_userdata.py
@@ -0,0 +1,60 @@
+"""Tests for handling of userdata within cloud init"""
+
+import logging
+import StringIO
+
+from mocker import MockerTestCase
+
+import cloudinit
+from cloudinit.DataSource import DataSource
+
+
+instance_id = "i-testing"
+
+
+class FakeDataSource(DataSource):
+
+ def __init__(self, userdata):
+ self.metadata = {'instance-id': instance_id}
+ self.userdata_raw = userdata
+
+
+class TestConsumeUserData(MockerTestCase):
+
+ def setUp(self):
+ self.mock_write = self.mocker.replace("cloudinit.util.write_file",
+ passthrough=False)
+ self.mock_write(self.get_ipath("cloud_config"), "", 0600)
+ self.capture_log()
+
+ def tearDown(self):
+ self._log.removeHandler(self._log_handler)
+
+ @staticmethod
+ def get_ipath(name):
+ return "%s/instances/%s%s" % (cloudinit.varlibdir, instance_id,
+ cloudinit.pathmap[name])
+
+ def capture_log(self):
+ self.log_file = StringIO.StringIO()
+ self._log_handler = logging.StreamHandler(self.log_file)
+ self._log_handler.setLevel(logging.DEBUG)
+ self._log = logging.getLogger(cloudinit.logger_name)
+ self._log.addHandler(self._log_handler)
+
+ def test_script(self):
+ script = "#!/bin/sh\necho hello\n"
+ outpath = cloudinit.get_ipath_cur("scripts") + "/part-001"
+ self.mock_write(outpath, script, 0700)
+ self.mocker.replay()
+ ci = cloudinit.CloudInit()
+ ci.datasource = FakeDataSource(script)
+ ci.consume_userdata()
+ self.assertEqual("", self.log_file.getvalue())
+
+ def test_unhandled_type_warning(self):
+ self.mocker.replay()
+ ci = cloudinit.CloudInit()
+ ci.datasource = FakeDataSource("arbitrary text\n")
+ ci.consume_userdata()
+ self.assertIn("Unhandled userdata part", self.log_file.getvalue())