summaryrefslogtreecommitdiff
path: root/cloudinit/tests/test_util.py
diff options
context:
space:
mode:
Diffstat (limited to 'cloudinit/tests/test_util.py')
-rw-r--r--cloudinit/tests/test_util.py23
1 files changed, 23 insertions, 0 deletions
diff --git a/cloudinit/tests/test_util.py b/cloudinit/tests/test_util.py
index ba6bf699..c3e2e404 100644
--- a/cloudinit/tests/test_util.py
+++ b/cloudinit/tests/test_util.py
@@ -44,3 +44,26 @@ class TestUtil(CiTestCase):
m_mount_info.return_value = ('/dev/sda1', 'btrfs', '/', 'ro,relatime')
is_rw = util.mount_is_read_write('/')
self.assertEqual(is_rw, False)
+
+
+class TestShellify(CiTestCase):
+
+ def test_input_dict_raises_type_error(self):
+ self.assertRaisesRegex(
+ TypeError, 'Input.*was.*dict.*xpected',
+ util.shellify, {'mykey': 'myval'})
+
+ def test_input_str_raises_type_error(self):
+ self.assertRaisesRegex(
+ TypeError, 'Input.*was.*str.*xpected', util.shellify, "foobar")
+
+ def test_value_with_int_raises_type_error(self):
+ self.assertRaisesRegex(
+ TypeError, 'shellify.*int', util.shellify, ["foo", 1])
+
+ def test_supports_strings_and_lists(self):
+ self.assertEqual(
+ '\n'.join(["#!/bin/sh", "echo hi mom", "'echo' 'hi dad'",
+ "'echo' 'hi' 'sis'", ""]),
+ util.shellify(["echo hi mom", ["echo", "hi dad"],
+ ('echo', 'hi', 'sis')]))