espressif/openthread
1
0
Fork 0
mirror of https://github.com/espressif/openthread.git synced 2026-08-05 10:27:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

[style] apply google python style guide (#4501)

Browse Source 
This commit applies and enforces Google's python style for tests.
This commit is contained in:
Yakun Xu
2020-02-04 10:27:50 -08:00
committed by GitHub
parent 8368d440dd
commit 33808ebfba
501 changed files with 6674 additions and 5515 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/scripts/thread-cert/simulator.py
+18 -31
View File
@@ -48,12 +48,12 @@ def dbg_print(*args):
class BaseSimulator(object):
def __init__(self):
self._nodes = {}
self.commissioning_messages = {}
self._payload_parse_factory = mesh_cop.MeshCopCommandFactory(
mesh_cop.create_default_mesh_cop_tlv_factories()
)
mesh_cop.create_default_mesh_cop_tlv_factories())
self._mesh_cop_msg_set = mesh_cop.create_mesh_cop_message_type_set()
def __del__(self):
@@ -81,18 +81,18 @@ class BaseSimulator(object):
if not node:
continue
for (
direction,
type,
payload,
direction,
type,
payload,
) in node.read_cert_messages_in_commissioning_log():
if direction == b'send':
msg = self._payload_parse_factory.parse(
type.decode("utf-8"), io.BytesIO(payload)
)
type.decode("utf-8"), io.BytesIO(payload))
self.commissioning_messages[nodeid].append(msg)
class RealTime(BaseSimulator):
def __init__(self):
super(RealTime, self).__init__()
self._sniffer = config.create_default_thread_sniffer()
@@ -103,9 +103,7 @@ class RealTime(BaseSimulator):
def get_messages_sent_by(self, nodeid):
messages = self._sniffer.get_messages_sent_by(nodeid).messages
ret = message.MessagesSet(
messages, self.commissioning_messages[nodeid]
)
ret = message.MessagesSet(messages, self.commissioning_messages[nodeid])
self.commissioning_messages[nodeid] = []
return ret
@@ -207,9 +205,7 @@ class VirtualTime(BaseSimulator):
messages = self.devices[addr]['msgs']
self.devices[addr]['msgs'] = []
ret = message.MessagesSet(
messages, self.commissioning_messages[nodeid]
)
ret = message.MessagesSet(messages, self.commissioning_messages[nodeid])
self.commissioning_messages[nodeid] = []
return ret
@@ -239,14 +235,9 @@ class VirtualTime(BaseSimulator):
def receive_events(self):
""" Receive events until all devices are asleep. """
while True:
if (
self.current_event
or len(self.awake_devices)
or (
self._next_event_time() > self._pause_time
and self.current_nodeid
)
):
if (self.current_event or len(self.awake_devices) or
(self._next_event_time() > self._pause_time and
self.current_nodeid)):
self.sock.settimeout(self.BLOCK_TIMEOUT)
try:
msg, addr = self.sock.recvfrom(self.MAX_MESSAGE)
@@ -311,10 +302,8 @@ class VirtualTime(BaseSimulator):
self.awake_devices.discard(addr)
if (
self.current_event
and self.current_event[self.EVENT_ADDR] == addr
):
if (self.current_event and
self.current_event[self.EVENT_ADDR] == addr):
# print "Done\t", self.current_event
self.current_event = None
@@ -335,9 +324,8 @@ class VirtualTime(BaseSimulator):
# print "-- Enqueue\t", event
bisect.insort(self.event_queue, event)
self._pcap.append(
data, (event_time // 1000000, event_time % 1000000)
)
self._pcap.append(data,
(event_time // 1000000, event_time % 1000000))
self._add_message(addr[1] - self.port, data)
# add radio transmit done events to event queue
@@ -461,9 +449,8 @@ class VirtualTime(BaseSimulator):
continue
dbg_print('syncing', addr, elapsed)
self.devices[addr]['time'] = self.current_time
message = struct.pack(
'=QBH', elapsed, self.OT_SIM_EVENT_ALARM_FIRED, 0
)
message = struct.pack('=QBH', elapsed,
self.OT_SIM_EVENT_ALARM_FIRED, 0)
self._send_message(message, addr)
self.awake_devices.add(addr)
self.receive_events()