def test_get_interfaces_invalid_ap_argument(self): """ Test get_interfaces method with a invalid AP argument """ # create fake interfaces interface0 = interfaces.NetworkAdapter("wlan0") interface1 = interfaces.NetworkAdapter("wlan1") # make both interfaces support AP and monitor interface0.set_ap_support(True) interface1.set_ap_support(True) # add the interfaces to the list network_manager = interfaces.NetworkManager(None, "wlan3") network_manager._interfaces.append(interface0) network_manager._interfaces.append(interface1) self.assertRaises(interfaces.ApInterfaceInvalidError, network_manager.get_interfaces)
def test_get_interfaces_valid_argument(self): """ Test get_interfaces method with a valid arguments for both jamming and AP arguments """ # create fake interfaces interface0 = interfaces.NetworkAdapter("wlan0") interface1 = interfaces.NetworkAdapter("wlan1") # make both interfaces support AP and monitor interface0.set_monitor_support(True) interface1.set_ap_support(True) # add the interfaces to the list network_manager = interfaces.NetworkManager("wlan0", "wlan1") network_manager._interfaces.append(interface0) network_manager._interfaces.append(interface1) expected = ("wlan0", "wlan1") self.assertEqual(network_manager.get_interfaces(), expected)
def test_get_interfaces_valid_monitor_argument_no_ap(self): """ Test get_interfaces method with a valid monitor argument but no interface with AP mode """ # create fake interfaces interface0 = interfaces.NetworkAdapter("wlan0") interface1 = interfaces.NetworkAdapter("wlan1") # make both interfaces support AP and monitor interface0.set_monitor_support(True) interface1.set_monitor_support(True) # add the interfaces to the list network_manager = interfaces.NetworkManager("wlan0", None) network_manager._interfaces.append(interface0) network_manager._interfaces.append(interface1) self.assertRaises(interfaces.NoApInterfaceFoundError, network_manager.get_interfaces)
def setUp(self): """ Set up the tests """ self.network_manager = interfaces.NetworkManager(None, None)