def test_get_network_info_virtual_devices(self, net_io_counters):
Stats = collections.namedtuple('interface', ['bytes_sent', 'bytes_recv',
'packets_sent', 'packets_recv'
])
net_stats = Stats(bytes_sent = 0, bytes_recv = 1,
packets_sent = 2, packets_recv = 3
)
all_net_stats = { 'etc11' : net_stats, 'etc2' : net_stats }
net_io_counters.return_value = all_net_stats
c = MagicMock()
#skip virtual devices
c.get_virtual_interfaces_skip.return_value = 'True'
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(network_info, {})
#do not skip virtual devices
c.get_virtual_interfaces_skip.return_value = 'False'
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
#len({'bytes_in': ..., 'pkts_in': ..., 'pkts_out': ..., 'bytes_out': ...}) == 4
self.assertEqual(len(network_info), 4)
def test_get_network_info_skip_by_pattern(self, net_io_counters):
Stats = collections.namedtuple('interface', ['bytes_sent', 'bytes_recv',
'packets_sent', 'packets_recv'
])
net_stats = Stats(bytes_sent = 0, bytes_recv = 1,
packets_sent = 2, packets_recv = 3
)
all_net_stats = { 'etc11' : net_stats, 'etc2' : net_stats }
net_io_counters.return_value = all_net_stats
c = MagicMock()
#skip all by pattern
c.get_virtual_interfaces_skip.return_value = 'False'
c.get_network_interfaces_skip_pattern.return_value = "^etc\d*$"
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(network_info, {})
#skip one by pattern
c.get_network_interfaces_skip_pattern.return_value = "^etc\d{1}$"
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(len(network_info), 4)
all_net_stats = { 'etc2' : net_stats }
net_io_counters.return_value = all_net_stats
c.get_network_interfaces_skip_pattern.return_value = "^etc\d{1}$"
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(network_info, {})
#skip by 'None' pattern
c.get_network_interfaces_skip_pattern.return_value = "None"
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(len(network_info), 4)
def test_get_network_info_skip_by_pattern_and_virtual(self, net_io_counters):
Stats = collections.namedtuple('interface', ['bytes_sent', 'bytes_recv',
'packets_sent', 'packets_recv'
])
net_stats = Stats(bytes_sent = 0, bytes_recv = 1,
packets_sent = 2, packets_recv = 3
)
all_net_stats = { 'etc11' : net_stats, 'etc2' : net_stats, 'etc333' : net_stats }
net_io_counters.return_value = all_net_stats
c = MagicMock()
#skip only one by pattern and other as virtual
c.get_virtual_interfaces_skip.return_value = 'True'
c.get_network_interfaces_skip_pattern.return_value = "^etc\d{3}$"
hostinfo = HostInfo(c)
network_info = hostinfo.get_network_info()
self.assertEqual(network_info, {})
