46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from subiquitycore.netplan import configs_in_root
|
||
|
|
from subiquitycore.tests import SubiTestCase, populate_dir
|
||
|
|
|
||
|
|
|
||
|
|
class TestConfigsInRoot(SubiTestCase):
|
||
|
|
def test_masked_true(self):
|
||
|
|
"""configs_in_root masked=False should not return masked files."""
|
||
|
|
my_dir = self.tmp_dir()
|
||
|
|
unmasked = [
|
||
|
|
"run/netplan/00base.yaml",
|
||
|
|
"lib/netplan/01system.yaml",
|
||
|
|
"etc/netplan/99end.yaml",
|
||
|
|
]
|
||
|
|
masked = ["etc/netplan/00base.yaml"]
|
||
|
|
populate_dir(my_dir, {f: "key: here\n" for f in unmasked + masked})
|
||
|
|
self.assertEqual(
|
||
|
|
[os.path.join(my_dir, p) for p in unmasked], configs_in_root(my_dir)
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_masked_false(self):
|
||
|
|
"""configs_in_root mask=True should return all configs."""
|
||
|
|
my_dir = self.tmp_dir()
|
||
|
|
yamls = [
|
||
|
|
"etc/netplan/00base.yaml",
|
||
|
|
"run/netplan/00base.yaml",
|
||
|
|
"lib/netplan/01system.yaml",
|
||
|
|
"etc/netplan/99end.yaml",
|
||
|
|
]
|
||
|
|
populate_dir(my_dir, {f: "someyaml: here\n" for f in yamls})
|
||
|
|
self.assertEqual(
|
||
|
|
[os.path.join(my_dir, p) for p in yamls],
|
||
|
|
configs_in_root(my_dir, masked=True),
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_only_includes_yaml(self):
|
||
|
|
"""configs_in_root should only return *.yaml files."""
|
||
|
|
my_dir = self.tmp_dir()
|
||
|
|
yamls = ["etc/netplan/00base.yaml", "etc/netplan/99end.yaml"]
|
||
|
|
nonyamls = ["etc/netplan/ignored.yaml.dist", "run/netplan/my.cfg"]
|
||
|
|
populate_dir(my_dir, {f: "someyaml: here\n" for f in yamls + nonyamls})
|
||
|
|
self.assertEqual(
|
||
|
|
[os.path.join(my_dir, p) for p in yamls], configs_in_root(my_dir)
|
||
|
|
)
|