Test uploading firmware

test/firmware.py

@@ -0,0 +1,62 @@
+"""Test uploading firmware"""
+
+import time
+import unittest
+from multiprocessing import Process, Queue
+from queue import Empty
+
+from gpiozero import DigitalOutputDevice
+
+RESET_GPIO_NUMBER = 4
+BOOT0_GPIO_NUMBER = 22
+
+
+def resethat():
+    """Reset the HAT"""
+    reset = DigitalOutputDevice(RESET_GPIO_NUMBER)
+    boot0 = DigitalOutputDevice(BOOT0_GPIO_NUMBER)
+    boot0.off()
+    reset.off()
+    time.sleep(0.01)
+    reset.on()
+    time.sleep(0.01)
+    boot0.close()
+    reset.close()
+    time.sleep(0.5)
+
+
+def reboot(exc):
+    """Reboot hat and load firmware
+
+    :param exc: Queue to pass exceptions
+    """
+    try:
+        from buildhat import Hat
+        resethat()
+        h = Hat(debug=True)
+        print(h.get())
+    except Exception as e:
+        exc.put(e)
+
+
+class TestFirmware(unittest.TestCase):
+    """Test firmware uploading functions"""
+
+    def test_upload(self):
+        """Test upload firmware
+
+        :raises exc.get: Raised if exception in subprocess
+        """
+        for _ in range(500):
+            exc = Queue()
+            p = Process(target=reboot, args=(exc,))
+            p.start()
+            p.join()
+            try:
+                raise exc.get(False)
+            except Empty:
+                pass
+
+
+if __name__ == '__main__':
+    unittest.main()