def testEnumerateInterfaces(self):
# Stub out wmi.WMI().Win32_NetworkAdapterConfiguration()
wmi_object = self.windows.wmi.WMI.return_value
wmi_object.Win32_NetworkAdapterConfiguration.return_value = [
client_test_lib.WMIWin32NetworkAdapterConfigurationMock()
]
enumif = self.windows.EnumerateInterfaces()
replies = []
def Collect(reply):
replies.append(reply)
enumif.SendReply = Collect
enumif.Run(None)
self.assertEqual(len(replies), 1)
interface = replies[0]
self.assertEqual(len(interface.addresses), 4)
addresses = [x.human_readable_address for x in interface.addresses]
self.assertItemsEqual(addresses, [
"192.168.1.20", "ffff::ffff:aaaa:1111:aaaa",
"dddd:0:8888:6666:bbbb:aaaa:eeee:bbbb",
"dddd:0:8888:6666:bbbb:aaaa:ffff:bbbb"
])
def testEnumerateInterfaces(self):
# Stub out wmi.WMI().Win32_NetworkAdapterConfiguration(IPEnabled=1)
wmi_object = self.windows.wmi.WMI.return_value
wmi_object.Win32_NetworkAdapterConfiguration.return_value = [
client_test_lib.WMIWin32NetworkAdapterConfigurationMock()
]
enumif = self.windows.EnumerateInterfaces()
interface_dict_list = list(enumif.RunNetAdapterWMIQuery())
self.assertEqual(len(interface_dict_list), 1)
interface = interface_dict_list[0]
self.assertEqual(len(interface["addresses"]), 4)
addresses = [x.human_readable_address for x in interface["addresses"]]
self.assertItemsEqual(addresses, [
"192.168.1.20", "ffff::ffff:aaaa:1111:aaaa",
"dddd:0:8888:6666:bbbb:aaaa:eeee:bbbb",
"dddd:0:8888:6666:bbbb:aaaa:ffff:bbbb"
])
def testEnumerateInterfacesMock(self):
# Stub out wmi.WMI().Win32_NetworkAdapterConfiguration()
wmi = mock.MagicMock()
wmi.Win32_NetworkAdapterConfiguration.return_value = [
client_test_lib.WMIWin32NetworkAdapterConfigurationMock()
]
replies = []
with mock.patch.object(windows.wmi, "WMI", return_value=wmi):
enumif = windows.EnumerateInterfaces()
enumif.SendReply = replies.append
enumif.Run(None)
self.assertLen(replies, 1)
interface = replies[0]
self.assertLen(interface.addresses, 4)
addresses = [x.human_readable_address for x in interface.addresses]
self.assertCountEqual(addresses, [
"192.168.1.20", "ffff::ffff:aaaa:1111:aaaa",
"dddd:0:8888:6666:bbbb:aaaa:eeee:bbbb",
"dddd:0:8888:6666:bbbb:aaaa:ffff:bbbb"
])