Use priming in speedtests

greatscottgadgets · Sep 6, 2024 · 7484c03 · 7484c03
1 parent 8385478
commit 7484c03
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/test/speedtest_fullspeed.py b/test/speedtest_fullspeed.py
@@ -66,6 +66,18 @@ def handle_data_requested(self, ep):
         print(f"sending {len(self.random_buffer)} bytes on {ep}")
         self.send(ep.number, self.random_buffer)
 
+    def handle_buffer_empty(self, endpoint: USBEndpoint):
+        """ Handler called when a given endpoint first has an empty buffer.
+
+        Often, an empty buffer indicates an opportunity to queue data
+        for sending ('prime an endpoint'), but doesn't necessarily mean
+        that the host is planning on reading the data.
+
+        This function is called only once per buffer.
+        """
+        print(f"priming {len(self.random_buffer)} bytes on {endpoint}")
+        self.send(endpoint.number, self.random_buffer)
+
 
 if __name__ == "__main__":
     default_main(USBSpeedtest)
diff --git a/test/speedtest_highspeed.py b/test/speedtest_highspeed.py
@@ -99,6 +99,17 @@ def handle_data_requested(self, ep):
         print(f"sending {len(self.random_buffer)} bytes on {ep}")
         self.send(ep.number, self.random_buffer)
 
+    def handle_buffer_empty(self, endpoint: USBEndpoint):
+        """ Handler called when a given endpoint first has an empty buffer.
+
+        Often, an empty buffer indicates an opportunity to queue data
+        for sending ('prime an endpoint'), but doesn't necessarily mean
+        that the host is planning on reading the data.
+
+        This function is called only once per buffer.
+        """
+        print(f"priming {len(self.random_buffer)} bytes on {endpoint}")
+        self.send(endpoint.number, self.random_buffer)
 
 if __name__ == "__main__":
     default_main(USBSpeedtest)