1 files changed, 41 insertions, 0 deletions
diff --git a/mmi2grpc/_description.py b/mmi2grpc/_description.py
new file mode 100644
index 0000000..0f3219b
--- /dev/null
+++ b/mmi2grpc/_description.py
@@ -0,0 +1,41 @@
+import functools
+import unittest
+import textwrap
+
+COMMENT_WIDTH = 80 - 8  # 80 cols - 8 indentation space
+
+
+def assert_description(f):
+    @functools.wraps(f)
+    def wrapper(*args, **kwargs):
+        description = textwrap.fill(
+            kwargs["description"], COMMENT_WIDTH, replace_whitespace=False)
+        docstring = textwrap.dedent(f.__doc__ or "")
+
+        if docstring.strip() != description.strip():
+            print(f'Expected description of {f.__name__}:')
+            print(description)
+
+            # Generate AssertionError
+            test = unittest.TestCase()
+            test.maxDiff = None
+            test.assertMultiLineEqual(
+                docstring.strip(), description.strip(),
+                f'description does not match with function docstring of {f.__name__}')
+
+        f(*args, **kwargs)
+    return wrapper
+
+
+def format_function(id, description):
+    wrapped = textwrap.fill(description, COMMENT_WIDTH,
+                            replace_whitespace=False)
+    return (
+        f'@assert_description\n'
+        f'def {id}(self, **kwargs):\n'
+        f'    """\n'
+        f'{textwrap.indent(wrapped, "    ")}\n'
+        f'    """\n'
+        f'\n'
+        f'    return "OK"\n'
+    )