61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from archinstall.lib.models.network import WifiNetwork
|
|
|
|
SAMPLE_SCAN_RESULTS = """bssid / frequency / signal level / flags / ssid
|
|
aa:bb:cc:dd:ee:01 2412 -40 [WPA2-PSK-CCMP][ESS] NoSpacesSSID
|
|
aa:bb:cc:dd:ee:02 2412 -50 [WPA2-PSK-CCMP][ESS] My Home Network
|
|
aa:bb:cc:dd:ee:03 5180 -60 [WPA2-PSK-CCMP][WPS][ESS] Free Public WiFi
|
|
aa:bb:cc:dd:ee:04 2412 -70 [WPA2-PSK-CCMP][ESS]
|
|
"""
|
|
|
|
|
|
def test_from_wpa_ssid_without_spaces() -> None:
|
|
networks = WifiNetwork.from_wpa(SAMPLE_SCAN_RESULTS)
|
|
ssids = [network.ssid for network in networks]
|
|
|
|
assert 'NoSpacesSSID' in ssids
|
|
|
|
|
|
def test_from_wpa_ssid_with_spaces() -> None:
|
|
networks = WifiNetwork.from_wpa(SAMPLE_SCAN_RESULTS)
|
|
ssids = [network.ssid for network in networks]
|
|
|
|
assert 'My Home Network' in ssids
|
|
|
|
|
|
def test_from_wpa_ssid_with_multiple_spaces() -> None:
|
|
networks = WifiNetwork.from_wpa(SAMPLE_SCAN_RESULTS)
|
|
ssids = [network.ssid for network in networks]
|
|
|
|
assert 'Free Public WiFi' in ssids
|
|
|
|
|
|
def test_from_wpa_empty_ssid() -> None:
|
|
networks = WifiNetwork.from_wpa(SAMPLE_SCAN_RESULTS)
|
|
empty = [network for network in networks if network.bssid == 'aa:bb:cc:dd:ee:04']
|
|
|
|
assert len(empty) == 1
|
|
assert empty[0].ssid == ''
|
|
|
|
|
|
def test_from_wpa_skips_header_and_blank_lines() -> None:
|
|
results = """
|
|
bssid / frequency / signal level / flags / ssid
|
|
|
|
aa:bb:cc:dd:ee:01 2412 -40 [WPA2-PSK-CCMP][ESS] SimpleNet
|
|
"""
|
|
networks = WifiNetwork.from_wpa(results)
|
|
|
|
assert len(networks) == 1
|
|
assert networks[0].ssid == 'SimpleNet'
|
|
assert networks[0].bssid == 'aa:bb:cc:dd:ee:01'
|
|
|
|
|
|
def test_from_wpa_preserves_flags_and_fields() -> None:
|
|
networks = WifiNetwork.from_wpa(SAMPLE_SCAN_RESULTS)
|
|
network = next(n for n in networks if n.ssid == 'My Home Network')
|
|
|
|
assert network.bssid == 'aa:bb:cc:dd:ee:02'
|
|
assert network.frequency == '2412'
|
|
assert network.signal_level == '-50'
|
|
assert network.flags == '[WPA2-PSK-CCMP][ESS]'
|