Esempio n. 1
0
 def setUp(self):
     """
     Simply setup our Guild obj before usage
     """
     self.guild = Guild(None, 15, {})
     self.guild.users = User(20, 15, {})
     self.guild.users = User(21, 15, {})
Esempio n. 2
0
    def propagate(self, message: discord.Message):
        """
        This method handles a message object and then adds it to
        the relevant user

        Parameters
        ==========
        message : discord.Message
            The message that needs to be propagated out
        """
        if not isinstance(message, discord.Message):
            raise ValueError("Expected message of type: discord.Message")

        if isinstance(self._channel, int):
            self.channel = message.channel

        user = User(
            self._bot,
            message.author.id,
            message.guild.id,
            self.options,
            logger=self.logger,
        )
        for userObj in self.users:
            if user == userObj:
                return userObj.propagate(message)

        self.users = user
        self.logger.info(f"Created User: {user.id}")

        user.propagate(message)
Esempio n. 3
0
 def setUp(self):
     """
     Simply setup our User obj before usage
     """
     self.user = User(
         None, 0, 3, Static.DEFAULTS, logger=logging.getLogger(__name__)
     )
     self.user.messages = Message(0, "Hello world", 0, 2, 3)
     self.user.messages = Message(1, "Foo Bar", 0, 2, 3)
Esempio n. 4
0
 def setUp(self):
     """
     Simply setup our User obj before usage
     """
     self.user = User(0, 3, {})
     self.user.messages = Message(0, "Hello world", 0, 2, 3)
     self.user.messages = Message(1, "Foo Bar", 0, 2, 3)
Esempio n. 5
0
 def test_messageRaisesMismatch(self):
     with self.assertRaises(ObjectMismatch):
         self.guild.users = User(None,
                                 22,
                                 16,
                                 Static.DEFAULTS,
                                 logger=logging.getLogger(__name__))
Esempio n. 6
0
 def test_userRaisesDuplicate(self):
     with self.assertRaises(DuplicateObject):
         self.guild.users = User(None,
                                 21,
                                 15,
                                 Static.DEFAULTS,
                                 logger=logging.getLogger(__name__))
Esempio n. 7
0
 def test_userAssignment(self):
     self.assertEqual(len(self.guild.users), 2)
     self.guild.users = User(None,
                             22,
                             15,
                             Static.DEFAULTS,
                             logger=logging.getLogger(__name__))
     self.assertEqual(len(self.guild.users), 3)
Esempio n. 8
0
 def setUp(self):
     """
     Simply setup our Guild obj before usage
     """
     self.guild = Guild(None,
                        15,
                        Static.DEFAULTS,
                        logger=logging.getLogger(__name__))
     self.guild.users = User(None,
                             20,
                             15,
                             Static.DEFAULTS,
                             logger=logging.getLogger(__name__))
     self.guild.users = User(None,
                             21,
                             15,
                             Static.DEFAULTS,
                             logger=logging.getLogger(__name__))
Esempio n. 9
0
 def setUp(self):
     """
     Simply setup our Guild obj before usage
     """
     self.ash = AntiSpamHandler(commands.Bot(command_prefix="!"))
     self.ash.guilds = Guild(None,
                             12,
                             Static.DEFAULTS,
                             logger=logging.getLogger(__name__))
     self.ash.guilds = Guild(None,
                             15,
                             Static.DEFAULTS,
                             logger=logging.getLogger(__name__))
     self.ash.guilds[0] = User(None,
                               20,
                               15,
                               Static.DEFAULTS,
                               logger=logging.getLogger(__name__))
Esempio n. 10
0
 def test_messageRaisesMismatch(self):
     with self.assertRaises(ObjectMismatch):
         self.guild.users = User(22, 16, {})
Esempio n. 11
0
 def test_userRaisesDuplicate(self):
     with self.assertRaises(DuplicateObject):
         self.guild.users = User(21, 15, {})
Esempio n. 12
0
 def test_userAssignment(self):
     self.assertEqual(len(self.guild.users), 2)
     self.guild.users = User(22, 15, {})
     self.assertEqual(len(self.guild.users), 3)
Esempio n. 13
0
class TestUser(unittest.TestCase):
    """
    Used to test the user object
    """

    def setUp(self):
        """
        Simply setup our User obj before usage
        """
        self.user = User(
            None, 0, 3, Static.DEFAULTS, logger=logging.getLogger(__name__)
        )
        self.user.messages = Message(0, "Hello world", 0, 2, 3)
        self.user.messages = Message(1, "Foo Bar", 0, 2, 3)

    def test_botAssignment(self):
        # Given we don't want to populate an entire bot,
        # lets just check its assigned correctly
        self.assertIsNone(self.user.bot)

    def test_intAssignment(self):
        self.assertIsInstance(self.user.id, int)
        self.assertIsInstance(self.user.guildId, int)

    def test_listAssignment(self):
        self.assertIsInstance(self.user.messages, list)

    def test_dictAssignment(self):
        self.assertIsInstance(self.user.options, dict)

    def test_valueAssignment(self):
        self.assertEqual(self.user.id, 0)
        self.assertEqual(self.user.guildId, 3)
        self.assertEqual(self.user.options, Static.DEFAULTS)

        self.user.id = 10
        self.user.guildId = 10

        self.assertEqual(self.user.id, 10)
        self.assertEqual(self.user.guildId, 10)

    def test_messageAssignment(self):
        self.assertEqual(len(self.user.messages), 2)
        self.user.messages = Message(3, "Test", 0, 2, 3)
        self.assertEqual(len(self.user.messages), 3)

    def test_messageRaises(self):
        with self.assertRaises(ValueError):
            self.user.messages = 1

    def test_messageRaisesDuplicate(self):
        with self.assertRaises(DuplicateObject):
            self.user.messages = Message(1, "Testing", 0, 2, 3)

    def test_messageRaisesMismatch(self):
        with self.assertRaises(ObjectMismatch):
            self.user.messages = Message(20, "Testing", 20, 20, 20)

    def test_repr(self):
        self.assertEqual(
            repr(self.user),
            (
                f"'{self.user.__class__.__name__} object. User id: {self.user.id}, Guild id: {self.user.guildId}, "
                f"Len Stored Messages {len(self.user.messages)}'"
            ),
        )

    def test_str(self):
        self.assertEqual(
            str(self.user), f"{self.user.__class__.__name__} object for {self.user.id}."
        )

    def test_eqEqual(self):
        obj = User(None, 0, 3, Static.DEFAULTS, logger=logging.getLogger(__name__))
        self.assertTrue(self.user == obj)

    def test_eqNotEqual(self):
        obj = User(None, 2, 2, Static.DEFAULTS, logger=logging.getLogger(__name__))
        self.assertFalse(self.user == obj)

    def test_duplicateCounter(self):
        self.assertNotEqual(
            self.user.duplicateCounter, self.user.GetCorrectDuplicateCount()
        )

        self.assertEqual(
            self.user.duplicateCounter - 1, self.user.GetCorrectDuplicateCount()
        )
Esempio n. 14
0
 def test_eqNotEqual(self):
     obj = User(None, 2, 2, Static.DEFAULTS, logger=logging.getLogger(__name__))
     self.assertFalse(self.user == obj)
Esempio n. 15
0
 def test_eqEqual(self):
     obj = User(None, 0, 3, Static.DEFAULTS, logger=logging.getLogger(__name__))
     self.assertTrue(self.user == obj)