1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
# Basic Wi-Fi MAC layer test where one device creates an Access Point and the
# other device connects to it as a Station. Also tests channel assignment (where
# possible) and disconnection.
try:
import network
network.WLAN
except (ImportError, NameError):
print("SKIP")
raise SystemExit
import os
import sys
import time
CHANNEL = 8
# Note that on slower Wi-Fi stacks this bumps up against the run-multitests.py
# timeout which expects <10s between lines of output. We work around this by
# logging something half way through the wait_for loop...
CONNECT_TIMEOUT = 15000
def wait_for(test_func):
has_printed = False
start = time.ticks_ms()
while not test_func():
time.sleep(0.1)
delta = time.ticks_diff(time.ticks_ms(), start)
if not has_printed and delta > CONNECT_TIMEOUT / 2:
print("...")
has_printed = True
elif delta > CONNECT_TIMEOUT:
break
if not has_printed:
print("...") # keep the output consistent
return test_func()
# AP
def instance0():
ap = network.WLAN(network.WLAN.IF_AP)
ssid = "MP-test-" + os.urandom(6).hex()
psk = "Secret-" + os.urandom(6).hex()
# stop any previous activity
network.WLAN(network.WLAN.IF_STA).active(False)
ap.active(False)
ap.active(True)
ap.config(ssid=ssid, key=psk, channel=CHANNEL, security=network.WLAN.SEC_WPA_WPA2)
# print("AP setup", ssid, psk)
print("AP started")
multitest.globals(SSID=ssid, PSK=psk)
multitest.next()
# Wait for station
if not wait_for(ap.isconnected):
raise RuntimeError("Timed out waiting for station, status ", ap.status())
print("AP got station")
time.sleep(
3
) # depending on port, may still need to negotiate DHCP lease for STA to see connection
print("AP disabling...")
ap.active(False)
# STA
def instance1():
sta = network.WLAN(network.WLAN.IF_STA)
# stop any previous activity
network.WLAN(network.WLAN.IF_AP).active(False)
sta.active(False)
multitest.next()
ssid = SSID
psk = PSK
# print("STA setup", ssid, psk)
sta.active(True)
sta.connect(ssid, psk)
print("STA connecting...")
if not wait_for(sta.isconnected):
raise RuntimeError("Timed out waiting to connect, status ", sta.status())
print("STA connected")
print("channel", sta.config("channel"))
print("STA waiting for disconnect...")
# Expect the AP to disconnect us immediately
if not wait_for(lambda: not sta.isconnected()):
raise RuntimeError("Timed out waiting for AP to disconnect us, status ", sta.status())
print("STA disconnected")
sta.active(False)
|
