def test_ok(self, manifest_dir):
"""Test loading the manifests into API objects successfully."""
objs = manifest.load_path(os.path.join(manifest_dir, 'manifests'))
assert len(objs) == 3
for obj in objs:
assert obj.kind in ['Deployment', 'ConfigMap', 'Service']
def apply_manifest_from_marker(item, client):
"""Load manifests and create the API objects for the specified files.
This gets called for every `pytest.mark.applymanifests` marker on
test cases.
Once a manifest has been loaded, the API objects will be registered
with the test cases' TestMeta. This allows some easier control via
the "kube" fixture, such as waiting for all objects to be created.
Args:
item: The pytest test item.
client (manager.TestMeta): The metainfo object for the marked
test case.
"""
for mark in item.iter_markers(name='applymanifests'):
dir_path = mark.args[0]
files = mark.kwargs.get('files')
# We expect the path specified to either be absolute or relative
# from the test file. If the path is relative, add the directory
# that the test file resides in as a prefix to the dir_path.
if not os.path.isabs(dir_path):
dir_path = os.path.abspath(
os.path.join(os.path.dirname(item.fspath), dir_path))
# If there are any files specified, we will only load those files.
# Otherwise, we'll load everything in the directory.
if files is None:
objs = load_path(dir_path)
else:
objs = [
obj for objs in load_file(os.path.join(dir_path, f))
for f in files
]
# For each of the loaded Kubernetes resources, we'll want to wrap it
# in the equivalent kubetest wrapper. If the resource does not have
# an equivalent kubetest wrapper, error out. We cannot reliably create
# the resource without our ApiObject wrapper semantics.
wrapped = []
for obj in objs:
found = False
for klass in ApiObject.__subclasses__():
if obj.kind == klass.__name__:
wrapped.append(klass(obj))
found = True
break
if not found:
raise ValueError(
'Unable to match loaded object to an internal wrapper '
'class: {}'.format(obj))
client.register_objects(wrapped)
def test_invalid_yaml(self, manifest_dir):
"""Test loading manifests when one of the files has invalid YAML."""
with pytest.raises(yaml.YAMLError):
manifest.load_path(manifest_dir)
def test_empty_dir(self, tmpdir):
"""Test loading manifests from an empty directory."""
d = tmpdir.mkdir('foo')
objs = manifest.load_path(d)
assert len(objs) == 0
def test_no_dir(self):
"""Test loading manifests when the specified path is not a directory."""
with pytest.raises(ValueError):
manifest.load_path('foobar')
def apply_manifests_from_marker(item: pytest.Item,
meta: manager.TestMeta) -> None:
"""Load manifests and create the API objects for the specified files.
This gets called for every `pytest.mark.applymanifests` marker on test cases.
Once a manifest has been loaded, the API objects will be registered with the
test cases' TestMeta. This allows some easier control via the "kube" fixture,
such as waiting for all objects to be created.
Args:
item: The pytest test item.
meta: The metainfo object for the marked test case.
"""
item_renderer = get_manifest_renderer_for_item(item)
for mark in item.iter_markers(name="applymanifests"):
dir_path = mark.args[0]
files = mark.kwargs.get("files")
renderer = mark.kwargs.get("renderer", item_renderer)
if not callable(renderer):
raise TypeError("renderer given is not callable")
# We expect the path specified to either be absolute or relative
# from the test file. If the path is relative, add the directory
# that the test file resides in as a prefix to the dir_path.
if not os.path.isabs(dir_path):
dir_path = os.path.abspath(
os.path.join(os.path.dirname(item.fspath), dir_path))
# Setup template rendering context
context = dict(
dir_path=dir_path,
namespace=meta.ns,
test_node_id=meta.node_id,
test_name=meta.name,
)
context_renderer = ContextRenderer(renderer, context)
# If there are any files specified, we will only load those files.
# Otherwise, we'll load everything in the directory.
if files is None:
objs = load_path(dir_path, renderer=context_renderer)
else:
objs = []
context_renderer.context["objs"] = objs
for f in files:
objs.extend(
load_file(os.path.join(dir_path, f),
renderer=context_renderer))
# For each of the loaded Kubernetes resources, we'll want to wrap it
# in the equivalent kubetest wrapper. If the resource does not have
# an equivalent kubetest wrapper, error out. We cannot reliably create
# the resource without our ApiObject wrapper semantics.
wrapped = []
for obj in objs:
found = False
for klass in ApiObject.__subclasses__():
if obj.kind == klass.__name__:
wrapped.append(klass(obj))
found = True
break
if not found:
raise ValueError(
f"Unable to match loaded object to an internal wrapper class: {obj}",
)
meta.register_objects(wrapped)
