test: add test for createDoneEvent() and improve test for transformPayloadForOpenAICompatibility() (#36)

Author: gr2m

test/parse.test.js

@@ -28,10 +28,23 @@ suite("request parsing", () => {
 
   test("transformPayloadForOpenAICompatibility()", (t) => {
     const payload = transformPayloadForOpenAICompatibility({
-      messages: [],
+      messages: [
+        {
+          role: "role",
+          name: "name",
+          content: "content",
+          someCopilotKey: "value",
+        },
+      ],
       someCopilotKey: "value",
     });
-    t.assert.deepStrictEqual(payload.messages, []);
+    t.assert.deepStrictEqual(payload.messages, [
+      {
+        role: "role",
+        name: "name",
+        content: "content",
+      },
+    ]);
   });
 
   test("verifyAndParseRequest()", async (t) => {

test/response.test.js

@@ -28,6 +28,13 @@ suite("response", () => {
     t.assert.snapshot(event.toString());
   });
 
+  test("createTextEvent()", (t) => {
+    const event = createTextEvent("test");
+    t.assert.equal(undefined, event.event);
+    t.assert.snapshot(event.data);
+    t.assert.snapshot(event.toString());
+  });
+
   test("createConfirmationEvent()", (t) => {
     const event = createConfirmationEvent({
       id: "123",

test/response.test.js.snapshot

@@ -99,3 +99,20 @@ exports[`response > createReferencesEvent() 1`] = `
 exports[`response > createReferencesEvent() 2`] = `
 "event: copilot_references\\ndata: [{\\"type\\":\\"test.story\\",\\"id\\":\\"test\\",\\"data\\":{\\"file\\":\\"test.js\\",\\"start\\":\\"1\\",\\"end\\":\\"42\\",\\"content\\":\\"function test() {...}\\"},\\"is_implicit\\":false,\\"metadata\\":{\\"display_name\\":\\"Lines 1-42 from test.js\\",\\"display_icon\\":\\"test-icon\\",\\"display_url\\":\\"http://github.com/monalisa/hello-world/blob/main/test.js#L1-L42\\"}}]\\n\\n"
 `;
+
+exports[`response > createTextEvent() 1`] = `
+{
+  "choices": [
+    {
+      "delta": {
+        "content": "test",
+        "role": "assistant"
+      }
+    }
+  ]
+}
+`;
+
+exports[`response > createTextEvent() 2`] = `
+"data: {\\"choices\\":[{\\"delta\\":{\\"content\\":\\"test\\",\\"role\\":\\"assistant\\"}}]}\\n\\n"
+`;