fix postMessage result

containers · Jan 30, 2024 · 82b0ae7 · 82b0ae7
1 parent f858c19
commit 82b0ae7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/backend/src/managers/playground.spec.ts b/packages/backend/src/managers/playground.spec.ts
@@ -59,13 +59,15 @@ beforeEach(() => {
 });
 
 test('startPlayground should fail if no provider', async () => {
+  mocks.postMessage.mockResolvedValue(undefined);
   mocks.getContainerConnections.mockReturnValue([]);
   await expect(manager.startPlayground('model1', '/path/to/model')).rejects.toThrowError(
     'Unable to find an engine to start playground',
   );
 });
 
 test('startPlayground should download image if not present then create container', async () => {
+  mocks.postMessage.mockResolvedValue(undefined);
   mocks.getContainerConnections.mockReturnValue([
     {
       connection: {
@@ -122,6 +124,7 @@ test('stopPlayground should fail if no playground is running', async () => {
 });
 
 test('stopPlayground should stop a started playground', async () => {
+  mocks.postMessage.mockResolvedValue(undefined);
   mocks.getContainerConnections.mockReturnValue([
     {
       connection: {