test: Add tests for device description, VID and PID

Author: nbuchwitz

tests/helper.py

@@ -45,6 +45,25 @@ def __power_status(self, port_number: int):
 
         return self.__status(self.status[port_number])
 
+    def register_port_details(
+        self,
+        fp: pytest_subprocess.FakeProcess,
+        port_number: int = None,
+        vendor_id: int = None,
+        product_id: int = None,
+        description: str = None,
+    ):
+        stdout = f"  Port {port_number}: 0103 power enable connect"
+        if vendor_id is not None and product_id is not None:
+            stdout += f" [{vendor_id:04x}:{product_id:04x}"
+
+            if description is not None:
+                stdout += f" {description}"
+
+            stdout += "]"
+
+        fp.register(["uhubctl", "-l", self.path, "-p", str(port_number)], stdout=stdout)
+
     def cmd(
         self,
         fp: pytest_subprocess.FakeProcess,

tests/test_port.py

@@ -61,3 +61,26 @@ def test_port_from_path():
 
         assert port.hub.path == path
         assert port.port_number == port_number
+
+
+def test_device_data(mock_hub: MockHub, fp: pytest_subprocess.FakeProcess):
+    port_number = 1
+    vendor_id = 0xDEAD
+    product_id = 0xBEEF
+    description = "Some fancy USB device"
+
+    port = Port(mock_hub, port_number)
+
+    mock_hub.register_port_details(fp, port_number)
+    assert port.description(cached_results=False) is None
+
+    mock_hub.register_port_details(fp, port_number)
+    assert port.vendor_id(cached_results=False) is None
+
+    mock_hub.register_port_details(fp, port_number)
+    assert port.product_id(cached_results=False) is None
+
+    mock_hub.register_port_details(fp, port_number, vendor_id, product_id, description)
+    assert port.description(cached_results=False) == description
+    assert port.vendor_id(cached_results=True) == vendor_id
+    assert port.product_id(cached_results=True) == product_id