From 680d7e01803a57f2ad5c16b12e4b81a62a130dd8 Mon Sep 17 00:00:00 2001 From: andrei Date: Mon, 24 Jul 2017 17:16:31 -0700 Subject: [PATCH] [tests] cleanup --- disco/gateway/events.py | 1 - tests/gateway/events.py | 24 +++++++++++++++++++++++- 2 files changed, 23 insertions(+), 2 deletions(-) diff --git a/disco/gateway/events.py b/disco/gateway/events.py index 7dc6d1f..8d5e882 100644 --- a/disco/gateway/events.py +++ b/disco/gateway/events.py @@ -414,7 +414,6 @@ class GuildRoleCreate(GatewayEvent): return self.client.state.guilds.get(self.guild_id) -@proxy('role') class GuildRoleUpdate(GuildRoleCreate): """ Sent when a role is updated. diff --git a/tests/gateway/events.py b/tests/gateway/events.py index 7609cd5..96e6fe2 100644 --- a/tests/gateway/events.py +++ b/tests/gateway/events.py @@ -20,7 +20,7 @@ def test_event_creation(benchmark): benchmark(create_resumed_payload) -def test_guild_role_guild_id_attach(): +def test_guild_role_create_guild_id_attach(): event = GatewayEvent.from_dispatch(None, { 't': 'GUILD_ROLE_CREATE', 'd': { @@ -40,3 +40,25 @@ def test_guild_role_guild_id_attach(): assert event.guild_id == 2 assert event.role.guild_id == 2 + + +def test_guild_role_update_guild_id_attach(): + event = GatewayEvent.from_dispatch(None, { + 't': 'GUILD_ROLE_UPDATE', + 'd': { + 'role': { + 'id': 1, + 'name': 'test', + 'color': 1, + 'hoist': True, + 'position': 0, + 'permissions': 0, + 'managed': False, + 'mentionable': False, + }, + 'guild_id': 2, + } + }) + + assert event.guild_id == 2 + assert event.role.guild_id == 2