Implemented goto "Test Controller Buttons" menu

controller_state.py

@@ -0,0 +1,38 @@
+import asyncio
+
+from button_state import ButtonState
+from protocol import ControllerProtocol
+
+
+class ControllerState:
+    def __init__(self, transport: asyncio.Transport, protocol: ControllerProtocol):
+        super().__init__()
+        self.transport = transport
+
+        self.protocol = protocol
+
+    async def send(self):
+        await self.protocol.button_input_report.write(self.transport)
+
+    async def connect(self):
+        """
+        Waits until the switch is paired with the controller and accepts button commands
+        """
+        # TODO HACK: Hard to say for now.
+        await self.protocol.wait_for_output_report()
+        # The switch sends data to our device, it shouldn't take long until the connection is fully established.
+        await asyncio.sleep(5)
+
+    def set_button_state(self, button_state: ButtonState):
+        """
+        Sets the button status bytes in the input report
+        """
+        self.protocol.button_input_report.set_button_status(button_state)
+
+    def set_stick_state(self):
+        """
+        TODO
+        """
+        raise NotImplementedError()
+
+