/bin/wifi: report association result for Quantenna.
Change-Id: I7c26ae109541982490f26f9c82b2382b84594e18
diff --git a/wifi/quantenna.py b/wifi/quantenna.py
index 10af967..b88ccac 100755
--- a/wifi/quantenna.py
+++ b/wifi/quantenna.py
@@ -99,6 +99,13 @@
# 'apply_security_config' must be called instead.
_qcsapi('apply_security_config', 'wifi0')
+ for _ in xrange(10):
+ if _qcsapi('get_status', 'wifi0') == 'Up':
+ break
+ time.sleep(1)
+ else:
+ raise utils.BinWifiException('wpa_supplicant failed to connect')
+
return True
diff --git a/wifi/quantenna_test.py b/wifi/quantenna_test.py
index 72f0333..b68b239 100755
--- a/wifi/quantenna_test.py
+++ b/wifi/quantenna_test.py
@@ -16,7 +16,9 @@
def fake_qcsapi(*args):
calls.append(list(args))
if args[0] == 'is_startprod_done':
- return '1\n' if ['startprod', 'wifi0'] in calls else '0\n'
+ return '1' if ['startprod', 'wifi0'] in calls else '0'
+ if args[0] == 'get_status':
+ return 'Up' if ['get_status', 'wifi0'] in calls else 'Down'
bridge_interfaces = set()
