Method test_get_handler_fallback

tests/test_websocket.py:83–92

(self)

        self.assertEqual(result, {"statusCode": 200, "body": "connected"})

    def test_get_handler_fallback(self):
        from zappa.websocket import get_handler, on_message

        @on_message
        def handle_msg(event, context):
            return {"statusCode": 200}

        # Unknown route should fall back to $default
        handler = get_handler("$unknown")
        self.assertIs(handler, handle_msg)

    def test_get_handler_returns_none_when_empty(self):
        from zappa.websocket import get_handler