From b446b9e596fc0487b88597d8e818dae61512f309 Mon Sep 17 00:00:00 2001
From: Rowan Goemans <RB.Goemans@student.han.nl>
Date: Fri, 29 Dec 2017 00:15:59 +0100
Subject: -- Added more unit tests. -- Added support for alignas.

---
 src/test/tools/TestSetup.ts | 46 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 src/test/tools/TestSetup.ts

(limited to 'src/test/tools/TestSetup.ts')

diff --git a/src/test/tools/TestSetup.ts b/src/test/tools/TestSetup.ts
new file mode 100644
index 0000000..1270616
--- /dev/null
+++ b/src/test/tools/TestSetup.ts
@@ -0,0 +1,46 @@
+import * as vscode from "vscode";
+import CodeParser from "../../CodeParser/CodeParser";
+import CParser from "../../CodeParser/CParser/CParser";
+import { IDocGen } from "../../DocGen/DocGen";
+import * as myExtension from "../../extension";
+import MockDocument from "./MockDocument";
+import MockEditor from "./MockEditor";
+import MockLine from "./MockLine";
+import MockPosition from "./MockPosition";
+import MockSelection from "./MockSelection";
+
+export default class TestSetup {
+    private editor: MockEditor;
+
+    constructor(method: string) {
+        this.SetLine(method);
+    }
+
+    public SetLine(method: string): TestSetup {
+        let position: MockPosition;
+        position = new MockPosition(0, 0);
+
+        let selection: MockSelection;
+        selection = new MockSelection(position);
+
+        let line: MockLine;
+        line = new MockLine(method);
+
+        let doc: MockDocument;
+        doc = new MockDocument(line);
+
+        this.editor = new MockEditor(selection, doc);
+
+        return this;
+    }
+
+    public GetResult(): string {
+        let parser: CodeParser;
+        parser = new CParser();
+
+        const gen: IDocGen = parser.Parse(this.editor);
+        gen.GenerateDoc(new vscode.Range(new vscode.Position(0, 0), new vscode.Position(0, 0)));
+
+        return this.editor.editBuilder.text;
+    }
+}
-- 
cgit v1.2.3