3 files changed, 54 insertions, 0 deletions
diff --git a/extension/src/test-suite/environmentSetup.test.ts b/extension/src/test-suite/environmentSetup.test.ts
new file mode 100644
index 00000000..9a478522
--- /dev/null
+++ b/extension/src/test-suite/environmentSetup.test.ts
@@ -0,0 +1,20 @@
+import { test, describe } from "mocha";
+import * as assert from "assert";
+
+import { getContinueServerUrl } from "../bridge";
+import { startContinuePythonServer } from "../activation/environmentSetup";
+import fetch from "node-fetch";
+
+describe("Can start python server", () => {
+  test("Can start python server in under 10 seconds", async function () {
+    this.timeout(17_000);
+    await startContinuePythonServer();
+
+    await new Promise((resolve) => setTimeout(resolve, 15_000));
+
+    // Check if server is running
+    const serverUrl = getContinueServerUrl();
+    const response = await fetch(`${serverUrl}/health`);
+    assert.equal(response.status, 200);
+  });
+});
diff --git a/extension/src/test-suite/extension.test.ts b/extension/src/test-suite/extension.test.ts
new file mode 100644
index 00000000..890820b2
--- /dev/null
+++ b/extension/src/test-suite/extension.test.ts
@@ -0,0 +1,16 @@
+import { test, describe } from "mocha";
+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 vscode from "vscode";
+// import * as myExtension from '../../extension';
+
+describe("Extension Test Suite", () => {
+  vscode.window.showInformationMessage("Start all tests.");
+
+  test("Sample test", () => {
+    assert.strictEqual(-1, [1, 2, 3].indexOf(5));
+    assert.strictEqual(-1, [1, 2, 3].indexOf(0));
+  });
+});
diff --git a/extension/src/test-suite/util.test.ts b/extension/src/test-suite/util.test.ts
new file mode 100644
index 00000000..2b301b0c
--- /dev/null
+++ b/extension/src/test-suite/util.test.ts
@@ -0,0 +1,18 @@
+import { test, describe } from "mocha";
+import assert from "assert";
+import { convertSingleToDoubleQuoteJSON } from "../util/util";
+
+describe("utils.ts", () => {
+  test("convertSingleToDoubleQuoteJson", () => {
+    let pairs = [
+      [`{'a': 'b'}`, `{"a": "b"}`],
+      [`{'a': "b", "c": 'd'}`, `{"a": "b", "c": "d"}`],
+      [`{'a': '\\'"'}`, `{"a": "'\\""}`],
+    ];
+    for (let pair of pairs) {
+      let result = convertSingleToDoubleQuoteJSON(pair[0]);
+      assert(result === pair[1]);
+      JSON.parse(result);
+    }
+  });
+});