style: format python files with isort and double-quote-string-fixer

examples/protocols/mqtt/ssl/mqtt_ssl_example_test.py

@@ -1,29 +1,28 @@
-from __future__ import print_function
-from __future__ import unicode_literals
-from builtins import str
-import re
+from __future__ import print_function, unicode_literals
+
 import os
-import sys
+import re
 import ssl
+import sys
+from builtins import str
+from threading import Event, Thread
+
 import paho.mqtt.client as mqtt
-from threading import Thread, Event
-
-from tiny_test_fw import DUT
 import ttfw_idf
-
+from tiny_test_fw import DUT
 
 event_client_connected = Event()
 event_stop_client = Event()
 event_client_received_correct = Event()
 event_client_received_binary = Event()
-message_log = ""
+message_log = ''
 
 
 # The callback for when the client receives a CONNACK response from the server.
 def on_connect(client, userdata, flags, rc):
-    print("Connected with result code " + str(rc))
+    print('Connected with result code ' + str(rc))
     event_client_connected.set()
-    client.subscribe("/topic/qos0")
+    client.subscribe('/topic/qos0')
 
 
 def mqtt_client_task(client):
@@ -36,33 +35,33 @@ def on_message(client, userdata, msg):
     global message_log
     global event_client_received_correct
     global event_client_received_binary
-    if msg.topic == "/topic/binary":
+    if msg.topic == '/topic/binary':
         binary = userdata
         size = os.path.getsize(binary)
-        print("Receiving binary from esp and comparing with {}, size {}...".format(binary, size))
-        with open(binary, "rb") as f:
+        print('Receiving binary from esp and comparing with {}, size {}...'.format(binary, size))
+        with open(binary, 'rb') as f:
             bin = f.read()
             if bin == msg.payload[:size]:
-                print("...matches!")
+                print('...matches!')
                 event_client_received_binary.set()
                 return
             else:
-                recv_binary = binary + ".received"
-                with open(recv_binary, "w") as fw:
+                recv_binary = binary + '.received'
+                with open(recv_binary, 'w') as fw:
                     fw.write(msg.payload)
                 raise ValueError('Received binary (saved as: {}) does not match the original file: {}'.format(recv_binary, binary))
     payload = msg.payload.decode()
-    if not event_client_received_correct.is_set() and payload == "data":
-        client.subscribe("/topic/binary")
-        client.publish("/topic/qos0", "send binary please")
-        if msg.topic == "/topic/qos0" and payload == "data":
+    if not event_client_received_correct.is_set() and payload == 'data':
+        client.subscribe('/topic/binary')
+        client.publish('/topic/qos0', 'send binary please')
+        if msg.topic == '/topic/qos0' and payload == 'data':
             event_client_received_correct.set()
-    message_log += "Received data:" + msg.topic + " " + payload + "\n"
+    message_log += 'Received data:' + msg.topic + ' ' + payload + '\n'
 
 
-@ttfw_idf.idf_example_test(env_tag="Example_WIFI")
+@ttfw_idf.idf_example_test(env_tag='Example_WIFI')
 def test_examples_protocol_mqtt_ssl(env, extra_data):
-    broker_url = ""
+    broker_url = ''
     broker_port = 0
     """
     steps: |
@@ -72,15 +71,15 @@ def test_examples_protocol_mqtt_ssl(env, extra_data):
       4. Test ESP32 client received correct qos0 message
       5. Test python client receives binary data from running partition and compares it with the binary
     """
-    dut1 = env.get_dut("mqtt_ssl", "examples/protocols/mqtt/ssl", dut_class=ttfw_idf.ESP32DUT)
+    dut1 = env.get_dut('mqtt_ssl', 'examples/protocols/mqtt/ssl', dut_class=ttfw_idf.ESP32DUT)
     # check and log bin size
-    binary_file = os.path.join(dut1.app.binary_path, "mqtt_ssl.bin")
+    binary_file = os.path.join(dut1.app.binary_path, 'mqtt_ssl.bin')
     bin_size = os.path.getsize(binary_file)
-    ttfw_idf.log_performance("mqtt_ssl_bin_size", "{}KB"
+    ttfw_idf.log_performance('mqtt_ssl_bin_size', '{}KB'
                              .format(bin_size // 1024))
     # Look for host:port in sdkconfig
     try:
-        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()["CONFIG_BROKER_URI"])
+        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()['CONFIG_BROKER_URI'])
         broker_url = value.group(1)
         broker_port = int(value.group(2))
     except Exception:
@@ -97,31 +96,31 @@ def test_examples_protocol_mqtt_ssl(env, extra_data):
                        None,
                        None, cert_reqs=ssl.CERT_NONE, tls_version=ssl.PROTOCOL_TLSv1_2, ciphers=None)
         client.tls_insecure_set(True)
-        print("Connecting...")
+        print('Connecting...')
         client.connect(broker_url, broker_port, 60)
     except Exception:
-        print("ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:".format(broker_url, sys.exc_info()[0]))
+        print('ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:'.format(broker_url, sys.exc_info()[0]))
         raise
     # Starting a py-client in a separate thread
     thread1 = Thread(target=mqtt_client_task, args=(client,))
     thread1.start()
     try:
-        print("Connecting py-client to broker {}:{}...".format(broker_url, broker_port))
+        print('Connecting py-client to broker {}:{}...'.format(broker_url, broker_port))
         if not event_client_connected.wait(timeout=30):
-            raise ValueError("ENV_TEST_FAILURE: Test script cannot connect to broker: {}".format(broker_url))
+            raise ValueError('ENV_TEST_FAILURE: Test script cannot connect to broker: {}'.format(broker_url))
         dut1.start_app()
         try:
-            ip_address = dut1.expect(re.compile(r" sta ip: ([^,]+),"), timeout=30)
-            print("Connected to AP with IP: {}".format(ip_address))
+            ip_address = dut1.expect(re.compile(r' sta ip: ([^,]+),'), timeout=30)
+            print('Connected to AP with IP: {}'.format(ip_address))
         except DUT.ExpectTimeout:
             print('ENV_TEST_FAILURE: Cannot connect to AP')
             raise
-        print("Checking py-client received msg published from esp...")
+        print('Checking py-client received msg published from esp...')
         if not event_client_received_correct.wait(timeout=30):
             raise ValueError('Wrong data received, msg log: {}'.format(message_log))
-        print("Checking esp-client received msg published from py-client...")
-        dut1.expect(re.compile(r"DATA=send binary please"), timeout=30)
-        print("Receiving binary data from running partition...")
+        print('Checking esp-client received msg published from py-client...')
+        dut1.expect(re.compile(r'DATA=send binary please'), timeout=30)
+        print('Receiving binary data from running partition...')
         if not event_client_received_binary.wait(timeout=30):
             raise ValueError('Binary not received within timeout')
     finally:

examples/protocols/mqtt/tcp/mqtt_tcp_example_test.py

@@ -1,20 +1,20 @@
-import re
 import os
-import sys
+import re
 import socket
-from threading import Thread
 import struct
+import sys
 import time
+from threading import Thread
 
-from tiny_test_fw import DUT
 import ttfw_idf
+from tiny_test_fw import DUT
 
 msgid = -1
 
 
 def get_my_ip():
     s1 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-    s1.connect(("8.8.8.8", 80))
+    s1.connect(('8.8.8.8', 80))
     my_ip = s1.getsockname()[0]
     s1.close()
     return my_ip
@@ -22,7 +22,7 @@ def get_my_ip():
 
 def mqqt_server_sketch(my_ip, port):
     global msgid
-    print("Starting the server on {}".format(my_ip))
+    print('Starting the server on {}'.format(my_ip))
     s = None
     try:
         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -31,29 +31,29 @@ def mqqt_server_sketch(my_ip, port):
         s.listen(1)
         q,addr = s.accept()
         q.settimeout(30)
-        print("connection accepted")
+        print('connection accepted')
     except Exception:
-        print("Local server on {}:{} listening/accepting failure: {}"
-              "Possibly check permissions or firewall settings"
-              "to accept connections on this address".format(my_ip, port, sys.exc_info()[0]))
+        print('Local server on {}:{} listening/accepting failure: {}'
+              'Possibly check permissions or firewall settings'
+              'to accept connections on this address'.format(my_ip, port, sys.exc_info()[0]))
         raise
     data = q.recv(1024)
     # check if received initial empty message
-    print("received from client {}".format(data))
+    print('received from client {}'.format(data))
     data = bytearray([0x20, 0x02, 0x00, 0x00])
     q.send(data)
     # try to receive qos1
     data = q.recv(1024)
-    msgid = struct.unpack(">H", data[15:17])[0]
-    print("received from client {}, msgid: {}".format(data, msgid))
+    msgid = struct.unpack('>H', data[15:17])[0]
+    print('received from client {}, msgid: {}'.format(data, msgid))
     data = bytearray([0x40, 0x02, data[15], data[16]])
     q.send(data)
     time.sleep(5)
     s.close()
-    print("server closed")
+    print('server closed')
 
 
-@ttfw_idf.idf_example_test(env_tag="Example_WIFI")
+@ttfw_idf.idf_example_test(env_tag='Example_WIFI')
 def test_examples_protocol_mqtt_qos1(env, extra_data):
     global msgid
     """
@@ -63,11 +63,11 @@ def test_examples_protocol_mqtt_qos1(env, extra_data):
       3. Test evaluates that qos1 message is queued and removed from queued after ACK received
       4. Test the broker received the same message id evaluated in step 3
     """
-    dut1 = env.get_dut("mqtt_tcp", "examples/protocols/mqtt/tcp", dut_class=ttfw_idf.ESP32DUT)
+    dut1 = env.get_dut('mqtt_tcp', 'examples/protocols/mqtt/tcp', dut_class=ttfw_idf.ESP32DUT)
     # check and log bin size
-    binary_file = os.path.join(dut1.app.binary_path, "mqtt_tcp.bin")
+    binary_file = os.path.join(dut1.app.binary_path, 'mqtt_tcp.bin')
     bin_size = os.path.getsize(binary_file)
-    ttfw_idf.log_performance("mqtt_tcp_bin_size", "{}KB".format(bin_size // 1024))
+    ttfw_idf.log_performance('mqtt_tcp_bin_size', '{}KB'.format(bin_size // 1024))
     # 1. start mqtt broker sketch
     host_ip = get_my_ip()
     thread1 = Thread(target=mqqt_server_sketch, args=(host_ip,1883))
@@ -76,23 +76,23 @@ def test_examples_protocol_mqtt_qos1(env, extra_data):
     dut1.start_app()
     # waiting for getting the IP address
     try:
-        ip_address = dut1.expect(re.compile(r" sta ip: ([^,]+),"), timeout=30)
-        print("Connected to AP with IP: {}".format(ip_address))
+        ip_address = dut1.expect(re.compile(r' sta ip: ([^,]+),'), timeout=30)
+        print('Connected to AP with IP: {}'.format(ip_address))
     except DUT.ExpectTimeout:
         raise ValueError('ENV_TEST_FAILURE: Cannot connect to AP')
 
-    print("writing to device: {}".format("mqtt://" + host_ip + "\n"))
-    dut1.write("mqtt://" + host_ip + "\n")
+    print('writing to device: {}'.format('mqtt://' + host_ip + '\n'))
+    dut1.write('mqtt://' + host_ip + '\n')
     thread1.join()
-    print("Message id received from server: {}".format(msgid))
+    print('Message id received from server: {}'.format(msgid))
     # 3. check the message id was enqueued and then deleted
-    msgid_enqueued = dut1.expect(re.compile(r"OUTBOX: ENQUEUE msgid=([0-9]+)"), timeout=30)
-    msgid_deleted = dut1.expect(re.compile(r"OUTBOX: DELETED msgid=([0-9]+)"), timeout=30)
+    msgid_enqueued = dut1.expect(re.compile(r'OUTBOX: ENQUEUE msgid=([0-9]+)'), timeout=30)
+    msgid_deleted = dut1.expect(re.compile(r'OUTBOX: DELETED msgid=([0-9]+)'), timeout=30)
     # 4. check the msgid of received data are the same as that of enqueued and deleted from outbox
     if (msgid_enqueued[0] == str(msgid) and msgid_deleted[0] == str(msgid)):
-        print("PASS: Received correct msg id")
+        print('PASS: Received correct msg id')
     else:
-        print("Failure!")
+        print('Failure!')
         raise ValueError('Mismatch of msgid: received: {}, enqueued {}, deleted {}'.format(msgid, msgid_enqueued, msgid_deleted))
 
 

examples/protocols/mqtt/ws/mqtt_ws_example_test.py

@@ -1,26 +1,26 @@
-from __future__ import print_function
-from __future__ import unicode_literals
-from builtins import str
-import re
-import os
-import sys
-import paho.mqtt.client as mqtt
-from threading import Thread, Event
+from __future__ import print_function, unicode_literals
 
-from tiny_test_fw import DUT
+import os
+import re
+import sys
+from builtins import str
+from threading import Event, Thread
+
+import paho.mqtt.client as mqtt
 import ttfw_idf
+from tiny_test_fw import DUT
 
 event_client_connected = Event()
 event_stop_client = Event()
 event_client_received_correct = Event()
-message_log = ""
+message_log = ''
 
 
 # The callback for when the client receives a CONNACK response from the server.
 def on_connect(client, userdata, flags, rc):
-    print("Connected with result code " + str(rc))
+    print('Connected with result code ' + str(rc))
     event_client_connected.set()
-    client.subscribe("/topic/qos0")
+    client.subscribe('/topic/qos0')
 
 
 def mqtt_client_task(client):
@@ -32,16 +32,16 @@ def mqtt_client_task(client):
 def on_message(client, userdata, msg):
     global message_log
     payload = msg.payload.decode()
-    if not event_client_received_correct.is_set() and payload == "data":
-        client.publish("/topic/qos0", "data_to_esp32")
-        if msg.topic == "/topic/qos0" and payload == "data":
+    if not event_client_received_correct.is_set() and payload == 'data':
+        client.publish('/topic/qos0', 'data_to_esp32')
+        if msg.topic == '/topic/qos0' and payload == 'data':
             event_client_received_correct.set()
-    message_log += "Received data:" + msg.topic + " " + payload + "\n"
+    message_log += 'Received data:' + msg.topic + ' ' + payload + '\n'
 
 
-@ttfw_idf.idf_example_test(env_tag="Example_WIFI")
+@ttfw_idf.idf_example_test(env_tag='Example_WIFI')
 def test_examples_protocol_mqtt_ws(env, extra_data):
-    broker_url = ""
+    broker_url = ''
     broker_port = 0
     """
     steps: |
@@ -50,14 +50,14 @@ def test_examples_protocol_mqtt_ws(env, extra_data):
       3. Test evaluates it received correct qos0 message
       4. Test ESP32 client received correct qos0 message
     """
-    dut1 = env.get_dut("mqtt_websocket", "examples/protocols/mqtt/ws", dut_class=ttfw_idf.ESP32DUT)
+    dut1 = env.get_dut('mqtt_websocket', 'examples/protocols/mqtt/ws', dut_class=ttfw_idf.ESP32DUT)
     # check and log bin size
-    binary_file = os.path.join(dut1.app.binary_path, "mqtt_websocket.bin")
+    binary_file = os.path.join(dut1.app.binary_path, 'mqtt_websocket.bin')
     bin_size = os.path.getsize(binary_file)
-    ttfw_idf.log_performance("mqtt_websocket_bin_size", "{}KB".format(bin_size // 1024))
+    ttfw_idf.log_performance('mqtt_websocket_bin_size', '{}KB'.format(bin_size // 1024))
     # Look for host:port in sdkconfig
     try:
-        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()["CONFIG_BROKER_URI"])
+        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()['CONFIG_BROKER_URI'])
         broker_url = value.group(1)
         broker_port = int(value.group(2))
     except Exception:
@@ -66,33 +66,33 @@ def test_examples_protocol_mqtt_ws(env, extra_data):
     client = None
     # 1. Test connects to a broker
     try:
-        client = mqtt.Client(transport="websockets")
+        client = mqtt.Client(transport='websockets')
         client.on_connect = on_connect
         client.on_message = on_message
-        print("Connecting...")
+        print('Connecting...')
         client.connect(broker_url, broker_port, 60)
     except Exception:
-        print("ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:".format(broker_url, sys.exc_info()[0]))
+        print('ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:'.format(broker_url, sys.exc_info()[0]))
         raise
     # Starting a py-client in a separate thread
     thread1 = Thread(target=mqtt_client_task, args=(client,))
     thread1.start()
     try:
-        print("Connecting py-client to broker {}:{}...".format(broker_url, broker_port))
+        print('Connecting py-client to broker {}:{}...'.format(broker_url, broker_port))
         if not event_client_connected.wait(timeout=30):
-            raise ValueError("ENV_TEST_FAILURE: Test script cannot connect to broker: {}".format(broker_url))
+            raise ValueError('ENV_TEST_FAILURE: Test script cannot connect to broker: {}'.format(broker_url))
         dut1.start_app()
         try:
-            ip_address = dut1.expect(re.compile(r" sta ip: ([^,]+),"), timeout=30)
-            print("Connected to AP with IP: {}".format(ip_address))
+            ip_address = dut1.expect(re.compile(r' sta ip: ([^,]+),'), timeout=30)
+            print('Connected to AP with IP: {}'.format(ip_address))
         except DUT.ExpectTimeout:
             print('ENV_TEST_FAILURE: Cannot connect to AP')
             raise
-        print("Checking py-client received msg published from esp...")
+        print('Checking py-client received msg published from esp...')
         if not event_client_received_correct.wait(timeout=30):
             raise ValueError('Wrong data received, msg log: {}'.format(message_log))
-        print("Checking esp-client received msg published from py-client...")
-        dut1.expect(re.compile(r"DATA=data_to_esp32"), timeout=30)
+        print('Checking esp-client received msg published from py-client...')
+        dut1.expect(re.compile(r'DATA=data_to_esp32'), timeout=30)
     finally:
         event_stop_client.set()
         thread1.join()

examples/protocols/mqtt/wss/mqtt_wss_example_test.py

@@ -1,28 +1,27 @@
 from __future__ import unicode_literals
-from __future__ import unicode_literals
-from builtins import str
-import re
+
 import os
-import sys
+import re
 import ssl
+import sys
+from builtins import str
+from threading import Event, Thread
+
 import paho.mqtt.client as mqtt
-from threading import Thread, Event
-
-from tiny_test_fw import DUT
 import ttfw_idf
-
+from tiny_test_fw import DUT
 
 event_client_connected = Event()
 event_stop_client = Event()
 event_client_received_correct = Event()
-message_log = ""
+message_log = ''
 
 
 # The callback for when the client receives a CONNACK response from the server.
 def on_connect(client, userdata, flags, rc):
-    print("Connected with result code " + str(rc))
+    print('Connected with result code ' + str(rc))
     event_client_connected.set()
-    client.subscribe("/topic/qos0")
+    client.subscribe('/topic/qos0')
 
 
 def mqtt_client_task(client):
@@ -34,16 +33,16 @@ def mqtt_client_task(client):
 def on_message(client, userdata, msg):
     global message_log
     payload = msg.payload.decode()
-    if not event_client_received_correct.is_set() and payload == "data":
-        client.publish("/topic/qos0", "data_to_esp32")
-        if msg.topic == "/topic/qos0" and payload == "data":
+    if not event_client_received_correct.is_set() and payload == 'data':
+        client.publish('/topic/qos0', 'data_to_esp32')
+        if msg.topic == '/topic/qos0' and payload == 'data':
             event_client_received_correct.set()
-    message_log += "Received data:" + msg.topic + " " + payload + "\n"
+    message_log += 'Received data:' + msg.topic + ' ' + payload + '\n'
 
 
-@ttfw_idf.idf_example_test(env_tag="Example_WIFI")
+@ttfw_idf.idf_example_test(env_tag='Example_WIFI')
 def test_examples_protocol_mqtt_wss(env, extra_data):
-    broker_url = ""
+    broker_url = ''
     broker_port = 0
     """
     steps: |
@@ -52,14 +51,14 @@ def test_examples_protocol_mqtt_wss(env, extra_data):
       3. Test evaluates it received correct qos0 message
       4. Test ESP32 client received correct qos0 message
     """
-    dut1 = env.get_dut("mqtt_websocket_secure", "examples/protocols/mqtt/wss", dut_class=ttfw_idf.ESP32DUT)
+    dut1 = env.get_dut('mqtt_websocket_secure', 'examples/protocols/mqtt/wss', dut_class=ttfw_idf.ESP32DUT)
     # check and log bin size
-    binary_file = os.path.join(dut1.app.binary_path, "mqtt_websocket_secure.bin")
+    binary_file = os.path.join(dut1.app.binary_path, 'mqtt_websocket_secure.bin')
     bin_size = os.path.getsize(binary_file)
-    ttfw_idf.log_performance("mqtt_websocket_secure_bin_size", "{}KB".format(bin_size // 1024))
+    ttfw_idf.log_performance('mqtt_websocket_secure_bin_size', '{}KB'.format(bin_size // 1024))
     # Look for host:port in sdkconfig
     try:
-        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()["CONFIG_BROKER_URI"])
+        value = re.search(r'\:\/\/([^:]+)\:([0-9]+)', dut1.app.get_sdkconfig()['CONFIG_BROKER_URI'])
         broker_url = value.group(1)
         broker_port = int(value.group(2))
     except Exception:
@@ -68,36 +67,36 @@ def test_examples_protocol_mqtt_wss(env, extra_data):
     client = None
     # 1. Test connects to a broker
     try:
-        client = mqtt.Client(transport="websockets")
+        client = mqtt.Client(transport='websockets')
         client.on_connect = on_connect
         client.on_message = on_message
         client.tls_set(None,
                        None,
                        None, cert_reqs=ssl.CERT_NONE, tls_version=ssl.PROTOCOL_TLSv1_2, ciphers=None)
-        print("Connecting...")
+        print('Connecting...')
         client.connect(broker_url, broker_port, 60)
     except Exception:
-        print("ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:".format(broker_url, sys.exc_info()[0]))
+        print('ENV_TEST_FAILURE: Unexpected error while connecting to broker {}: {}:'.format(broker_url, sys.exc_info()[0]))
         raise
     # Starting a py-client in a separate thread
     thread1 = Thread(target=mqtt_client_task, args=(client,))
     thread1.start()
     try:
-        print("Connecting py-client to broker {}:{}...".format(broker_url, broker_port))
+        print('Connecting py-client to broker {}:{}...'.format(broker_url, broker_port))
         if not event_client_connected.wait(timeout=30):
-            raise ValueError("ENV_TEST_FAILURE: Test script cannot connect to broker: {}".format(broker_url))
+            raise ValueError('ENV_TEST_FAILURE: Test script cannot connect to broker: {}'.format(broker_url))
         dut1.start_app()
         try:
-            ip_address = dut1.expect(re.compile(r" sta ip: ([^,]+),"), timeout=30)
-            print("Connected to AP with IP: {}".format(ip_address))
+            ip_address = dut1.expect(re.compile(r' sta ip: ([^,]+),'), timeout=30)
+            print('Connected to AP with IP: {}'.format(ip_address))
         except DUT.ExpectTimeout:
             print('ENV_TEST_FAILURE: Cannot connect to AP')
             raise
-        print("Checking py-client received msg published from esp...")
+        print('Checking py-client received msg published from esp...')
         if not event_client_received_correct.wait(timeout=30):
             raise ValueError('Wrong data received, msg log: {}'.format(message_log))
-        print("Checking esp-client received msg published from py-client...")
-        dut1.expect(re.compile(r"DATA=data_to_esp32"), timeout=30)
+        print('Checking esp-client received msg published from py-client...')
+        dut1.expect(re.compile(r'DATA=data_to_esp32'), timeout=30)
     finally:
         event_stop_client.set()
         thread1.join()