Renamed unclear message_id
This commit is contained in:
@@ -1,2 +1,2 @@
|
|||||||
from ._bot import Bot
|
from ._bot import Bot
|
||||||
from ._discord_utils import delete_message, message_id
|
from ._discord_utils import delete_message, channel_id, sender_id
|
||||||
|
|||||||
@@ -11,10 +11,16 @@ async def delete_message(message: discord.Message) -> bool:
|
|||||||
pass
|
pass
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
def channel_id(message: discord.Message) -> str:
|
||||||
def message_id(message: discord.Message) -> str:
|
|
||||||
is_direct = message.channel.type == discord.ChannelType.private
|
is_direct = message.channel.type == discord.ChannelType.private
|
||||||
if not is_direct:
|
if not is_direct:
|
||||||
return f'{message.guild.id}/{message.channel.id}/{message.author.id}'
|
return f'{message.guild.id}/{message.channel.id}'
|
||||||
|
else:
|
||||||
|
return message.author.id
|
||||||
|
|
||||||
|
def sender_id(message: discord.Message) -> str:
|
||||||
|
is_direct = message.channel.type == discord.ChannelType.private
|
||||||
|
if not is_direct:
|
||||||
|
return f'{channel_id(message)}/{message.author.id}'
|
||||||
else:
|
else:
|
||||||
return message.author.id
|
return message.author.id
|
||||||
|
|||||||
@@ -136,6 +136,18 @@ class TestRegisterCommand(TestCase):
|
|||||||
self.assertEqual("^test$", cmd.regex)
|
self.assertEqual("^test$", cmd.regex)
|
||||||
|
|
||||||
|
|
||||||
|
class TestRegisterFallback(TestCase):
|
||||||
|
@skip
|
||||||
|
def test_todo(self):
|
||||||
|
self.fail("not implemented")
|
||||||
|
|
||||||
|
|
||||||
|
class TestRegisterWatcher(TestCase):
|
||||||
|
@skip
|
||||||
|
def test_todo(self):
|
||||||
|
self.fail("not implemented")
|
||||||
|
|
||||||
|
|
||||||
class TestOnMessage(AsyncTestCase):
|
class TestOnMessage(AsyncTestCase):
|
||||||
@skip
|
@skip
|
||||||
def test_todo(self):
|
def test_todo(self):
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ from unittest.mock import Mock, AsyncMock
|
|||||||
from tests.utils import AsyncTestCase
|
from tests.utils import AsyncTestCase
|
||||||
|
|
||||||
import discord
|
import discord
|
||||||
from miniscord._discord_utils import delete_message, message_id
|
from miniscord._discord_utils import delete_message, channel_id, sender_id
|
||||||
|
|
||||||
|
|
||||||
class TestDeleteMessage(AsyncTestCase):
|
class TestDeleteMessage(AsyncTestCase):
|
||||||
@@ -25,12 +25,12 @@ class TestDeleteMessage(AsyncTestCase):
|
|||||||
message.delete.assert_awaited_once()
|
message.delete.assert_awaited_once()
|
||||||
|
|
||||||
|
|
||||||
class TestMessageId(TestCase):
|
class TestChannelId(TestCase):
|
||||||
def test_direct(self):
|
def test_direct(self):
|
||||||
message = Mock()
|
message = Mock()
|
||||||
message.channel.type = discord.ChannelType.private
|
message.channel.type = discord.ChannelType.private
|
||||||
message.author.id = "TEST"
|
message.author.id = "TEST"
|
||||||
self.assertEqual("TEST", message_id(message))
|
self.assertEqual("TEST", channel_id(message))
|
||||||
|
|
||||||
def test_not_direct(self):
|
def test_not_direct(self):
|
||||||
message = Mock()
|
message = Mock()
|
||||||
@@ -38,4 +38,20 @@ class TestMessageId(TestCase):
|
|||||||
message.guild.id = "TEST1"
|
message.guild.id = "TEST1"
|
||||||
message.channel.id = "TEST2"
|
message.channel.id = "TEST2"
|
||||||
message.author.id = "TEST3"
|
message.author.id = "TEST3"
|
||||||
self.assertEqual("TEST1/TEST2/TEST3", message_id(message))
|
self.assertEqual("TEST1/TEST2", channel_id(message))
|
||||||
|
|
||||||
|
|
||||||
|
class TestSenderId(TestCase):
|
||||||
|
def test_direct(self):
|
||||||
|
message = Mock()
|
||||||
|
message.channel.type = discord.ChannelType.private
|
||||||
|
message.author.id = "TEST"
|
||||||
|
self.assertEqual("TEST", sender_id(message))
|
||||||
|
|
||||||
|
def test_not_direct(self):
|
||||||
|
message = Mock()
|
||||||
|
message.channel.type = discord.ChannelType.text
|
||||||
|
message.guild.id = "TEST1"
|
||||||
|
message.channel.id = "TEST2"
|
||||||
|
message.author.id = "TEST3"
|
||||||
|
self.assertEqual("TEST1/TEST2/TEST3", sender_id(message))
|
||||||
|
|||||||
Reference in New Issue
Block a user