mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
227 lines
6.6 KiB
Python
227 lines
6.6 KiB
Python
import re
|
|
from sys import flags
|
|
|
|
from argenta.command.flag import Flag, InputFlag, PossibleValues
|
|
from argenta.command.flag.flags import Flags, InputFlags
|
|
import pytest
|
|
|
|
|
|
def test_get_string_entity():
|
|
assert Flag(name='test').string_entity == '--test'
|
|
|
|
|
|
def test_get_string_entity2():
|
|
assert Flag(name='test', prefix='---').string_entity == '---test'
|
|
|
|
|
|
def test_get_flag_name():
|
|
assert Flag(name='test').name == 'test'
|
|
|
|
|
|
def test_get_flag_prefix():
|
|
assert Flag(name='test').prefix == '--'
|
|
|
|
|
|
def test_get_flag_prefix2():
|
|
assert Flag(name='test', prefix='--').prefix == '--'
|
|
|
|
|
|
def test_get_flag_value_without_set():
|
|
assert InputFlag(name='test', input_value='', status=None).input_value == ''
|
|
|
|
|
|
def test_get_flag_value_with_set():
|
|
flag = InputFlag(name='test', input_value='example', status=None)
|
|
assert flag.input_value == 'example'
|
|
|
|
|
|
def test_validate_incorrect_flag_value_with_list_of_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=['1', '2', '3'])
|
|
assert flag.validate_input_flag_value('bad value') is False
|
|
|
|
|
|
def test_validate_correct_flag_value_with_list_of_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=['1', '2', '3'])
|
|
assert flag.validate_input_flag_value('1') is True
|
|
|
|
|
|
def test_validate_incorrect_flag_value_with_pattern_of_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=re.compile(r'192.168.\d+.\d+'))
|
|
assert flag.validate_input_flag_value('152.123.9.8') is False
|
|
|
|
|
|
def test_validate_correct_flag_value_with_pattern_of_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=re.compile(r'192.168.\d+.\d+'))
|
|
assert flag.validate_input_flag_value('192.168.9.8') is True
|
|
|
|
|
|
def test_validate_correct_empty_flag_value_without_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=PossibleValues.NEITHER)
|
|
assert flag.validate_input_flag_value('') is True
|
|
|
|
|
|
def test_validate_correct_empty_flag_value_with_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=PossibleValues.NEITHER)
|
|
assert flag.validate_input_flag_value('') is True
|
|
|
|
|
|
def test_validate_incorrect_random_flag_value_without_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=PossibleValues.NEITHER)
|
|
assert flag.validate_input_flag_value('random value') is False
|
|
|
|
|
|
def test_validate_correct_random_flag_value_with_possible_flag_values():
|
|
flag = Flag(name='test', possible_values=PossibleValues.ALL)
|
|
assert flag.validate_input_flag_value('random value') is True
|
|
|
|
|
|
def test_get_input_flag1():
|
|
flag = InputFlag(name='test', input_value='', status=None)
|
|
input_flags = InputFlags([flag])
|
|
assert input_flags.get_flag_by_name('test') == flag
|
|
|
|
|
|
def test_get_input_flag2():
|
|
flag = InputFlag(name='test', input_value='', status=None)
|
|
flag2 = InputFlag(name='some', input_value='', status=None)
|
|
input_flags = InputFlags([flag, flag2])
|
|
assert input_flags.get_flag_by_name('some') == flag2
|
|
|
|
|
|
def test_get_undefined_input_flag():
|
|
flag = InputFlag(name='test', input_value='', status=None)
|
|
flag2 = InputFlag(name='some', input_value='', status=None)
|
|
input_flags = InputFlags([flag, flag2])
|
|
assert input_flags.get_flag_by_name('case') is None
|
|
|
|
|
|
def test_get_flags():
|
|
flags = Flags()
|
|
list_of_flags = [
|
|
Flag('test1'),
|
|
Flag('test2'),
|
|
Flag('test3'),
|
|
]
|
|
flags.add_flags(list_of_flags)
|
|
assert flags.flags == list_of_flags
|
|
|
|
|
|
def test_add_flag():
|
|
flags = Flags()
|
|
flags.add_flag(Flag('test'))
|
|
assert len(flags.flags) == 1
|
|
|
|
|
|
def test_add_flags():
|
|
flags = Flags()
|
|
flags.add_flags([Flag('test'), Flag('test2')])
|
|
assert len(flags.flags) == 2
|
|
|
|
def test_eq_flags():
|
|
flags = Flags([Flag('some')])
|
|
flags2 = Flags([Flag('some')])
|
|
assert flags == flags2
|
|
|
|
def test_contains_flags():
|
|
flags = Flags([Flag('some')])
|
|
flag = Flag('some')
|
|
assert flag in flags
|
|
|
|
def test_eq_flags2():
|
|
flags = Flags([Flag('some')])
|
|
flags2 = Flags([Flag('other')])
|
|
assert flags != flags2
|
|
|
|
def test_eq_flags3():
|
|
flags = Flags([Flag('some')])
|
|
flags2 = Flags([Flag('some'), Flag('other')])
|
|
assert flags != flags2
|
|
|
|
def test_eq_flags4():
|
|
flags = Flags([Flag('some')])
|
|
not_flags = object()
|
|
assert flags != not_flags
|
|
|
|
def test_contains_flags2():
|
|
flags = Flags([Flag('some')])
|
|
flag = Flag('nonexists')
|
|
assert flag not in flags
|
|
|
|
def test_contains_flags3():
|
|
flags = Flags([Flag('some')])
|
|
not_flag = object
|
|
with pytest.raises(TypeError):
|
|
not_flag in flags # pyright: ignore[reportUnusedExpression]
|
|
|
|
def test_get_flag_by_name():
|
|
flags = Flags([Flag('some')])
|
|
assert flags.get_flag_by_name('some') == Flag('some')
|
|
|
|
def test_eq_input_flags3():
|
|
flags = InputFlags([InputFlag('some', input_value='')])
|
|
flags2 = InputFlags([
|
|
InputFlag('some', input_value=''),
|
|
InputFlag('some2', input_value='')
|
|
])
|
|
assert flags != flags2
|
|
|
|
def test_eq_input_flags4():
|
|
flags = InputFlags([InputFlag('some', input_value='')])
|
|
not_flags = object()
|
|
assert flags != not_flags
|
|
|
|
def test_contains_input_flags2():
|
|
flags = InputFlags([InputFlag('some', input_value='')])
|
|
flag = InputFlag('nonexists', input_value='')
|
|
assert flag not in flags
|
|
|
|
def test_contains_input_flags3():
|
|
flags = InputFlags([InputFlag('some', input_value='')])
|
|
not_flag = object
|
|
with pytest.raises(TypeError):
|
|
not_flag in flags # pyright: ignore[reportUnusedExpression]
|
|
|
|
def test_len_flags():
|
|
flags = Flags([Flag('one'), Flag('two')])
|
|
assert len(flags) == 2
|
|
|
|
def test_bool_flags():
|
|
flags = Flags([Flag('one'), Flag('two')])
|
|
assert bool(flags)
|
|
|
|
def test_bool_flags2():
|
|
flags = Flags([])
|
|
assert not bool(flags)
|
|
|
|
def test_getitem_flags():
|
|
flags = Flags([Flag('one'), Flag('two')])
|
|
assert flags[1] == Flag('two')
|
|
|
|
def test_str_flag():
|
|
flag = Flag('two')
|
|
assert str(flag) == '--two'
|
|
|
|
def test_repr_flag():
|
|
flag = Flag('two')
|
|
assert repr(flag) == 'Flag<name=two, prefix=-->'
|
|
|
|
def test_eq_flag():
|
|
flag = Flag('two')
|
|
not_flag = object()
|
|
with pytest.raises(NotImplementedError):
|
|
flag == not_flag # pyright: ignore[reportUnusedExpression]
|
|
|
|
def test_str_input_flag():
|
|
flag = InputFlag('two', input_value='value')
|
|
assert str(flag) == '--two value'
|
|
|
|
def test_repr_input_flag():
|
|
flag = InputFlag('two', input_value='some_value')
|
|
assert repr(flag) == 'InputFlag<name=two, prefix=--, value=some_value, status=None>'
|
|
|
|
def test_eq_input_flag():
|
|
flag = InputFlag('two', input_value='')
|
|
not_flag = object()
|
|
with pytest.raises(NotImplementedError):
|
|
flag == not_flag # pyright: ignore[reportUnusedExpression]
|
|
|