tests: unit tests testing compatibility of our decorator with dataclasses

diff --git a/manager/tests/utils/test_dataclasses_yaml.py b/manager/tests/utils/test_dataclasses_yaml.py
index 513048455fecbb26dbfcbcbb7fde5229fdd22665..cbad3c203769d1c39665c419aae801ed021e0e1b 100644 (file)
--- a/manager/tests/utils/test_dataclasses_yaml.py
+++ b/manager/tests/utils/test_dataclasses_yaml.py
@@ -142,3 +142,25 @@ def test_chained_classes():
     assert _schema_eq(
         CompoundClass.STRICTYAML_SCHEMA, Map({"c": Map({"field": Int()})})
     )
+
+
+def test_combined_with_dataclass():
+    from dataclasses import dataclass
+
+    @dataclass
+    @dataclasses_strictyaml_schema
+    class TestClass:
+        field: int
+
+    assert _schema_eq(TestClass.STRICTYAML_SCHEMA, Map({"field": Int()}))
+
+
+def test_combined_with_dataclass2():
+    from dataclasses import dataclass
+
+    @dataclasses_strictyaml_schema
+    @dataclass
+    class TestClass:
+        field: int
+
+    assert _schema_eq(TestClass.STRICTYAML_SCHEMA, Map({"field": Int()}))