mirror of https://github.com/stijndcl/didier
Additional snipe tests
parent
ab6819d963
commit
02bd10adcb
|
@ -0,0 +1,27 @@
|
|||
from data.snipe import should_snipe
|
||||
import unittest
|
||||
from unittest.mock import Mock
|
||||
|
||||
|
||||
class TestSnipe(unittest.TestCase):
|
||||
def test_should_snipe(self):
|
||||
mock_message = Mock()
|
||||
mock_guild = Mock()
|
||||
mock_author = Mock()
|
||||
|
||||
# Guild is None
|
||||
mock_message.guild = None
|
||||
self.assertFalse(should_snipe(mock_message))
|
||||
mock_message.guild = mock_guild
|
||||
|
||||
# Author is a bot
|
||||
mock_message.author = mock_author
|
||||
mock_author.bot = True
|
||||
self.assertFalse(should_snipe(mock_message))
|
||||
mock_author.bot = False
|
||||
|
||||
mock_message.content = "Some string that contains A123B-CE68S-Z6B34 a Steam code"
|
||||
self.assertFalse(should_snipe(mock_message))
|
||||
|
||||
mock_message.content = "Some string that does NOT contain a Steam code"
|
||||
self.assertTrue(should_snipe(mock_message))
|
Loading…
Reference in New Issue