blob: 177f55777ef0a96349d7cb53ce5ff15c7bb369a9 [file] [log] [blame]
#!/usr/bin/python -S
"""Tests for quantenna.py."""
import quantenna
import utils
from wvtest import wvtest
@wvtest.wvtest
def get_external_mac_test():
old_get_mac_address_for_interface = utils.get_mac_address_for_interface
utils.get_mac_address_for_interface = lambda _: '02:00:00:00:00:00'
wvtest.WVPASSEQ(quantenna._get_external_mac('wlan0'), '00:00:00:00:00:00')
utils.get_mac_address_for_interface = old_get_mac_address_for_interface
@wvtest.wvtest
def get_vlan_test():
old_read_or_empty = utils.read_or_empty
utils.read_or_empty = lambda _: 'wlan0 VID: 3 REORDER_HDR: 1'
wvtest.WVPASSEQ(quantenna._get_vlan('wlan0'), 3)
utils.read_or_empty = lambda _: ''
wvtest.WVEXCEPT(utils.BinWifiException, quantenna._get_vlan, 'wlan0')
utils.read_or_empty = old_read_or_empty
@wvtest.wvtest
def get_interface_test():
old_get_quantenna_interfaces = quantenna._get_quantenna_interfaces
old_get_external_mac = quantenna._get_external_mac
old_get_vlan = quantenna._get_vlan
quantenna._get_quantenna_interfaces = lambda: ['wlan0', 'wlan0_portal']
quantenna._get_external_mac = lambda _: '00:00:00:00:00:00'
quantenna._get_vlan = lambda _: 3
wvtest.WVPASSEQ(quantenna._get_interface('ap', ''),
('wlan0', 'wifi1', '00:00:00:00:00:00', 3))
wvtest.WVPASSEQ(quantenna._get_interface('ap', 'portal'),
('wlan0_portal', 'wifi1', '00:00:00:00:00:00', 3))
wvtest.WVPASSEQ(quantenna._get_interface('sta', ''),
(None, None, None, None))
quantenna._get_vlan = old_get_vlan
quantenna._get_external_mac = old_get_external_mac
quantenna._get_quantenna_interfaces = old_get_quantenna_interfaces
@wvtest.wvtest
def parse_scan_result_test():
result = ' " ssid with "quotes" " 00:11:22:33:44:55 40 25 0 0 0 0 0 1 40 '
wvtest.WVPASSEQ(quantenna._parse_scan_result(result),
(' ssid with "quotes" ', '00:11:22:33:44:55', 40, -25, 0, 0))
if __name__ == '__main__':
wvtest.wvtest_main()