@quenty/studio-bridge 0.7.1 → 0.9.0

package/src/server/web-socket-protocol-basic.test.ts

@@ -0,0 +1,131 @@
+import { describe, it, expect } from 'vitest';
+import { encodeMessage, decodePluginMessage } from './web-socket-protocol.js';
+
+describe('encodeMessage', () => {
+  it('encodes a shutdown message', () => {
+    const json = encodeMessage({
+      type: 'shutdown',
+      sessionId: 'abc-123',
+      payload: {},
+    });
+    const parsed = JSON.parse(json);
+    expect(parsed).toEqual({
+      type: 'shutdown',
+      sessionId: 'abc-123',
+      payload: {},
+    });
+  });
+
+  it('encodes an execute message', () => {
+    const json = encodeMessage({
+      type: 'execute',
+      sessionId: 'abc-123',
+      payload: { script: 'print("hi")' },
+    });
+    const parsed = JSON.parse(json);
+    expect(parsed).toEqual({
+      type: 'execute',
+      sessionId: 'abc-123',
+      payload: { script: 'print("hi")' },
+    });
+  });
+});
+
+describe('decodePluginMessage', () => {
+  describe('scriptComplete', () => {
+    it('decodes a successful scriptComplete', () => {
+      const msg = decodePluginMessage(
+        JSON.stringify({
+          type: 'scriptComplete',
+          sessionId: 'test-session',
+          payload: { success: true },
+        })
+      );
+      expect(msg).toEqual({
+        type: 'scriptComplete',
+        sessionId: 'test-session',
+        payload: { success: true, error: undefined },
+      });
+    });
+
+    it('decodes a failed scriptComplete with error', () => {
+      const msg = decodePluginMessage(
+        JSON.stringify({
+          type: 'scriptComplete',
+          sessionId: 'test-session',
+          payload: { success: false, error: 'Script errored' },
+        })
+      );
+      expect(msg).toEqual({
+        type: 'scriptComplete',
+        sessionId: 'test-session',
+        payload: { success: false, error: 'Script errored' },
+      });
+    });
+
+    it('returns null if success is not boolean', () => {
+      const msg = decodePluginMessage(
+        JSON.stringify({
+          type: 'scriptComplete',
+          sessionId: 'test-session',
+          payload: { success: 'yes' },
+        })
+      );
+      expect(msg).toBeNull();
+    });
+
+    it('returns null for scriptComplete without top-level sessionId', () => {
+      const msg = decodePluginMessage(
+        JSON.stringify({
+          type: 'scriptComplete',
+          payload: { success: true },
+        })
+      );
+      expect(msg).toBeNull();
+    });
+  });
+
+  describe('malformed messages', () => {
+    it('returns null for invalid JSON', () => {
+      expect(decodePluginMessage('not json')).toBeNull();
+    });
+
+    it('returns null for non-object JSON', () => {
+      expect(decodePluginMessage('"just a string"')).toBeNull();
+    });
+
+    it('returns null for missing type', () => {
+      expect(
+        decodePluginMessage(JSON.stringify({ sessionId: 's', payload: {} }))
+      ).toBeNull();
+    });
+
+    it('returns null for missing payload', () => {
+      expect(
+        decodePluginMessage(
+          JSON.stringify({ type: 'register', sessionId: 's' })
+        )
+      ).toBeNull();
+    });
+
+    it('returns null for missing sessionId', () => {
+      expect(
+        decodePluginMessage(
+          JSON.stringify({ type: 'register', payload: { sessionId: 's' } })
+        )
+      ).toBeNull();
+    });
+
+    it('returns null for unknown message type', () => {
+      expect(
+        decodePluginMessage(
+          JSON.stringify({
+            type: 'unknown',
+            sessionId: 'test',
+            payload: {},
+          })
+        )
+      ).toBeNull();
+    });
+  });
+});