def test_get_templates(self, *args):
"""test get_templates"""
kv_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3'], [{}, {}], [{}, {}]],
}
auto_tpls = {
'foo': [['check3', 'check5'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check6'], [{}, {}], [{}, {}]],
'foobar': [['check4'], [{}], [{}]],
}
cache = _TemplateCache(issue_read, '')
cache.kv_templates = kv_tpls
cache.auto_conf_templates = auto_tpls
self.assertEquals(cache.get_templates('foo'),
{CONFIG_FROM_TEMPLATE: [['check0', 'check1'], [{}, {}], [{}, {}]],
CONFIG_FROM_AUTOCONF: [['check3', 'check5'], [{}, {}], [{}, {}]]}
)
self.assertEquals(cache.get_templates('bar'),
# check3 must come from template not autoconf
{CONFIG_FROM_TEMPLATE: [['check2', 'check3'], [{}, {}], [{}, {}]],
CONFIG_FROM_AUTOCONF: [['check6'], [{}], [{}]]}
)
self.assertEquals(cache.get_templates('foobar'),
{CONFIG_FROM_TEMPLATE: None,
CONFIG_FROM_AUTOCONF: [['check4'], [{}], [{}]]}
)
self.assertEquals(cache.get_templates('baz'), None)
def test_get_templates(self, *args):
"""test get_templates"""
kv_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3'], [{}, {}], [{}, {}]],
}
auto_tpls = {
'foo': [['check3', 'check5'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check6'], [{}, {}], [{}, {}]],
'foobar': [['check4'], [{}], [{}]],
}
cache = _TemplateCache(issue_read, '')
cache.kv_templates = kv_tpls
cache.auto_conf_templates = auto_tpls
self.assertEquals(cache.get_templates('foo'),
{CONFIG_FROM_TEMPLATE: [['check0', 'check1'], [{}, {}], [{}, {}]],
CONFIG_FROM_AUTOCONF: [['check3', 'check5'], [{}, {}], [{}, {}]]}
)
self.assertEquals(cache.get_templates('bar'),
# check3 must come from template not autoconf
{CONFIG_FROM_TEMPLATE: [['check2', 'check3'], [{}, {}], [{}, {}]],
CONFIG_FROM_AUTOCONF: [['check6'], [{}], [{}]]}
)
self.assertEquals(cache.get_templates('foobar'),
{CONFIG_FROM_TEMPLATE: None,
CONFIG_FROM_AUTOCONF: [['check4'], [{}], [{}]]}
)
self.assertEquals(cache.get_templates('baz'), None)
def test_populate_auto_conf(self, mock_get_auto_conf_images):
"""test _populate_auto_conf"""
auto_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3', 'check3'], [{}, {}, {}], [{}, {'foo': 'bar'}, {'bar': 'foo'}]],
}
cache = _TemplateCache(issue_read, '')
cache.auto_conf_templates = defaultdict(lambda: [[]] * 3)
mock_get_auto_conf_images.return_value = auto_tpls
cache._populate_auto_conf()
self.assertEquals(cache.auto_conf_templates['foo'], auto_tpls['foo'])
self.assertEquals(cache.auto_conf_templates['bar'],
[['check2', 'check3'], [{}, {}], [{}, {'foo': 'bar'}]])
def test_populate_auto_conf(self, mock_get_auto_conf_images):
"""test _populate_auto_conf"""
auto_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3', 'check3'], [{}, {}, {}], [{}, {'foo': 'bar'}, {'bar': 'foo'}]],
}
cache = _TemplateCache(issue_read, '')
cache.auto_conf_templates = defaultdict(lambda: [[]] * 3)
mock_get_auto_conf_images.return_value = auto_tpls
cache._populate_auto_conf()
self.assertEquals(cache.auto_conf_templates['foo'], auto_tpls['foo'])
self.assertEquals(cache.auto_conf_templates['bar'],
[['check2', 'check3'], [{}, {}], [{}, {'foo': 'bar'}]])
def test_get_check_names(self, mock_get_auto_confd_path):
"""Test get_check_names"""
kv_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3'], [{}, {}], [{}, {}]],
}
auto_tpls = {
'foo': [['check4', 'check5'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check6'], [{}, {}], [{}, {}]],
'foobar': None,
}
cache = _TemplateCache(issue_read, '')
cache.kv_templates = kv_tpls
cache.auto_conf_templates = auto_tpls
self.assertEquals(cache.get_check_names('foo'), set(['check0', 'check1', 'check4', 'check5']))
self.assertEquals(cache.get_check_names('bar'), set(['check2', 'check3', 'check6']))
self.assertEquals(cache.get_check_names('foobar'), set())
self.assertEquals(cache.get_check_names('baz'), set())
def test_get_check_names(self, mock_get_auto_confd_path):
"""Test get_check_names"""
kv_tpls = {
'foo': [['check0', 'check1'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check3'], [{}, {}], [{}, {}]],
}
auto_tpls = {
'foo': [['check4', 'check5'], [{}, {}], [{}, {}]],
'bar': [['check2', 'check6'], [{}, {}], [{}, {}]],
'foobar': None,
}
cache = _TemplateCache(issue_read, '')
cache.kv_templates = kv_tpls
cache.auto_conf_templates = auto_tpls
self.assertEquals(cache.get_check_names('foo'), set(['check0', 'check1', 'check4', 'check5']))
self.assertEquals(cache.get_check_names('bar'), set(['check2', 'check3', 'check6']))
self.assertEquals(cache.get_check_names('foobar'), set())
self.assertEquals(cache.get_check_names('baz'), set())
