Fixed an ``AttributeError`` caused by improper handling of ``dataclasses`` inference in ``pyreverse``2.4

Close #3256

1 files changed, 14 insertions, 0 deletions

diff --git a/tests/unittest_pyreverse_diadefs.py b/tests/unittest_pyreverse_diadefs.py
index 0659a5b11..307f22a94 100644
--- a/tests/unittest_pyreverse_diadefs.py
+++ b/tests/unittest_pyreverse_diadefs.py
@@ -13,6 +13,9 @@
 unit test for the extensions.diadefslib modules
 """
 
+import sys
+from pathlib import Path
+
 import astroid
 import pytest
 
@@ -173,3 +176,14 @@ def test_known_values2(HANDLER, PROJECT):
         (True, "DoNothing"),
         (True, "Specialization"),
     ]
+
+
+@pytest.mark.skipif(sys.version_info < (3, 8), reason="Requires dataclasses")
+def test_regression_dataclasses_inference(HANDLER):
+    project_path = Path("regrtest_data") / "dataclasses_pyreverse"
+    path = get_project(str(project_path))
+
+    cdg = ClassDiadefGenerator(Linker(path), HANDLER)
+    special = "regrtest_data.dataclasses_pyreverse.InventoryItem"
+    cd = cdg.class_diagram(path, special)
+    assert cd.title == special