def test_optable_path(self):
fname = 'local-get-interface-information.xml'
path = os.path.join(os.path.dirname(__file__),
'rpc-reply', fname)
lppt = PhyPortStatsTable(path=path)
lppt.get()
self.assertEqual(len(lppt), 2)
def test_optable_path(self):
fname = 'local-get-interface-information.xml'
path = os.path.join(os.path.dirname(__file__),
'rpc-reply', fname)
lppt = PhyPortStatsTable(path=path)
lppt.get()
self.assertEqual(len(lppt), 2)
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.ppt = PhyPortStatsTable(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.ppt = PhyPortStatsTable(self.dev)
def setUp(self):
self.dev = Device(host='1.1.1.1', user='******', password='******',
gather_facts=False)
self.ppt = PhyPortStatsTable(self.dev)
ret_val = '<physical-interface> \
<name>ge-0/0/0</name> \
<admin-status>up</admin-status> \
<oper-status>up</oper-status> \
</physical-interface>'
xml = etree.fromstring(ret_val)
self.v = PhyPortStatsView(self.ppt, xml)
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.ppt = PhyPortStatsTable(self.dev)
class TestFactoryOpTable(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.ppt = PhyPortStatsTable(self.dev)
@patch('jnpr.junos.Device.execute')
def test_optable_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
self.assertEqual(len(self.ppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_get_key(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt.GET_KEY, 'interface_name')
def test_optable_local(self):
fname = 'local-get-interface-information.xml'
self.ppt._path = os.path.join(os.path.dirname(__file__),
'rpc-reply', fname)
self.ppt.get()
self.assertEqual(len(self.ppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
v = self.ppt['ge-0/0/0']
self.assertEqual(v['rx_packets'], 1207)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool(self, mock_execute):
mock_execute.side_effect = self._mock_manager
et = EthPortTable(self.dev)
et.get()
v = et['ge-0/0/0']
self.assertEqual(v['present'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool_regex(self, mock_execute):
mock_execute.side_effect = self._mock_manager
from jnpr.junos.op.bfd import BfdSessionTable
bfd = BfdSessionTable(self.dev)
bfd.get()
v = bfd['10.92.20.4']
self.assertEqual(v['no_absorb'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_unknown_field(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
def bad(key):
v = self.ppt['ge-0/0/0']
return v[key]
self.assertRaises(ValueError, bad, 'bunk')
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:
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')
def test_optable_xml(self):
fname = 'get-interface-information.xml'
xml = self._read_file(fname)
lppt = PhyPortStatsTable(xml=xml)
lppt.get()
self.assertEqual(len(lppt), 2)
class TestFactoryOpTable(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.ppt = PhyPortStatsTable(self.dev)
@patch('jnpr.junos.Device.execute')
def test_optable_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
self.assertEqual(len(self.ppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_get_key(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt.GET_KEY, 'interface_name')
def test_optable_path(self):
fname = 'local-get-interface-information.xml'
path = os.path.join(os.path.dirname(__file__), 'rpc-reply', fname)
lppt = PhyPortStatsTable(path=path)
lppt.get()
self.assertEqual(len(lppt), 2)
def test_optable_xml(self):
fname = 'get-interface-information.xml'
xml = self._read_file(fname)
lppt = PhyPortStatsTable(xml=xml)
lppt.get()
self.assertEqual(len(lppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
v = self.ppt['ge-0/0/0']
self.assertEqual(v['rx_packets'], 1207)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool(self, mock_execute):
mock_execute.side_effect = self._mock_manager
et = EthPortTable(self.dev)
et.get()
v = et['ge-0/0/0']
self.assertEqual(v['present'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool_regex(self, mock_execute):
mock_execute.side_effect = self._mock_manager
from jnpr.junos.op.bfd import BfdSessionTable
bfd = BfdSessionTable(self.dev)
bfd.get()
v = bfd['10.92.20.4']
self.assertEqual(v['no_absorb'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_unknown_field(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
def bad(key):
v = self.ppt['ge-0/0/0']
return v[key]
self.assertRaises(ValueError, bad, 'bunk')
def test_generate_sax_parser_fields_with_many_slash(self):
yaml_data = """
---
bgpNeighborTable:
rpc: get-bgp-neighbor-information
item: bgp-peer
key: peer-address
view: bgpNeighborView
bgpNeighborView:
fields:
prefix-count: bgp-option-information/prefix-limit/prefix-count
prefix-dummy: bgp-option-information/prefix-limit/prefix-dummy
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = bgpNeighborTable(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, 'bgp-peer')
self.assertEqual(
len(etree.tostring(data)),
len(b'<bgp-peer><peer-address/><bgp-option-information><prefix-limit>'
b'<prefix-count/><prefix-dummy/></prefix-limit>'
b'</bgp-option-information></bgp-peer>'))
def test_generate_sax_parser_item_with_many_slash(self):
yaml_data = """
---
taskmallocdetail:
rpc: get-task-memory-information
args:
level: detail
item: task-memory-malloc-usage-report/task-malloc-list/task-malloc
key: tm-name
view: taskmallocview
taskmallocview:
fields:
tmallocs: tm-allocs
tmallocbytes: tm-alloc-bytes
tmmaxallocs: tm-max-allocs
tmmaxallocbytes: tm-max-alloc-bytes
tmfunctioncalls: tm-function-calls
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = taskmallocdetail(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, 'task-memory-malloc-usage-report')
self.assertEqual(
len(etree.tostring(data)),
len(b'<task-memory-malloc-usage-report><task-malloc-list><task-malloc><tm-name/><t'
b'm-allocs/><tm-alloc-bytes/><tm-max-allocs/><tm-max-alloc-bytes/><tm-function'
b'-calls/></task-malloc></task-malloc-list></task-memory-malloc-usage-report>'
))
def test_generate_sax_parser_same_parents_with_diff_fields(self):
yaml_data = """
---
VtepTable:
rpc: get-interface-information
args:
interface-name: "vtep"
extensive: True
item: physical-interface
key: name
view: VtepView
VtepView:
fields:
admin-status: admin-status
oper-status: oper-status
link-level-type: link-level-type
input-bytes: traffic-statistics/input-bytes
output-bytes: traffic-statistics/output-bytes
input-errors: input-error-list/input-errors
output-errors: output-error-list/output-errors
carrier-transitions: output-error-list/carrier-transitions
vtep-tunnel-stats: VtepTunnelTable
VtepTunnelTable:
item: logical-interface
key: name
view: VtepTunnelView
VtepTunnelView:
fields:
vtep-type: vtep-info/vtep-type
vtep-address: vtep-info/vtep-address
tunnel-input-bytes: traffic-statistics/input-bytes
tunnel-output-bytes: traffic-statistics/output-bytes
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = VtepTable(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, 'physical-interface')
self.assertEqual(
len(etree.tostring(data)),
len(b'<physical-interface><name/><admin-status/><oper-status/>'
b'<link-level-type/><traffic-statistics><input-bytes/>'
b'<output-bytes/></traffic-statistics><input-error-list>'
b'<input-errors/></input-error-list><output-error-list>'
b'<output-errors/><carrier-transitions/></output-error-list>'
b'<logical-interface><name/><vtep-info><vtep-type/><vtep-address/>'
b'</vtep-info><traffic-statistics><input-bytes/><output-bytes/>'
b'</traffic-statistics></logical-interface></physical-interface>'
))
@patch('jnpr.junos.Device.execute')
def test_key_pipe_delim_with_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status
item: //utmd-status
view: UTMStatusView
key: ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(json.loads(data.to_json()), {
'node0': {
'running': True
},
'node1': {
'running': True
}
})
@patch('jnpr.junos.Device.execute')
def test_key_pipe_delim_with_Null_use_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status_use_Null
item: //utmd-status
view: UTMStatusView
key: ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(json.loads(data.to_json()), {'running': True})
@patch('jnpr.junos.Device.execute')
def test_key_and_item_pipe_delim_with_Null_use_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status_use_Null
item: //multi-routing-engine-item/utmd-status | //utmd-status
view: UTMStatusView
key:
- ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(json.loads(data.to_json()), {'running': True})
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()
reply = RPCReply(foo)
reply.parse()
rpc_reply = NCElement(reply, self.dev._conn.
_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 TestFactoryOpTable(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.ppt = PhyPortStatsTable(self.dev)
@patch('jnpr.junos.Device.execute')
def test_optable_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
self.assertEqual(len(self.ppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_get_key(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get('ge-0/0/0')
self.assertEqual(self.ppt.GET_KEY, 'interface_name')
def test_optable_path(self):
fname = 'local-get-interface-information.xml'
path = os.path.join(os.path.dirname(__file__), 'rpc-reply', fname)
lppt = PhyPortStatsTable(path=path)
lppt.get()
self.assertEqual(len(lppt), 2)
def test_optable_xml(self):
fname = 'get-interface-information.xml'
xml = self._read_file(fname)
lppt = PhyPortStatsTable(xml=xml)
lppt.get()
self.assertEqual(len(lppt), 2)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
v = self.ppt['ge-0/0/0']
self.assertEqual(v['rx_packets'], 1207)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool(self, mock_execute):
mock_execute.side_effect = self._mock_manager
et = EthPortTable(self.dev)
et.get()
v = et['ge-0/0/0']
self.assertEqual(v['present'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_astype_bool_regex(self, mock_execute):
mock_execute.side_effect = self._mock_manager
from jnpr.junos.op.bfd import BfdSessionTable
bfd = BfdSessionTable(self.dev)
bfd.get()
v = bfd['10.92.20.4']
self.assertEqual(v['no_absorb'], True)
@patch('jnpr.junos.Device.execute')
def test_optable_view_get_unknown_field(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
def bad(key):
v = self.ppt['ge-0/0/0']
return v[key]
self.assertRaises(ValueError, bad, 'bunk')
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')
from jnpr.junos.op.arp import ArpTable arp_entries = ArpTable(a_device) arp_entries.get() arp_entries.items() arp_entries.keys() arp_entries.values() from jnpr.junos.op.phyport import PhyPortTable phy_ports = PhyPortTable(a_device) phy_ports.get() phy_ports.items() phy_ports.keys() phy_ports.values() phy_ports['fe-0/0/1'] phy_ports['fe-0/0/1'].keys() phy_ports['fe-0/0/1'].items() from jnpr.junos.op.phyport import PhyPortStatsTable port_stats = PhyPortStatsTable(a_device) port_stats.get() port_stats.keys() port_stats['fe-0/0/1'].items() if __name__ == "__main__" def __main__:
def test_optable_xml(self):
fname = 'get-interface-information.xml'
xml = self._read_file(fname)
lppt = PhyPortStatsTable(xml=xml)
lppt.get()
self.assertEqual(len(lppt), 2)
from jnpr.junos.op.ethport import EthPortTable
from jnpr.junos.op.arp import ArpTable
from jnpr.junos.op.phyport import PhyPortTable
from jnpr.junos.op.phyport import PhyPortStatsTable
from getpass import getpass
from pprint import pprint
a_device = Device(host='184.105.247.76', user='******', password='******')
a_device.open()
ports = EthPortTable(a_device)
jports=ports.get()
pprint(jports)
#pprint(jports['fe-0/0/1'].items())
phy_ports = PhyPortTable(a_device)
pp = phy_ports.get()
print "PP type %s" %type(pp)
p = pp.items()
for key,values in p:
print key
for k,v in values:
print " %s %s" %(k,v)
port_stats = PhyPortStatsTable(a_device)
ps = port_stats.get()
p = ps.items()
for key,values in p:
print key
for k,v in values:
print " %s %s" %(k,v)
class TestFactoryOpTable(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.ppt = PhyPortStatsTable(self.dev)
@patch("jnpr.junos.Device.execute")
def test_optable_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
self.assertEqual(len(self.ppt), 2)
@patch("jnpr.junos.Device.execute")
def test_optable_get_key(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get("ge-0/0/0")
self.assertEqual(self.ppt.GET_KEY, "interface_name")
def test_optable_path(self):
fname = "local-get-interface-information.xml"
path = os.path.join(os.path.dirname(__file__), "rpc-reply", fname)
lppt = PhyPortStatsTable(path=path)
lppt.get()
self.assertEqual(len(lppt), 2)
def test_optable_xml(self):
fname = "get-interface-information.xml"
xml = self._read_file(fname)
lppt = PhyPortStatsTable(xml=xml)
lppt.get()
self.assertEqual(len(lppt), 2)
@patch("jnpr.junos.Device.execute")
def test_optable_view_get(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
v = self.ppt["ge-0/0/0"]
self.assertEqual(v["rx_packets"], 1207)
@patch("jnpr.junos.Device.execute")
def test_optable_view_get_astype_bool(self, mock_execute):
mock_execute.side_effect = self._mock_manager
et = EthPortTable(self.dev)
et.get()
v = et["ge-0/0/0"]
self.assertEqual(v["present"], True)
@patch("jnpr.junos.Device.execute")
def test_optable_view_get_astype_bool_regex(self, mock_execute):
mock_execute.side_effect = self._mock_manager
from jnpr.junos.op.bfd import BfdSessionTable
bfd = BfdSessionTable(self.dev)
bfd.get()
v = bfd["10.92.20.4"]
self.assertEqual(v["no_absorb"], True)
@patch("jnpr.junos.Device.execute")
def test_optable_view_get_unknown_field(self, mock_execute):
mock_execute.side_effect = self._mock_manager
self.ppt.get()
def bad(key):
v = self.ppt["ge-0/0/0"]
return v[key]
self.assertRaises(ValueError, bad, "bunk")
def test_generate_sax_parser_fields_with_many_slash(self):
yaml_data = """
---
bgpNeighborTable:
rpc: get-bgp-neighbor-information
item: bgp-peer
key: peer-address
view: bgpNeighborView
bgpNeighborView:
fields:
prefix-count: bgp-option-information/prefix-limit/prefix-count
prefix-dummy: bgp-option-information/prefix-limit/prefix-dummy
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = bgpNeighborTable(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, "bgp-peer")
self.assertEqual(
len(etree.tostring(data)),
len(b"<bgp-peer><peer-address/><bgp-option-information><prefix-limit>"
b"<prefix-count/><prefix-dummy/></prefix-limit>"
b"</bgp-option-information></bgp-peer>"),
)
def test_generate_sax_parser_fields_with_diff_child_xpaths(self):
yaml_data = """
---
twampProbeTable:
rpc: twamp-get-probe-results
item: probe-test-results
key: test-name
view: probeResultsView
probeResultsView:
fields:
min-delay: probe-test-global-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/min-delay
max-delay: probe-test-global-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/max-delay
avg-delay: probe-test-global-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/avg-delay
positive-rtt-jitter: probe-test-global-results/probe-test-generic-results/probe-test-positive-round-trip-jitter/probe-summary-results/avg-delay
loss-percentage: probe-test-global-results/probe-test-generic-results/loss-percentage
current-min-delay: probe-last-test-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/min-delay
current-max-delay: probe-last-test-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/max-delay
current-avg-delay: probe-last-test-results/probe-test-generic-results/probe-test-rtt/probe-summary-results/avg-delay
current-positive-rtt-jitter: probe-last-test-results/probe-test-generic-results/probe-test-positive-round-trip-jitter/probe-summary-results/avg-delay
current-loss-percentage: probe-last-test-results/probe-test-generic-results/loss-percentage
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = twampProbeTable(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, "probe-test-results")
self.assertEqual(
len(etree.tostring(data)),
len(b"<probe-test-results><test-name/><probe-last-test-results>"
b"<probe-test-generic-results><probe-test-rtt><probe-summary"
b"-results><min-delay/><avg-delay/><max-delay/></probe-summary"
b"-results></probe-test-rtt><loss-percentage/>"
b"<probe-test-positive-round-trip-jitter><probe-summary-results>"
b"<avg-delay/></probe-summary-results></probe-test-positive-round"
b"-trip-jitter></probe-test-generic-results></probe-last-test-resu"
b"lts><probe-test-global-results><probe-test-generic-results>"
b"<probe-test-rtt><probe-summary-results><min-delay/><avg-delay/>"
b"<max-delay/></probe-summary-results></probe-test-rtt>"
b"<loss-percentage/><probe-test-positive-round-trip-jitter>"
b"<probe-summary-results><avg-delay/></probe-summary-results>"
b"</probe-test-positive-round-trip-jitter></probe-test-generic-"
b"results></probe-test-global-results></probe-test-results>"),
)
def test_generate_sax_parser_item_with_many_slash(self):
yaml_data = """
---
taskmallocdetail:
rpc: get-task-memory-information
args:
level: detail
item: task-memory-malloc-usage-report/task-malloc-list/task-malloc
key: tm-name
view: taskmallocview
taskmallocview:
fields:
tmallocs: tm-allocs
tmallocbytes: tm-alloc-bytes
tmmaxallocs: tm-max-allocs
tmmaxallocbytes: tm-max-alloc-bytes
tmfunctioncalls: tm-function-calls
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = taskmallocdetail(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, "task-memory-malloc-usage-report")
self.assertEqual(
len(etree.tostring(data)),
len(b"<task-memory-malloc-usage-report><task-malloc-list><task-malloc><tm-name/><t"
b"m-allocs/><tm-alloc-bytes/><tm-max-allocs/><tm-max-alloc-bytes/><tm-function"
b"-calls/></task-malloc></task-malloc-list></task-memory-malloc-usage-report>"
),
)
def test_generate_sax_parser_same_parents_with_diff_fields(self):
yaml_data = """
---
VtepTable:
rpc: get-interface-information
args:
interface-name: "vtep"
extensive: True
item: physical-interface
key: name
view: VtepView
VtepView:
fields:
admin-status: admin-status
oper-status: oper-status
link-level-type: link-level-type
input-bytes: traffic-statistics/input-bytes
output-bytes: traffic-statistics/output-bytes
input-errors: input-error-list/input-errors
output-errors: output-error-list/output-errors
carrier-transitions: output-error-list/carrier-transitions
vtep-tunnel-stats: VtepTunnelTable
VtepTunnelTable:
item: logical-interface
key: name
view: VtepTunnelView
VtepTunnelView:
fields:
vtep-type: vtep-info/vtep-type
vtep-address: vtep-info/vtep-address
tunnel-input-bytes: traffic-statistics/input-bytes
tunnel-output-bytes: traffic-statistics/output-bytes
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = VtepTable(self.dev)
data = generate_sax_parser_input(tbl)
self.assertEqual(data.tag, "physical-interface")
self.assertEqual(
len(etree.tostring(data)),
len(b"<physical-interface><name/><admin-status/><oper-status/>"
b"<link-level-type/><traffic-statistics><input-bytes/>"
b"<output-bytes/></traffic-statistics><input-error-list>"
b"<input-errors/></input-error-list><output-error-list>"
b"<output-errors/><carrier-transitions/></output-error-list>"
b"<logical-interface><name/><vtep-info><vtep-type/><vtep-address/>"
b"</vtep-info><traffic-statistics><input-bytes/><output-bytes/>"
b"</traffic-statistics></logical-interface></physical-interface>"
),
)
@patch("jnpr.junos.Device.execute")
def test_key_pipe_delim_with_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status
item: //utmd-status
view: UTMStatusView
key: ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(
json.loads(data.to_json()),
{
"node0": {
"running": True
},
"node1": {
"running": True
}
},
)
@patch("jnpr.junos.Device.execute")
def test_key_pipe_delim_with_Null_use_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status_use_Null
item: //utmd-status
view: UTMStatusView
key: ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(json.loads(data.to_json()), {"running": True})
@patch("jnpr.junos.Device.execute")
def test_key_and_item_pipe_delim_with_Null_use_Null(self, mock_execute):
mock_execute.side_effect = self._mock_manager
yaml_data = """
---
UTMStatusTable:
rpc: show-utmd-status_use_Null
item: //multi-routing-engine-item/utmd-status | //utmd-status
view: UTMStatusView
key:
- ../re-name | Null
UTMStatusView:
fields:
running: { running: flag }
"""
globals().update(FactoryLoader().load(
yaml.load(yaml_data, Loader=yaml.Loader)))
tbl = UTMStatusTable(self.dev)
data = tbl.get()
self.assertEqual(json.loads(data.to_json()), {"running": True})
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()
reply = RPCReply(foo)
reply.parse()
rpc_reply = NCElement(reply,
self.dev._conn._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")
