mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
from argenta.command.params.flag import FlagsGroup, Flag
|
|
from argenta.router import Router
|
|
from argenta.command import Command
|
|
from argenta.router.exceptions import RepeatedCommandException
|
|
|
|
import unittest
|
|
|
|
|
|
class TestRouter(unittest.TestCase):
|
|
def test_get_router_name(self):
|
|
self.assertEqual(Router(name='test name').get_name(), 'test name')
|
|
|
|
def test_get_router_title(self):
|
|
self.assertEqual(Router(title='test title').get_title(), 'test title')
|
|
|
|
def test_input_correct_command(self):
|
|
router = Router()
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
self.assertEqual(router.input_command_handler(Command(command='test')), 'correct result')
|
|
|
|
def test_input_command_with_invalid_flag(self):
|
|
router = Router()
|
|
router.set_invalid_input_flag_handler(lambda x: x)
|
|
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
input_command = Command(command='test')
|
|
input_command.set_input_flags(FlagsGroup([Flag('host')]))
|
|
|
|
self.assertEqual(router.input_command_handler(input_command), None)
|
|
|
|
def test_input_correct_command_with_one_register_and_ignore_command_register(self):
|
|
router = Router()
|
|
router.set_ignore_command_register(True)
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
self.assertEqual(router.input_command_handler(Command(command='test')), 'correct result')
|
|
|
|
def test_input_correct_command_with_different_register_and_ignore_command_register(self):
|
|
router = Router()
|
|
router.set_ignore_command_register(True)
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
self.assertEqual(router.input_command_handler(Command(command='TeSt')), 'correct result')
|
|
|
|
def test_input_incorrect_command_with_ignore_command_register(self):
|
|
router = Router()
|
|
router.set_ignore_command_register(True)
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
self.assertEqual(router.input_command_handler(Command(command='Test2')), None)
|
|
|
|
def test_register_repeated_commands_with_one_register(self):
|
|
router = Router()
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
with self.assertRaises(RepeatedCommandException):
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
def test_register_commands_with_different_register(self):
|
|
router = Router()
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
try:
|
|
@router.command(Command(command='Test'))
|
|
def test():
|
|
return 'correct result'
|
|
except RepeatedCommandException:
|
|
self.fail('RepeatedCommandException should not have been thrown')
|
|
|
|
def test_register_repeated_commands_with_one_register_and_set_ignore_command_register(self):
|
|
router = Router()
|
|
router.set_ignore_command_register(True)
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
with self.assertRaises(RepeatedCommandException):
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
def test_register_repeated_commands_with_different_register_and_set_ignore_command_register(self):
|
|
router = Router()
|
|
router.set_ignore_command_register(True)
|
|
@router.command(Command(command='test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
with self.assertRaises(RepeatedCommandException):
|
|
@router.command(Command(command='Test'))
|
|
def test():
|
|
return 'correct result'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|