Add all kinds of comments (#632)

* Add all kinds of comments

* Format

* Fix get_comment

* Improve test

* Lint

tests/inputs/documentation/documentation.proto

@@ -1,20 +1,44 @@
 syntax = "proto3";
 package documentation;
 
-// Documentation of message
-message Test {
-    // Documentation of field
-    uint32 x = 1;
+// Documentation of message 1
+// other line 1
+
+// Documentation of message 2
+// other line 2
+message Test { // Documentation of message 3
+    // Documentation of field 1
+    // other line 1
+
+    // Documentation of field 2
+    // other line 2
+    uint32 x = 1; // Documentation of field 3
 }
 
-// Documentation of enum
-enum Enum {
-    // Documentation of variant
-    Enum_Variant = 0;
+// Documentation of enum 1
+// other line 1
+
+// Documentation of enum 2
+// other line 2
+enum Enum { // Documentation of enum 3
+    // Documentation of variant 1
+    // other line 1
+
+    // Documentation of variant 2
+    // other line 2
+    Enum_Variant = 0; // Documentation of variant 3
 }
 
-// Documentation of service
-service Service {
-    // Documentation of method
-    rpc get(Test) returns (Test);
+// Documentation of service 1
+// other line 1
+
+// Documentation of service 2
+// other line 2
+service Service { // Documentation of service 3
+    // Documentation of method 1
+    // other line 1
+
+    // Documentation of method 2
+    // other line 2
+    rpc get(Test) returns (Test); // Documentation of method 3
 }

tests/test_documentation.py

@@ -2,7 +2,15 @@ import ast
 import inspect
 
 
-def test_documentation():
+def check(generated_doc: str, type: str) -> None:
+    assert f"Documentation of {type} 1" in generated_doc
+    assert "other line 1" in generated_doc
+    assert f"Documentation of {type} 2" in generated_doc
+    assert "other line 2" in generated_doc
+    assert f"Documentation of {type} 3" in generated_doc
+
+
+def test_documentation() -> None:
     from .output_betterproto.documentation import (
         Enum,
         ServiceBase,
@@ -10,20 +18,20 @@ def test_documentation():
         Test,
     )
 
-    assert Test.__doc__ == "Documentation of message"
+    check(Test.__doc__, "message")
 
     source = inspect.getsource(Test)
     tree = ast.parse(source)
-    assert tree.body[0].body[2].value.value == "Documentation of field"
+    check(tree.body[0].body[2].value.value, "field")
 
-    assert Enum.__doc__ == "Documentation of enum"
+    check(Enum.__doc__, "enum")
 
     source = inspect.getsource(Enum)
     tree = ast.parse(source)
-    assert tree.body[0].body[2].value.value == "Documentation of variant"
+    check(tree.body[0].body[2].value.value, "variant")
 
-    assert ServiceBase.__doc__ == "Documentation of service"
-    assert ServiceBase.get.__doc__ == "Documentation of method"
+    check(ServiceBase.__doc__, "service")
+    check(ServiceBase.get.__doc__, "method")
 
-    assert ServiceStub.__doc__ == "Documentation of service"
-    assert ServiceStub.get.__doc__ == "Documentation of method"
+    check(ServiceStub.__doc__, "service")
+    check(ServiceStub.get.__doc__, "method")