tiny-test-fw: handle case when ESPPORT is not set

Fix bug introduced in 8a88dd35, where .decode was called on result of os.getenv, which was None if ESPPORT was not set.
2018-07-02 21:45:27 +08:00
parent 3d880f59b9
commit eb85c4ec15
1 changed files with 5 additions and 1 deletions
@@ -131,7 +131,11 @@ class IDFDUT(DUT.SerialDUT):
    @classmethod
    def list_available_ports(cls):
        ports = [x.device for x in list_ports.comports()]
-        port_hint = os.getenv('ESPPORT').decode('utf8')
+        espport = os.getenv('ESPPORT')
+        if not espport:
+            return ports
+            
+        port_hint = espport.decode('utf8')

        # If $ESPPORT is a valid port, make it appear first in the list
        if port_hint in ports: