fixes test

dlt-hub · Mar 24, 2024 · 177b902 · 177b902
1 parent 4c5b0e3
commit 177b902
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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"