1 files changed, 6 insertions, 0 deletions
diff --git a/packages/sol-doc/test/solidity_doc_generator_test.ts b/packages/sol-doc/test/solidity_doc_generator_test.ts
index bae67001b..23ffe7b9e 100644
--- a/packages/sol-doc/test/solidity_doc_generator_test.ts
+++ b/packages/sol-doc/test/solidity_doc_generator_test.ts
@@ -84,6 +84,9 @@ describe('#SolidityDocGenerator', () => {
             it('return type name', () => {
                 expect(methodDoc.returnType.name).to.equal('r');
             });
+            it('return comment', () => {
+                expect(methodDoc.returnComment).to.equal('publicMethod @return');
+            });
         });
         describe('should emit external method documentation for', () => {
             let methodDoc: SolidityMethod;
@@ -111,6 +114,9 @@ describe('#SolidityDocGenerator', () => {
             it('return type name', () => {
                 expect(methodDoc.returnType.name).to.equal('r');
             });
+            it('return comment', () => {
+                expect(methodDoc.returnComment).to.equal('externalMethod @return');
+            });
         });
         it('should not truncate a multi-line devdoc comment', () => {
             // tslint:disable-next-line:no-unnecessary-type-assertion