From 17e851e95f62f4672aac8415a042514349da942c Mon Sep 17 00:00:00 2001 From: Luna Date: Sat, 13 Aug 2022 22:39:41 -0300 Subject: [PATCH] gateway: add resume_gateway_url to ready close #143 --- litecord/gateway/websocket.py | 2 ++ tests/test_websocket.py | 1 + 2 files changed, 3 insertions(+) diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index c755185..de1489f 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -66,6 +66,7 @@ from litecord.gateway.schemas import ( ) from litecord.storage import int_ +from litecord.blueprints.gateway import get_gw log = Logger(__name__) @@ -523,6 +524,7 @@ class GatewayWebsocket: "session_id": self.state.session_id, "_trace": ["transbian"], "shard": [self.state.current_shard, self.state.shard_count], + "resume_gateway_url": get_gw(), } # base_ready and user_ready are normalized as v6. from here onwards diff --git a/tests/test_websocket.py b/tests/test_websocket.py index e8f7f00..8aad925 100644 --- a/tests/test_websocket.py +++ b/tests/test_websocket.py @@ -217,6 +217,7 @@ async def extract_and_verify_ready(conn, **kwargs): assert isinstance(data["guilds"], list) assert isinstance(data["session_id"], str) assert isinstance(data["_trace"], list) + assert isinstance(data["resume_gateway_url"], str) if "shard" in data: assert isinstance(data["shard"], list)