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())