|
Packit Service |
a04d08 |
# This file is part of cloud-init. See LICENSE file for license information.
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
from cloudinit import util
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
from cloudinit.tests.helpers import TestCase, populate_dir
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
import shutil
|
|
Packit Service |
a04d08 |
import tempfile
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
class TestPathPrefix2Dict(TestCase):
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def setUp(self):
|
|
Packit Service |
a04d08 |
super(TestPathPrefix2Dict, self).setUp()
|
|
Packit Service |
a04d08 |
self.tmp = tempfile.mkdtemp()
|
|
Packit Service |
a04d08 |
self.addCleanup(shutil.rmtree, self.tmp)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_required_only(self):
|
|
Packit Service |
a04d08 |
dirdata = {'f1': b'f1content', 'f2': b'f2content'}
|
|
Packit Service |
a04d08 |
populate_dir(self.tmp, dirdata)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
ret = util.pathprefix2dict(self.tmp, required=['f1', 'f2'])
|
|
Packit Service |
a04d08 |
self.assertEqual(dirdata, ret)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_required_missing(self):
|
|
Packit Service |
a04d08 |
dirdata = {'f1': b'f1content'}
|
|
Packit Service |
a04d08 |
populate_dir(self.tmp, dirdata)
|
|
Packit Service |
a04d08 |
kwargs = {'required': ['f1', 'f2']}
|
|
Packit Service |
a04d08 |
self.assertRaises(ValueError, util.pathprefix2dict, self.tmp, **kwargs)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_no_required_and_optional(self):
|
|
Packit Service |
a04d08 |
dirdata = {'f1': b'f1c', 'f2': b'f2c'}
|
|
Packit Service |
a04d08 |
populate_dir(self.tmp, dirdata)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
ret = util.pathprefix2dict(self.tmp, required=None,
|
|
Packit Service |
a04d08 |
optional=['f1', 'f2'])
|
|
Packit Service |
a04d08 |
self.assertEqual(dirdata, ret)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_required_and_optional(self):
|
|
Packit Service |
a04d08 |
dirdata = {'f1': b'f1c', 'f2': b'f2c'}
|
|
Packit Service |
a04d08 |
populate_dir(self.tmp, dirdata)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
ret = util.pathprefix2dict(self.tmp, required=['f1'], optional=['f2'])
|
|
Packit Service |
a04d08 |
self.assertEqual(dirdata, ret)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
# vi: ts=4 expandtab
|