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)
