def test__make_virtual_c_dict(self): data = 'str' self.assertEqual(config._make_virtual_c_dict(cfg, data), data) config._config__copy_t_dict(cfg) data = {'r0': {'interfaces': {'r0r1_2': {'pic': 'ge-1/0/7', 'management': 0, 'name': 'ge-1/0/7.0', 'link': 'link2', 'type': ['10x', '1GE', 'LAN', 'EQ', 'SFP-SX', 'QDPCE-R-40X', 'B3', 'I3', 'ether', 'ge'], 'unit': 0}, 'ae0': {'management': 0, 'type': ['ae'], 'pic': 'ae0', 'link': 'ae0' }, }}} config._make_virtual_c_dict(cfg, data) self.assertIsInstance(cfg.c_dict['r0']['interfaces']['ae0'], dict) del(data['r0']['interfaces']['ae0']['link']) try: config._make_virtual_c_dict(cfg, data) except: self.assertRaises(Exception)
def test___make_c_dict(self): config._config__copy_t_dict(cfg) dh = Mock() cfg._get_dev_handle = MagicMock(return_value = dh) cfg.__get_cvar = MagicMock(return_value = 'Junos') cfg._make_ifd_cvar = MagicMock(return_value=True) config._config__make_c_dict(cfg) self.assertIsInstance(cfg.c_dict, dict)
def test___copy_t_dict(self): ''' __copy_t_dict ''' try: config._config__copy_t_dict(cfg) assertTrue('r0' in cfg.c_dict) #assertEqual(cfg.c_dict['system']['osname'] except: pass
def test__make_ifl_cvar(self): cmd_list = ['set others', 'set interfaces ge-0/0/0 unit 0 family inet address address 10.1.1.1/24', 'set interfaces ge-0/0/0 unit 1 family inet address address 10.2.1.1/24', 'set interfaces ge-0/0/0 unit 0 family inet6 address address a::1:1:1:1/64', ] dev_tag = 'r0' ifd_tag = 'r0r1_1' config._config__copy_t_dict(cfg) config_utils.nested_set(cfg.c_dict, [dev_tag, 'interfaces', ifd_tag], {}) config._make_ifl_cvar(cfg, dev_tag=dev_tag, ifd_tag=ifd_tag, cmd_list=cmd_list) self.assertIsInstance(cfg.c_dict[dev_tag]['interfaces'][ifd_tag]['unit']['1'], dict) cmd_list = ['set interfaces ge-0/0/0 unit 0 family inet address address 333.1.1.1/44'] try: config._make_ifl_cvar(cfg, dev_tag=dev_tag, ifd_tag=ifd_tag, cmd_list=cmd_list) except: self.assertRaises(Exception)
def test__make_ifd_cvar(self): config._config__copy_t_dict(cfg) config._make_ifd_cvar(cfg) self.assertEqual(cfg.cv_flat['r1__r0r1_1__slot'], '0/0/2')
t.get_handle = MagicMock() t.get_handle.execute_rpc = MagicMock(return_value = Response(status=True, response='test')) cfg = MagicMock(spec=config) cfg.c_dict = {} cfg.cv_flat = {} cfg.templates = {} cfg.device_handle_map = {} cfg.cfg = defaultdict(list) cfg.cached = False cfg.cached_cfg = {} #cfg.__make_c_dict = MagicMock(return_value={'a':1}) #cfg._process_cv = MagicMock() cfg._config__get_cvar.return_value = 'JunOS' cfg._config__copy_t_dict = MagicMock(side_effect=config._config__copy_t_dict(cfg)) class TestConfig(unittest.TestCase): def test__init(self): new_cfg = config() self.assertIsInstance(new_cfg, config) def test_config_engine_device_list(self): # load file kwargs = {'load_timeout':60} cfg._process_cv.return_value=kwargs dh = MagicMock() dh.current_node.current_controller.os = 'JunOS' dh.current_node.current_controller.host = '1.1.1.1' # dh.execute_rpc.return_value = Response(status=True, response='test') cfg._get_dev_handle.return_value = dh