def test_get_multiple_wifi_devices(mocker): from ideascube.serveradmin.wifi import get_wifi_device, WifiError NM = mocker.patch('ideascube.serveradmin.wifi.NetworkManager') NM.Devices = [NMDevice(True), NMDevice(True)] with pytest.raises(WifiError): get_wifi_device()
def test_get_wifi_device(mocker): from ideascube.serveradmin.wifi import get_wifi_device, NM_DEVICE_TYPE_WIFI NM = mocker.patch('ideascube.serveradmin.wifi.NetworkManager') NM.Devices = [NMDevice(False), NMDevice(True)] device = get_wifi_device() assert device.DeviceType == NM_DEVICE_TYPE_WIFI