Fix unexpected return tag on non-functions (#214)

1 files changed, 9 insertions, 1 deletions

diff --git a/src/test/CppTests/Config.test.ts b/src/test/CppTests/Config.test.ts
index f39b7d3..ec52c61 100644
--- a/src/test/CppTests/Config.test.ts
+++ b/src/test/CppTests/Config.test.ts
@@ -65,7 +65,7 @@ suite("C++ - Configuration Tests", () => {
             + " * @return bool \n */", result);
     });
 
-    test("Disable including return type.", () => {
+    test("Disable including return type on function", () => {
         testSetup.cfg = new Config();
         testSetup.cfg.Generic.includeTypeAtReturn = false;
 
@@ -74,6 +74,14 @@ suite("C++ - Configuration Tests", () => {
             + " * @return  \n */", result);
     });
 
+    test("Disable including return type on non-function", () => {
+        testSetup.cfg = new Config();
+        testSetup.cfg.Generic.includeTypeAtReturn = false;
+
+        const result = testSetup.SetLine("bool b;").GetResult();
+        assert.strictEqual("/**\n * @brief \n * \n */", result);
+    });
+
     test("Newlines after params and tparams but not after brief", () => {
         testSetup.cfg = new Config();
         testSetup.cfg.Generic.order = ["brief", "tparam", "empty", "param", "empty", "return"];