mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
from argenta.command import Command
|
|
from argenta.command.params.flag import Flag, FlagsGroup
|
|
from argenta.command.exceptions import (UnprocessedInputFlagException,
|
|
RepeatedInputFlagsException,
|
|
EmptyInputCommandException)
|
|
|
|
import unittest
|
|
|
|
|
|
class TestCommand(unittest.TestCase):
|
|
def test_parse_correct_raw_command(self):
|
|
self.assertEqual(Command.parse_input_command('ssh --host 192.168.0.3').get_string_entity(), 'ssh')
|
|
|
|
def test_parse_raw_command_with_flag_name_without_value(self):
|
|
with self.assertRaises(UnprocessedInputFlagException):
|
|
Command.parse_input_command('ssh --host')
|
|
|
|
def test_parse_raw_command_without_flag_name_with_value(self):
|
|
with self.assertRaises(UnprocessedInputFlagException):
|
|
Command.parse_input_command('ssh 192.168.0.3')
|
|
|
|
def test_parse_raw_command_with_repeated_flag_name(self):
|
|
with self.assertRaises(RepeatedInputFlagsException):
|
|
Command.parse_input_command('ssh --host 192.168.0.3 --host 172.198.0.43')
|
|
|
|
def test_parse_empty_raw_command(self):
|
|
with self.assertRaises(EmptyInputCommandException):
|
|
Command.parse_input_command('')
|
|
|
|
def test_get_command_description(self):
|
|
self.assertEqual(Command(command='test', description='test description').get_description(), 'test description')
|
|
|