Fix documentation for nested enums (#351)

2022-03-18 23:36:27 +01:00
parent 1aaf7728cc
commit 5f7e4d58ef
3 changed files with 54 additions and 23 deletions
--- a/tests/inputs/nestedtwice/test_nestedtwice.py
+++ b/tests/inputs/nestedtwice/test_nestedtwice.py
@@ -0,0 +1,25 @@
+import pytest
+
+from tests.output_betterproto.nestedtwice import (
+    Test,
+    TestTop,
+    TestTopMiddle,
+    TestTopMiddleBottom,
+    TestTopMiddleEnumBottom,
+    TestTopMiddleTopMiddleBottom,
+)
+
+
+@pytest.mark.parametrize(
+    ("cls", "expected_comment"),
+    [
+        (Test, "Test doc."),
+        (TestTopMiddleEnumBottom, "EnumBottom doc."),
+        (TestTop, "Top doc."),
+        (TestTopMiddle, "Middle doc."),
+        (TestTopMiddleTopMiddleBottom, "TopMiddleBottom doc."),
+        (TestTopMiddleBottom, "Bottom doc."),
+    ],
+)
+def test_comment(cls, expected_comment):
+    assert cls.__doc__ == expected_comment