def __init__(self, dev):
Table.__init__(self, dev) # call parent constructor
self._data_dict = self.DEFINE # crutch
self.ITEM_NAME_XPATH = self._data_dict.get('key', 'name')
self.ITEM_XPATH = self._data_dict['get']
self.view = self._data_dict.get('view')
def __init__(self, dev):
Table.__init__(self, dev) # call parent constructor
self._data_dict = self.DEFINE # crutch
self.ITEM_NAME_XPATH = self._data_dict.get('key', 'name')
self.ITEM_XPATH = self._data_dict['get']
self.view = self._data_dict.get('view')
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host='1.1.1.1', user='******', password='******',
gather_facts=False)
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def __init__(self, dev=None, xml=None, path=None):
Table.__init__(self, dev, xml, path) # call parent constructor
self._data_dict = self.DEFINE # crutch
self.ITEM_NAME_XPATH = self._data_dict.get('key', 'name')
self.ITEM_XPATH = self._data_dict['get']
self.view = self._data_dict.get('view')
self._options = self._data_dict.get('options')
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host='1.1.1.1',
user='******',
password='******',
gather_facts=False)
self.device_handler = JunosDeviceHandler(device_params={
'name': 'junos',
'local': False
})
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host="1.1.1.1",
user="******",
password="******",
gather_facts=False)
self.device_handler = JunosDeviceHandler(device_params={
"name": "junos",
"local": False
})
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host='1.1.1.1', user='******', password='******',
gather_facts=False)
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def __init__(self, dev=None, xml=None, path=None, mode=None):
Table.__init__(self, dev, xml, path) # call parent constructor
self._init_get()
self._data_dict = self.DEFINE # crutch
self.ITEM_NAME_XPATH = self._data_dict.get('key', 'name')
self.view = self._data_dict.get('view')
self._options = self._data_dict.get('options')
self.mode = mode
if 'set' in self._data_dict:
Config.__init__(self, dev, mode) # call parent constructor
self._init_set()
if self._view:
self.fields = self._view.FIELDS.copy()
else:
raise ValueError(
"%s set table view is not defined.\n"
% (self.__class__.__name__)
)
if 'key-field' in self._data_dict:
key_name = self._data_dict.get('key-field', None)
if isinstance(key_name, list):
self.key_field = key_name
elif isinstance(key_name, str):
self.key_field = [key_name]
else:
raise TypeError(
"Key-field %s is of invalid type %s.\n"
% (key_name, type(key_name))
)
else:
raise ValueError(
"Table should have key-field attribute defined\n"
)
self._type = 'set'
self._init_field()
else:
self._type = 'get'
self.ITEM_XPATH = self._data_dict[self._type]
# no new attributes.
self._freeze()
def __init__(self, dev=None, xml=None, path=None, mode=None):
Table.__init__(self, dev, xml, path) # call parent constructor
self._init_get()
self._data_dict = self.DEFINE # crutch
self.ITEM_NAME_XPATH = self._data_dict.get('key', 'name')
self.view = self._data_dict.get('view')
self._options = self._data_dict.get('options')
self.mode = mode
if 'set' in self._data_dict:
Config.__init__(self, dev, mode) # call parent constructor
self._init_set()
if self._view:
self.fields = self._view.FIELDS.copy()
else:
raise ValueError(
"%s set table view is not defined.\n"
% (self.__class__.__name__)
)
if 'key-field' in self._data_dict:
key_name = self._data_dict.get('key-field', None)
if isinstance(key_name, list):
self.key_field = key_name
elif isinstance(key_name, str):
self.key_field = [key_name]
else:
raise TypeError(
"Key-field %s is of invalid type %s.\n"
% (key_name, type(key_name))
)
else:
raise ValueError(
"Table should have key-field attribute defined\n"
)
self._type = 'set'
self._init_field()
else:
self._type = 'get'
self.ITEM_XPATH = self._data_dict[self._type]
# no new attributes.
self._freeze()
class TestFactoryTable(unittest.TestCase):
@patch('ncclient.manager.connect')
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host='1.1.1.1', user='******', password='******',
gather_facts=False)
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def test_config_constructor(self):
self.assertTrue(isinstance(self.table.D, Device))
def test_table_hostname(self):
self.assertEqual(self.table.hostname, '1.1.1.1')
def test_table_is_container(self):
self.assertTrue(self.table.is_container)
def test_table_repr_xml_none(self):
self.assertEqual(repr(self.table), 'Table:1.1.1.1 - Table empty')
def test_table_view_setter_ValueError(self):
try:
self.table.view = 'test'
except Exception as ex:
self.assertEqual(ex.__class__, ValueError)
@patch('jnpr.junos.Device.execute')
def test_keys_RuntimeError(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.ppt.ITEM_NAME_XPATH = 1
self.assertRaises(RuntimeError, self.ppt.keys)
@patch('jnpr.junos.Device.execute')
def test_keys__keys_composite(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.ppt.ITEM_NAME_XPATH = ['name', 'missing', 'mtu']
self.assertEqual(self.ppt.keys(),
[('ge-0/0/0', None, '1514'), ('ge-0/0/1', None, '1514')])
@patch('jnpr.junos.Device.execute')
def test_keys__keys_pipe(self, mock_execute):
from jnpr.junos.op.lldp import LLDPNeighborTable
mock_execute.side_effect = self._mock_manager
self.lldp = LLDPNeighborTable(self.dev)
self.lldp.get()
self.assertEqual(self.lldp.keys(), ['et-0/0/48', 'et-0/0/49', 'xe-0/0/13'])
@patch('jnpr.junos.Device.execute')
def test_table_repr_xml_not_none(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.table.xml = self.ppt.xml
self.table.ITEM_XPATH = self.ppt.ITEM_XPATH
self.assertEqual(repr(self.table), 'Table:1.1.1.1: 2 items')
@patch('jnpr.junos.Device.execute')
def test_table_get_keys_values(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt.keys(), ['ge-0/0/0', 'ge-0/0/1'])
self.assertEqual(len(self.ppt.values()), 2)
self.ppt.view = None
self.assertEqual(len(self.ppt.values()), 2)
@patch('jnpr.junos.Device.execute')
def test_table__getitem__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[0].ITEM_NAME_XPATH, 'name')
@patch('jnpr.junos.Device.execute')
def test_table__getitem__slice(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[:1][0].__class__.__name__, 'PhyPortView')
@patch('jnpr.junos.Device.execute')
def test_table__getitem__tuple(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[('ge-0/0/0',)], None)
@patch('jnpr.junos.Device.execute')
def test_table__contains__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertTrue('ge-0/0/0' in self.ppt)
@patch('jnpr.junos.Device.execute')
def test_table_items(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(len(self.ppt.items()[1][1]), 8)
def test_table_get_return_none(self):
self.assertEqual(self.table.get('ge-0/0/0'), None)
def test_table_get_RuntimeError(self):
self.assertRaises(RuntimeError, self.table._keys)
@patch('jnpr.junos.Device.execute')
@patch('__builtin__.file')
def test_table_savexml(self, mock_file, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.xml = etree.XML('<root><a>test</a></root>')
self.ppt.savexml('/vasr/tmssp/foo.xml', hostname=True, append='test')
mock_file.assert_called_once_with('/vasr/tmssp/foo_1.1.1.1_test.xml',
'w')
self.ppt.savexml('/vasr/tmssp/foo.xml', hostname=True, timestamp=True)
self.assertEqual(mock_file.call_count, 2)
def _read_file(self, fname):
from ncclient.xml_ import NCElement
fpath = os.path.join(os.path.dirname(__file__),
'rpc-reply', fname)
foo = open(fpath).read()
rpc_reply = NCElement(foo, self.dev._conn.
_device_handler.transform_reply())\
._NCElement__doc[0]
return rpc_reply
def _mock_manager(self, *args, **kwargs):
if kwargs:
if 'normalize' in kwargs and args:
return self._read_file(args[0].tag + '.xml')
device_params = kwargs['device_params']
device_handler = make_device_handler(device_params)
session = SSHSession(device_handler)
return Manager(session, device_handler)
if args:
return self._read_file(args[0].tag + '.xml')
class TestFactoryTable(unittest.TestCase):
@patch('ncclient.manager.connect')
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host='1.1.1.1',
user='******',
password='******',
gather_facts=False)
self.device_handler = JunosDeviceHandler(device_params={
'name': 'junos',
'local': False
})
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def test_config_constructor(self):
self.assertTrue(isinstance(self.table.D, Device))
def test_table_hostname(self):
self.assertEqual(self.table.hostname, '1.1.1.1')
def test_table_is_container(self):
self.assertTrue(self.table.is_container)
def test_table_repr_xml_none(self):
self.assertEqual(repr(self.table), 'Table:1.1.1.1 - Table empty')
def test_table_view_setter_ValueError(self):
try:
self.table.view = 'test'
except Exception as ex:
self.assertEqual(ex.__class__, ValueError)
@patch('jnpr.junos.Device.execute')
def test_keys_RuntimeError(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.ppt.ITEM_NAME_XPATH = 1
self.assertRaises(RuntimeError, self.ppt.keys)
@patch('jnpr.junos.Device.execute')
def test_keys__keys_composite(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.ppt.ITEM_NAME_XPATH = ['name', 'missing', 'mtu']
self.assertEqual(self.ppt.keys(), [('ge-0/0/0', None, '1514'),
('ge-0/0/1', None, '1514')])
@patch('jnpr.junos.Device.execute')
def test_keys__keys_pipe(self, mock_execute):
from jnpr.junos.op.lldp import LLDPNeighborTable
mock_execute.side_effect = self._mock_manager
self.lldp = LLDPNeighborTable(self.dev)
self.lldp.get()
self.assertEqual(self.lldp.keys(),
['et-0/0/48', 'et-0/0/49', 'xe-0/0/13'])
@patch('jnpr.junos.Device.execute')
def test_table_repr_xml_not_none(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.table.xml = self.ppt.xml
self.table.ITEM_XPATH = self.ppt.ITEM_XPATH
self.assertEqual(repr(self.table), 'Table:1.1.1.1: 2 items')
@patch('jnpr.junos.Device.execute')
def test_table_get_keys_values(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt.keys(), ['ge-0/0/0', 'ge-0/0/1'])
self.assertEqual(len(self.ppt.values()), 2)
self.ppt.view = None
self.assertEqual(len(self.ppt.values()), 2)
@patch('jnpr.junos.Device.execute')
def test_table__getitem__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[0].ITEM_NAME_XPATH, 'name')
@patch('jnpr.junos.Device.execute')
def test_table__getitem__slice(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[:1][0].__class__.__name__, 'PhyPortView')
@patch('jnpr.junos.Device.execute')
def test_table__getitem__tuple(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt[('ge-0/0/0', )], None)
@patch('jnpr.junos.Device.execute')
def test_table__contains__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertTrue('ge-0/0/0' in self.ppt)
@patch('jnpr.junos.Device.execute')
def test_table_items(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
print(self.ppt.items())
self.assertEqual(len(self.ppt.items()[1][1]), 8)
def test_table_get_return_none(self):
self.assertEqual(self.table.get('ge-0/0/0'), None)
def test_table_get_RuntimeError(self):
self.assertRaises(RuntimeError, self.table._keys)
@patch('jnpr.junos.Device.execute')
@patch(builtin_string + '.open')
def test_table_savexml(self, mock_file, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.xml = etree.XML('<root><a>test</a></root>')
self.ppt.savexml('/vasr/tmssp/foo.xml', hostname=True, append='test')
mock_file.assert_called_once_with('/vasr/tmssp/foo_1.1.1.1_test.xml',
'wb')
self.ppt.savexml('/vasr/tmssp/foo.xml', hostname=True, timestamp=True)
self.assertEqual(mock_file.call_count, 2)
def _read_file(self, fname):
from ncclient.xml_ import NCElement
fpath = os.path.join(os.path.dirname(__file__), 'rpc-reply', fname)
foo = open(fpath).read()
rpc_reply = NCElement(foo, self.device_handler.transform_reply())\
._NCElement__doc[0]
return rpc_reply
def _mock_manager(self, *args, **kwargs):
if kwargs:
if args and ('normalize' in kwargs or 'filter_xml' in kwargs):
return self._read_file(args[0].tag + '.xml')
device_params = kwargs['device_params']
device_handler = make_device_handler(device_params)
session = SSHSession(device_handler)
return Manager(session, device_handler)
if args:
return self._read_file(args[0].tag + '.xml')
class TestFactoryTable(unittest.TestCase):
@patch("ncclient.manager.connect")
def setUp(self, mock_connect):
mock_connect.side_effect = self._mock_manager
self.dev = Device(host="1.1.1.1",
user="******",
password="******",
gather_facts=False)
self.device_handler = JunosDeviceHandler(device_params={
"name": "junos",
"local": False
})
self.dev.open()
self.table = Table(dev=self.dev)
self.ppt = PhyPortTable(self.dev)
def test_config_constructor(self):
self.assertTrue(isinstance(self.table.D, Device))
def test_table_hostname(self):
self.assertEqual(self.table.hostname, "1.1.1.1")
def test_table_is_container(self):
self.assertTrue(self.table.is_container)
def test_table_repr_xml_none(self):
self.assertEqual(repr(self.table), "Table:1.1.1.1 - Table empty")
def test_table_view_setter_ValueError(self):
try:
self.table.view = "test"
except Exception as ex:
self.assertEqual(ex.__class__, ValueError)
@patch("jnpr.junos.Device.execute")
def test_keys_RuntimeError(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.ppt.ITEM_NAME_XPATH = 1
self.assertRaises(RuntimeError, self.ppt.keys)
@patch("jnpr.junos.Device.execute")
def test_keys__keys_composite(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.ppt.ITEM_NAME_XPATH = ["name", "missing", "mtu"]
self.assertEqual(self.ppt.keys(), [("ge-0/0/0", None, "1514"),
("ge-0/0/1", None, "1514")])
@patch("jnpr.junos.Device.execute")
def test_keys__keys_pipe(self, mock_execute):
from jnpr.junos.op.lldp import LLDPNeighborTable
mock_execute.side_effect = self._mock_manager
self.lldp = LLDPNeighborTable(self.dev)
self.lldp.get()
self.assertEqual(self.lldp.keys(),
["et-0/0/48", "et-0/0/49", "xe-0/0/13"])
@patch("jnpr.junos.Device.execute")
def test_table_repr_xml_not_none(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.table.xml = self.ppt.xml
self.table.ITEM_XPATH = self.ppt.ITEM_XPATH
self.assertEqual(repr(self.table), "Table:1.1.1.1: 2 items")
@patch("jnpr.junos.Device.execute")
def test_table_get_keys_values(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(self.ppt.keys(), ["ge-0/0/0", "ge-0/0/1"])
self.assertEqual(len(self.ppt.values()), 2)
self.ppt.view = None
self.assertEqual(len(self.ppt.values()), 2)
@patch("jnpr.junos.Device.execute")
def test_table__getitem__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(self.ppt[0].ITEM_NAME_XPATH, "name")
@patch("jnpr.junos.Device.execute")
def test_table__getitem__slice(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(self.ppt[:1][0].__class__.__name__, "PhyPortView")
@patch("jnpr.junos.Device.execute")
def test_table__getitem__tuple(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(self.ppt[("ge-0/0/0", )], None)
@patch("jnpr.junos.Device.execute")
def test_table__contains__(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertTrue("ge-0/0/0" in self.ppt)
@patch("jnpr.junos.Device.execute")
def test_table_items(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(len(self.ppt.items()[1][1]), 8)
def test_table_get_return_none(self):
self.assertEqual(self.table.get("ge-0/0/0"), None)
def test_table_get_RuntimeError(self):
self.assertRaises(RuntimeError, self.table._keys)
@patch("jnpr.junos.Device.execute")
@patch(builtin_string + ".open")
def test_table_savexml(self, mock_file, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.xml = etree.XML("<root><a>test</a></root>")
self.ppt.savexml("/vasr/tmssp/foo.xml", hostname=True, append="test")
mock_file.assert_called_once_with("/vasr/tmssp/foo_1.1.1.1_test.xml",
"wb")
self.ppt.savexml("/vasr/tmssp/foo.xml", hostname=True, timestamp=True)
self.assertEqual(mock_file.call_count, 2)
def _read_file(self, fname):
from ncclient.xml_ import NCElement
fpath = os.path.join(os.path.dirname(__file__), "rpc-reply", fname)
foo = open(fpath).read()
rpc_reply = NCElement(
foo, self.device_handler.transform_reply())._NCElement__doc[0]
return rpc_reply
def _mock_manager(self, *args, **kwargs):
if kwargs:
if args and ("normalize" in kwargs or "filter_xml" in kwargs):
return self._read_file(args[0].tag + ".xml")
device_params = kwargs["device_params"]
device_handler = make_device_handler(device_params)
session = SSHSession(device_handler)
return Manager(session, device_handler)
if args:
return self._read_file(args[0].tag + ".xml")
