mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
kolo
ce7e24b924
The entire public api is covered with documentation in two languages - Russian and English. the library now supports the latest three versions of python - 3.12, 3.13 and 3.14 minor design changes: now, when a Boolean flag is entered, its value is an empty string, not None. tests have been adapted to the supported versions of python, readmi has been redesigned in two languages, German is no longer available.
104 lines
6.0 KiB
Python
104 lines
6.0 KiB
Python
import re
|
|
import unittest
|
|
|
|
from argenta.command import Command
|
|
from argenta.command.flag import Flag, InputFlag
|
|
from argenta.command.flag.flags import Flags, InputFlags
|
|
from argenta.command.flag.models import PossibleValues, ValidationStatus
|
|
from argenta.response.entity import Response
|
|
from argenta.router import Router
|
|
from argenta.router.entity import _structuring_input_flags, _validate_command, _validate_func_args # pyright: ignore[reportPrivateUsage]
|
|
from argenta.router.exceptions import (RepeatedFlagNameException,
|
|
RequiredArgumentNotPassedException,
|
|
TriggerContainSpacesException)
|
|
|
|
|
|
class TestRouter(unittest.TestCase):
|
|
def test_register_command_with_spaces_in_trigger(self):
|
|
with self.assertRaises(TriggerContainSpacesException):
|
|
_validate_command(Command(trigger='command with spaces'))
|
|
|
|
def test_register_command_with_repeated_flags(self):
|
|
with self.assertRaises(RepeatedFlagNameException):
|
|
_validate_command(Command(trigger='command', flags=Flags([Flag('test'), Flag('test')])))
|
|
|
|
def test_structuring_input_flags1(self):
|
|
cmd = Command('cmd')
|
|
input_flags = InputFlags([InputFlag('ssh', input_value=None, status=None)])
|
|
self.assertEqual(_structuring_input_flags(cmd, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value=None, status=ValidationStatus.UNDEFINED)]))
|
|
|
|
def test_structuring_input_flags2(self):
|
|
cmd = Command('cmd')
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='some', status=None)])
|
|
self.assertEqual(_structuring_input_flags(cmd, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='some', status=ValidationStatus.UNDEFINED)]))
|
|
|
|
def test_structuring_input_flags3(self):
|
|
cmd = Command('cmd', flags=Flag('port'))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='some2', status=None)])
|
|
self.assertEqual(_structuring_input_flags(cmd, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='some2', status=ValidationStatus.UNDEFINED)]))
|
|
|
|
def test_structuring_input_flags4(self):
|
|
command = Command('cmd', flags=Flag('ssh', possible_values=PossibleValues.NEITHER))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='some3', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='some3', status=ValidationStatus.INVALID)]))
|
|
|
|
def test_structuring_input_flags5(self):
|
|
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'some[1-5]$')))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='some40', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='some40', status=ValidationStatus.INVALID)]))
|
|
|
|
def test_structuring_input_flags6(self):
|
|
command = Command('cmd', flags=Flag('ssh', possible_values=['example']))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='example2', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='example2', status=ValidationStatus.INVALID)]))
|
|
|
|
def test_structuring_input_flags7(self):
|
|
command = Command('cmd', flags=Flag('port'))
|
|
input_flags = InputFlags([InputFlag('port', input_value='some2', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('port', input_value='some2', status=ValidationStatus.VALID)]))
|
|
|
|
def test_structuring_input_flags8(self):
|
|
command = Command('cmd', flags=Flag('port', possible_values=['some2', 'some3']))
|
|
input_flags = InputFlags([InputFlag('port', input_value='some2', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('port', input_value='some2', status=ValidationStatus.VALID)]))
|
|
|
|
def test_structuring_input_flags9(self):
|
|
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'more[1-5]$')))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value='more5', status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value='more5', status=ValidationStatus.VALID)]))
|
|
|
|
def test_structuring_input_flags10(self):
|
|
command = Command('cmd', flags=Flag('ssh', possible_values=PossibleValues.NEITHER))
|
|
input_flags = InputFlags([InputFlag('ssh', input_value=None, status=None)])
|
|
self.assertEqual(_structuring_input_flags(command, input_flags).input_flags, InputFlags([InputFlag('ssh', input_value=None, status=ValidationStatus.VALID)]))
|
|
|
|
def test_validate_incorrect_func_args1(self):
|
|
def handler():
|
|
pass
|
|
with self.assertRaises(RequiredArgumentNotPassedException):
|
|
_validate_func_args(handler) # pyright: ignore[reportArgumentType]
|
|
|
|
def test_get_router_aliases(self):
|
|
router = Router()
|
|
@router.command(Command('some', aliases=['test', 'case']))
|
|
def handler(response: Response) -> None: # pyright: ignore[reportUnusedFunction]
|
|
pass
|
|
self.assertListEqual(router.aliases, ['test', 'case'])
|
|
|
|
def test_get_router_aliases2(self):
|
|
router = Router()
|
|
@router.command(Command('some', aliases=['test', 'case']))
|
|
def handler(response: Response): # pyright: ignore[reportUnusedFunction]
|
|
pass
|
|
@router.command(Command('ext', aliases=['more', 'foo']))
|
|
def handler2(response: Response): # pyright: ignore[reportUnusedFunction]
|
|
pass
|
|
self.assertListEqual(router.aliases, ['test', 'case', 'more', 'foo'])
|
|
|
|
def test_get_router_aliases3(self):
|
|
router = Router()
|
|
@router.command(Command('some'))
|
|
def handler(response: Response): # pyright: ignore[reportUnusedFunction]
|
|
pass
|
|
self.assertListEqual(router.aliases, [])
|