🐛 models: Fix crash on api error (#137)

lowczarc · web-flow · commit d5019a53b588 · 2024-04-30T20:27:27.000+09:00
* 🐛 models: Fix crash on api error

* 🐛 tests: Increase test timeout
diff --git a/lib/generate.ts b/lib/generate.ts
@@ -366,10 +366,11 @@ function stream(
         ws.onopen = () => ws.send(JSON.stringify(requestBody));
         ws.onmessage = (data: unknown) => onMessage(data, resultStream);
         ws.onclose = () => resultStream.push(null);
+        ws.onerror = () => resultStream.push(null);
         resultStream.stop = () => {
             ws.send("STOP");
         };
-    })();
+    })().catch(console.error);
     return resultStream;
 }
 
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "polyfire-js",
-    "version": "0.2.58",
+    "version": "0.2.59",
     "main": "index.js",
     "types": "index.d.ts",
     "author": "Lancelot Owczarczak <lancelot@owczarczak.fr>",
diff --git a/tests/generation.test.js b/tests/generation.test.js
@@ -96,4 +96,5 @@ test(
 
         expect(message).toBe("banana42");
     }),
+    10000,
 );

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "polyfire-js",`
`3`		`- "version": "0.2.58",`
	`3`	`+ "version": "0.2.59",`
`4`	`4`	`"main": "index.js",`
`5`	`5`	`"types": "index.d.ts",`
`6`	`6`	`"author": "Lancelot Owczarczak <[email protected]>",`
-Original file line number
+Diff line change
         expect(message).toBe("banana42");
     }),
 +    10000,
 );