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, {})