DEV test update

Author: cdcabrera

src/__tests__/__snapshots__/server.toolsHost.test.ts.snap

@@ -346,3 +346,54 @@ exports[`requestShutdown should send shutdown:ack and exit, with valid request 1
   ],
 ]
 `;
+
+exports[`setHandlers should set up message handlers and attempt handle requests, hello 1`] = `
+[
+  [
+    {
+      "id": "test-id",
+      "t": "hello:ack",
+    },
+  ],
+]
+`;
+
+exports[`setHandlers should set up message handlers and attempt handle requests, invoke 1`] = `
+[
+  [
+    {
+      "error": {
+        "message": "Unknown toolId",
+      },
+      "id": "test-id",
+      "ok": false,
+      "t": "invoke:result",
+    },
+  ],
+]
+`;
+
+exports[`setHandlers should set up message handlers and attempt handle requests, load 1`] = `
+[
+  [
+    {
+      "errors": [],
+      "id": "test-id",
+      "t": "load:ack",
+      "warnings": [],
+    },
+  ],
+]
+`;
+
+exports[`setHandlers should set up message handlers and attempt handle requests, manifest:get 1`] = `
+[
+  [
+    {
+      "id": "test-id",
+      "t": "manifest:result",
+      "tools": [],
+    },
+  ],
+]
+`;

src/__tests__/server.toolsHost.test.ts

@@ -390,3 +390,47 @@ describe('requestFallback', () => {
     }).not.toThrow();
   });
 });
+
+describe('setHandlers', () => {
+  let mockOn: jest.Mock;
+  let mockSend: jest.Mock;
+
+  beforeEach(() => {
+    mockSend = jest.fn();
+    mockOn = jest.fn();
+
+    process.on = mockOn;
+    process.send = mockSend;
+  });
+
+  afterEach(() => {
+    jest.clearAllMocks();
+    delete (process as any).send;
+  });
+
+  it.each([
+    {
+      description: 'hello',
+      request: { t: 'hello', id: 'test-id' }
+    },
+    {
+      description: 'load',
+      request: { t: 'load', id: 'test-id' }
+    },
+    {
+      description: 'manifest:get',
+      request: { t: 'manifest:get', id: 'test-id' }
+    },
+    {
+      description: 'invoke',
+      request: { t: 'invoke', id: 'test-id' }
+    }
+  ])('should set up message handlers and attempt handle requests, $description', async ({ request }) => {
+    const handler = setHandlers();
+
+    await handler(request as any);
+
+    expect(mockOn).toHaveBeenCalledWith('message', expect.any(Function));
+    expect(mockSend.mock.calls).toMatchSnapshot();
+  });
+});