Fixup message link (#27)

* fixup: Fetch guild even when guild_id is nil for Message#initialize
shardlab · Feb 1, 2021 · b9c65bb · b9c65bb
1 parent 23467b4
commit b9c65bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/discordrb/data/message.rb b/lib/discordrb/data/message.rb
@@ -83,7 +83,7 @@ def initialize(data, bot)
       @referenced_message = Message.new(data['referenced_message'], bot) if data['referenced_message']
       @message_reference = data['message_reference']
 
-      @server = bot.server(data['guild_id'].to_i) if data['guild_id']
+      @server = @channel.server
 
       @author = if data['author']
                   if data['author']['discriminator'] == ZERO_DISCRIM

diff --git a/spec/data/message_spec.rb b/spec/data/message_spec.rb
@@ -103,6 +103,8 @@
       data['channel_id'] = channel_id
 
       message = described_class.new(data, bot)
+      message.instance_variable_set(:@server, nil)
+
       expect(message.link).to eq 'https://discord.com/channels/@me/channel_id/message_id'
     end
   end