Add tests

2 files changed, 43 insertions, 0 deletions

diff --git a/src/test/CppTests/Config.test.ts b/src/test/CppTests/Config.test.ts
index 0459dcb..bb16c20 100644
--- a/src/test/CppTests/Config.test.ts
+++ b/src/test/CppTests/Config.test.ts
@@ -8,6 +8,7 @@ import * as assert from "assert";
 
 // You can import and use all API from the 'vscode' module
 // as well as import your extension to test it
+import * as moment from "moment";
 import * as vscode from "vscode";
 import { Config } from "../../Config";
 import TestSetup from "./TestSetup";
@@ -92,4 +93,13 @@ suite("C++ - Configuration Tests", () => {
         const negativeResult = testSetup.SetLines(["template<typename T> bool \n", "foo(T a);"]).GetResult();
         assert.equal("/**\n * @brief \n * \n */", negativeResult);
     });
+
+    test("File description order test", () => {
+        testSetup.cfg = new Config();
+        testSetup.firstLine = 0;
+        testSetup.cfg.fileOrder = ["brief", "author", "date", "file"];
+        const result = testSetup.SetLine("").GetResult();
+        assert.equal("/**\n * @brief \n * \n * @author your name\n" +
+            " * @date " + moment().format("YYYY-MM-DD") + "\n * @file MockDocument.h\n */", result);
+    });
 });
diff --git a/src/test/CppTests/FileDescription.test.ts b/src/test/CppTests/FileDescription.test.ts
new file mode 100644
index 0000000..ab6071f
--- /dev/null
+++ b/src/test/CppTests/FileDescription.test.ts
@@ -0,0 +1,33 @@
+//
+// Note: This example test is leveraging the Mocha test framework.
+// Please refer to their documentation on https://mochajs.org/ for help.
+//
+
+// The module 'assert' provides assertion methods from node
+import * as assert from "assert";
+
+// You can import and use all API from the 'vscode' module
+// as well as import your extension to test it
+import * as moment from "moment";
+import * as vscode from "vscode";
+import TestSetup from "./TestSetup";
+
+// Defines a Mocha test suite to group tests of similar kind together
+suite("File Description Tests", () => {
+    const testSetup: TestSetup = new TestSetup("void foo();");
+    const date = moment().format("YYYY-MM-DD");
+
+    // Tests
+    test("#include on next line", () => {
+        const result = testSetup.SetLine("#include <iostream>").GetResult();
+        assert.equal("/**\n * @brief \n * \n * @file MockDocument.h\n * @author your name\n" +
+            " * @date " + date + "\n */", result);
+    });
+
+    test("On first line of document", () => {
+        const result = testSetup.SetLine("").GetResult();
+        assert.equal("/**\n * @brief \n * \n * @file MockDocument.h\n * @author your name\n" +
+            " * @date " + date + "\n */", result);
+    });
+
+});