diff --git a/tests/common/test_utils.py b/tests/common/test_utils.py index 586f8d6546..229ce17085 100644 --- a/tests/common/test_utils.py +++ b/tests/common/test_utils.py @@ -245,7 +245,7 @@ def test_exception_traces() -> None: raise IdentifierTooLongException("postgres", "table", "too_long_table", 8) except Exception as exc: trace = get_exception_trace(exc) - assert trace["exception_type"] == "dlt.common.exceptions.IdentifierTooLongException" + assert trace["exception_type"] == "dlt.common.destination.exceptions.IdentifierTooLongException" assert isinstance(trace["stack_trace"], list) assert trace["exception_attrs"] == { "destination_name": "postgres", @@ -280,7 +280,10 @@ def test_exception_trace_chain() -> None: # outer exception first assert len(traces) == 3 assert traces[0]["exception_type"] == "dlt.common.exceptions.PipelineException" - assert traces[1]["exception_type"] == "dlt.common.exceptions.IdentifierTooLongException" + assert ( + traces[1]["exception_type"] + == "dlt.common.destination.exceptions.IdentifierTooLongException" + ) assert traces[2]["exception_type"] == "dlt.common.exceptions.TerminalValueError"