Пример #1
0
    def test_simple_functions(self):
        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        def_commands = [
            "test_func Hello there, {user}",
            "-ul=reg test_func2 Hello, {0}",
        ]

        for line in def_commands:
            cm.add_simple_command(line.split(" "))

        retval = cm.run_command("username", "mod", "test_func", [],
                                threaded=False)
        self.assertEquals(retval, "Hello there, username")

        retval = cm.run_command(
            "username", "reg", "test_func2", ["target"], threaded=False
        )
        self.assertEquals(retval, "Hello, target")

        self.assertRaises(
            bot.commandmanager.CommandPermissionError,
            cm.run_command,
            "username",
            "user",
            "test_func2"
        )
Пример #2
0
    def test_functions(self):
        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        def_commands = [
            "-ul=user -a=value test_func return tonumber(value) + 1",
            "-ul=reg --args=value test_func2 return __chat__test_func(value)"
            " + 10",
            "--args=... test_args local s = 0; for k, v in ipairs("
            "{...}) do s = s + tonumber(v); end; return s"
        ]

        for line in def_commands:
            cm.add_command(line.split(" "))

        retval = cm.run_command("username", "mod", "test_func2", ["10"],
                                threaded=False)
        self.assertEquals(retval, 21)

        retval = cm.run_command(
            "username", "mod", "test_args", ["1", "2", "3"], threaded=False
        )
        self.assertEquals(retval, 6)
Пример #3
0
    def test_cooldown(self):

        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        def_commands = [
            "-c=5 cd_test Cooldown test",
        ]

        for line in def_commands:
            cm.add_simple_command(line.split(" "))

        def run_cmd(timestamp):
            def _run():
                return cm.run_command(
                    "username", "mod", "cd_test", [], timestamp=timestamp,
                    threaded=False
                )

            return _run

        retval = run_cmd(1)()
        self.assertEquals(retval, "Cooldown test")

        self.assertRaises(
            bot.commandmanager.CommandCooldownError, run_cmd(2)
        )

        retval = run_cmd(6)()
        self.assertEquals(retval, "Cooldown test")
Пример #4
0
    def test_unicode(self):
        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        line = u"test ヽ༼ຈل͜ຈ༽ノ AMENO ヽ༼ຈل͜ຈ༽ノ"
        cm.add_simple_command(line.split(" "))

        cm.run_command("username", "mod", "test", threaded=False)
        chat.message.assert_called_with(u"ヽ༼ຈل͜ຈ༽ノ AMENO ヽ༼ຈل͜ຈ༽ノ")
Пример #5
0
    def test_want_user(self):
        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        line = "-w test return user"
        cm.add_command(line.split(" "))

        retval = cm.run_command("fakeuser", "mod", "test", threaded=False)
        self.assertEquals(retval, "fakeuser")
Пример #6
0
    def test_quoted(self):
        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(), chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        def_commands = [
            "-q -a=name,job test_quoted return name .. ': ' .. job",
            "-a=name,job test_not_quoted return name .. ': ' .. job",
        ]

        for line in def_commands:
            cm.add_command(line.split(" "))

        args = '"John Doe" "Car Salesman"'.split(" ")
        retval = cm.run_command("fakeuser", "mod", "test_quoted", args,
                                threaded=False)
        self.assertEquals(retval, "John Doe: Car Salesman")

        retval = cm.run_command("fakeuser", "mod", "test_not_quoted", args,
                                threaded=False)
        self.assertEquals(retval, '"John: Doe"')
Пример #7
0
    def test_permissions(self):

        chat = Chat(None, None)
        chat.message = Mock()
        cm = bot.commandmanager.CommandManager("#tmp", FakeBot(),
                                               chat=chat)
        cm.load_lua(self.UTILS)

        # Some test command definitions
        def_commands = [
            "-ul=owner owner_func return 0",
            "-ul=mod mod_func return 1",
            "-ul=reg reg_func return 2",
            "-ul=user user_func return 3"
        ]

        for line in def_commands:
            cm.add_command(line.split(" "))

        # owner_func

        self.assertRaises(
            bot.commandmanager.CommandPermissionError,
            cm.run_command,
            "username",
            "mod",
            "owner_func"
        )

        cm.run_command("username", "owner", "owner_func", threaded=False)
        chat.message.assert_called_with(0)

        # mod_func

        self.assertRaises(
            bot.commandmanager.CommandPermissionError,
            cm.run_command,
            "username",
            "reg",
            "mod_func"
        )

        cm.run_command("username", "mod", "mod_func", threaded=False)
        chat.message.assert_called_with(1)

        # reg_func

        self.assertRaises(
            bot.commandmanager.CommandPermissionError,
            cm.run_command,
            "username",
            "user",
            "reg_func"
        )

        cm.run_command("username", "reg", "reg_func", threaded=False)
        chat.message.assert_called_with(2)

        cm.run_command("username", "owner", "reg_func", threaded=False)
        chat.message.assert_called_with(2)

        # user_func

        cm.run_command("username", "user", "user_func", threaded=False)
        chat.message.assert_called_with(3)