1 files changed, 11 insertions, 0 deletions
diff --git a/src/test/CppTests/Config.test.ts b/src/test/CppTests/Config.test.ts
index d33b5c1..8a8e939 100644
--- a/src/test/CppTests/Config.test.ts
+++ b/src/test/CppTests/Config.test.ts
@@ -80,4 +80,15 @@ suite("C++ - Configuration Tests", () => {
             + "\n           * @param a \n           * "
             + "@return true \n           * @return false \n           */", result);
     });
+
+    test("Lines to get test", () => {
+        testSetup.cfg = new Config();
+        testSetup.cfg.linesToGet = 1;
+        const negativeResult = testSetup.SetLines(["template<typename T> bool \n", "foo(T a);"]).GetResult();
+        assert.equal("/**\n * @brief \n * \n */", negativeResult);
+        testSetup.cfg.linesToGet = 2;
+        const positiveResult = testSetup.SetLines(["template<typename T> bool \n", "foo(T a);"]).GetResult();
+        assert.equal("/**\n * @brief \n * \n * @tparam T \n * @param a \n * "
+            + "@return true \n * @return false \n */", positiveResult);
+    });
 });