summaryrefslogtreecommitdiff
path: root/tests/unittests
diff options
context:
space:
mode:
authorScott Moser <smoser@ubuntu.com>2014-01-27 13:05:11 -0500
committerScott Moser <smoser@ubuntu.com>2014-01-27 13:05:11 -0500
commit50c744fc4ebc0de5fc7fdfee6a874cb9cc62bba8 (patch)
tree26818911da3eef64c076a9b2f4071bf75147c090 /tests/unittests
parentf405a5283ede1e0af09eb8bf0dbe5046680641b5 (diff)
downloadvyos-cloud-init-50c744fc4ebc0de5fc7fdfee6a874cb9cc62bba8.tar.gz
vyos-cloud-init-50c744fc4ebc0de5fc7fdfee6a874cb9cc62bba8.zip
add 'pathprefix2dict' utility for use by DataSourceNoCloud
Diffstat (limited to 'tests/unittests')
-rw-r--r--tests/unittests/test_pathprefix2dict.py40
1 files changed, 40 insertions, 0 deletions
diff --git a/tests/unittests/test_pathprefix2dict.py b/tests/unittests/test_pathprefix2dict.py
new file mode 100644
index 00000000..c68c263c
--- /dev/null
+++ b/tests/unittests/test_pathprefix2dict.py
@@ -0,0 +1,40 @@
+from cloudinit import util
+
+from mocker import MockerTestCase
+from tests.unittests.helpers import populate_dir
+
+
+class TestPathPrefix2Dict(MockerTestCase):
+
+ def setUp(self):
+ self.tmp = self.makeDir()
+
+ def test_required_only(self):
+ dirdata = {'f1': 'f1content', 'f2': 'f2content'}
+ populate_dir(self.tmp, dirdata)
+
+ ret = util.pathprefix2dict(self.tmp, required=['f1', 'f2'])
+ self.assertEqual(dirdata, ret)
+
+ def test_required_missing(self):
+ dirdata = {'f1': 'f1content'}
+ populate_dir(self.tmp, dirdata)
+ kwargs = {'required': ['f1', 'f2']}
+ self.assertRaises(ValueError, util.pathprefix2dict, self.tmp, **kwargs)
+
+ def test_no_required_and_optional(self):
+ dirdata = {'f1': 'f1c', 'f2': 'f2c'}
+ populate_dir(self.tmp, dirdata)
+
+ ret = util.pathprefix2dict(self.tmp, required=None,
+ optional=['f1', 'f2'])
+ self.assertEqual(dirdata, ret)
+
+ def test_required_and_optional(self):
+ dirdata = {'f1': 'f1c', 'f2': 'f2c'}
+ populate_dir(self.tmp, dirdata)
+
+ ret = util.pathprefix2dict(self.tmp, required=['f1'], optional=['f2'])
+ self.assertEqual(dirdata, ret)
+
+# vi: ts=4 expandtab