ca605001b3
Основные изменения: - Добавлена иерархия исключений (17 классов) с кодами ошибок и контекстом - Улучшена обработка ошибок: детальные сообщения с подсказками - Добавлено 24 теста для экстремальных случаев (комбинаторика, циклы, async) - Добавлено 23 теста для системы обработки ошибок - Исправлен баг с optional-аргументами в renderer.py - Обновлены импорты в тестах (src.breakshaft → breakshaft) Документация: - ERROR_DESIGN.md — проектирование системы ошибок - COMMUTATIVITY_DESIGN.md — анализ проблемы некоммутативности (10 вариантов решений) Файлы: - src/breakshaft/exceptions.py (новый) — модуль исключений - tests/test_error_handling.py (новый) — тесты ошибок - tests/test_extreme_cases.py (новый) — экстремальные кейсы Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>
145 lines
3.7 KiB
Python
145 lines
3.7 KiB
Python
from dataclasses import dataclass
|
|
|
|
from breakshaft.convertor import ConvRepo
|
|
|
|
|
|
@dataclass
|
|
class A:
|
|
a: int
|
|
|
|
|
|
@dataclass
|
|
class B:
|
|
b: float
|
|
|
|
|
|
type optC = str
|
|
|
|
|
|
def test_default_consumer_args():
|
|
repo = ConvRepo()
|
|
|
|
@repo.mark_injector()
|
|
def b_to_a(b: B) -> A:
|
|
return A(int(b.b))
|
|
|
|
@repo.mark_injector()
|
|
def a_to_b(a: A) -> B:
|
|
return B(float(a.a))
|
|
|
|
@repo.mark_injector()
|
|
def int_to_a(i: int) -> A:
|
|
return A(i)
|
|
|
|
def consumer(dep: A, opt_dep: optC = '42') -> tuple[int, str]:
|
|
return dep.a, opt_dep
|
|
|
|
fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn1(B(42.1))
|
|
assert dep == (42, '42')
|
|
|
|
fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn2(123)
|
|
assert dep == (123, '42')
|
|
|
|
fn3 = repo.get_conversion((int, optC), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn3(123, '1')
|
|
assert dep == (123, '1')
|
|
|
|
|
|
def test_optional_default_none_consumer_args():
|
|
repo = ConvRepo()
|
|
|
|
@repo.mark_injector()
|
|
def b_to_a(b: B | None = None) -> A:
|
|
return A(int(b.b))
|
|
|
|
@repo.mark_injector()
|
|
def a_to_b(a: A) -> B | None:
|
|
return B(float(a.a))
|
|
|
|
@repo.mark_injector()
|
|
def int_to_a(i: int) -> A:
|
|
return A(i)
|
|
|
|
def consumer(dep: A, opt_dep: optC = '42') -> tuple[int, str]:
|
|
return dep.a, opt_dep
|
|
|
|
fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn1(B(42.1))
|
|
assert dep == (42, '42')
|
|
|
|
fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn2(123)
|
|
assert dep == (123, '42')
|
|
|
|
fn3 = repo.get_conversion((int, optC), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn3(123, '1')
|
|
assert dep == (123, '1')
|
|
|
|
|
|
def test_default_inj_args():
|
|
repo = ConvRepo()
|
|
|
|
@repo.mark_injector()
|
|
def b_to_a(b: B) -> A:
|
|
return A(int(b.b))
|
|
|
|
@repo.mark_injector()
|
|
def a_to_b(a: A) -> B:
|
|
return B(float(a.a))
|
|
|
|
@repo.mark_injector()
|
|
def int_to_a(i: int, opt_dep: optC = '42') -> A:
|
|
return A(i + int(opt_dep))
|
|
|
|
def consumer(dep: A) -> int:
|
|
return dep.a
|
|
|
|
fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn1(B(42.1))
|
|
assert dep == 42
|
|
|
|
fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn2(123)
|
|
assert dep == 123 + 42
|
|
|
|
fn3 = repo.get_conversion((int, optC,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn3(123, '0')
|
|
assert dep == 123
|
|
|
|
|
|
def test_default_graph_override():
|
|
repo = ConvRepo()
|
|
|
|
@repo.mark_injector()
|
|
def b_to_a(b: B) -> A:
|
|
return A(int(b.b))
|
|
|
|
@repo.mark_injector()
|
|
def a_to_b(a: A) -> B:
|
|
return B(float(a.a))
|
|
|
|
@repo.mark_injector()
|
|
def int_to_a(i: int, opt_dep: optC = '42') -> A:
|
|
return A(i + int(opt_dep))
|
|
|
|
@repo.mark_injector()
|
|
def inject_opt_dep() -> optC:
|
|
return '12345'
|
|
|
|
def consumer(dep: A) -> int:
|
|
return dep.a
|
|
|
|
fn1 = repo.get_conversion((B,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn1(B(42.1))
|
|
assert dep == 42
|
|
|
|
fn2 = repo.get_conversion((int,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn2(123)
|
|
assert dep == 123 + 12345
|
|
|
|
fn3 = repo.get_conversion((int, optC,), consumer, force_commutative=True, force_async=False, allow_async=False)
|
|
dep = fn3(123, '0')
|
|
assert dep == 123
|