import ceph quincy 17.2.4

[ceph.git] / ceph / src / pybind / mgr / cephadm / tests / test_template.py

import pathlib

import pytest

from cephadm.template import TemplateMgr, UndefinedError, TemplateNotFoundError


def test_render(cephadm_module, fs):
    template_base = (pathlib.Path(__file__).parent / '../templates').resolve()
    fake_template = template_base / 'foo/bar'
    fs.create_file(fake_template, contents='{{ cephadm_managed }}{{ var }}')

    template_mgr = TemplateMgr(cephadm_module)
    value = 'test'

    # with base context
    expected_text = '{}{}'.format(template_mgr.base_context['cephadm_managed'], value)
    assert template_mgr.render('foo/bar', {'var': value}) == expected_text

    # without base context
    with pytest.raises(UndefinedError):
        template_mgr.render('foo/bar', {'var': value}, managed_context=False)

    # override the base context
    context = {
        'cephadm_managed': 'abc',
        'var': value
    }
    assert template_mgr.render('foo/bar', context) == 'abc{}'.format(value)

    # template not found
    with pytest.raises(TemplateNotFoundError):
        template_mgr.render('foo/bar/2', {})