fix: update mocks

OpenFn · Dec 1, 2024 · cc070a6 · cc070a6
1 parent 6293248
commit cc070a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/packages/cli/test/docgen/handler.test.ts b/packages/cli/test/docgen/handler.test.ts
@@ -31,6 +31,7 @@ const loadJSON = async (path: string) => {
 
 // Mock doc gen function
 const mockGen: DocGenFn = async () => ({
+  namespaces: [{ type: 'namespace', name: 'smth' }],
   name: 'test',
   version: '1.0.0',
   functions: [
@@ -41,6 +42,7 @@ const mockGen: DocGenFn = async () => ({
       magic: false,
       parameters: [],
       examples: [],
+      type: 'function',
     },
   ],
 });

diff --git a/packages/ws-worker/src/mock/sockets.ts b/packages/ws-worker/src/mock/sockets.ts
@@ -78,13 +78,15 @@ export class MockSocket {
   private allChannels: ChannelMap;
   private callbacks: Record<string, EventHandler>;
 
+  endpoint: string;
   constructor(
-    private _endpoint: string = '',
-    private channels: ChannelMap = {},
+    endpoint: string = '',
+    channels: ChannelMap = {},
     private _connect: () => Promise<void> = async () => {}
   ) {
     this.allChannels = channels;
     this.callbacks = {};
+    this.endpoint = endpoint;
   }
 
   onOpen(callback: EventHandler): void {