From 838279a61349be6b7ee09ac1b1775b0d6da12028 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Isaac=20Rold=C3=A1n?= <isaac.roldan@shopify.com>
Date: Wed, 20 Nov 2024 15:31:32 +0100
Subject: [PATCH] add tests

---
 .../dev/app-events/app-event-watcher.test.ts  |  2 +-
 .../app-events/app-watcher-esbuild.test.ts    | 37 +++++++++++--------
 2 files changed, 23 insertions(+), 16 deletions(-)

diff --git a/packages/app/src/cli/services/dev/app-events/app-event-watcher.test.ts b/packages/app/src/cli/services/dev/app-events/app-event-watcher.test.ts
index 64cf2932187..eb1fbeb4865 100644
--- a/packages/app/src/cli/services/dev/app-events/app-event-watcher.test.ts
+++ b/packages/app/src/cli/services/dev/app-events/app-event-watcher.test.ts
@@ -323,7 +323,7 @@ describe('app-event-watcher build extension errors', () => {
       }
 
       const mockManager = new MockESBuildContextManager()
-      mockManager.contexts.h1.rebuild.mockRejectedValueOnce(esbuildError)
+      mockManager.rebuildContext = vi.fn().mockRejectedValueOnce(esbuildError)
 
       const buildOutputPath = joinPath(tmpDir, '.shopify', 'bundle')
       const app = testAppLinked({
diff --git a/packages/app/src/cli/services/dev/app-events/app-watcher-esbuild.test.ts b/packages/app/src/cli/services/dev/app-events/app-watcher-esbuild.test.ts
index e2ab43b8534..800f6d78c5a 100644
--- a/packages/app/src/cli/services/dev/app-events/app-watcher-esbuild.test.ts
+++ b/packages/app/src/cli/services/dev/app-events/app-watcher-esbuild.test.ts
@@ -2,6 +2,7 @@ import {DevAppWatcherOptions, ESBuildContextManager} from './app-watcher-esbuild
 import {AppEvent, EventType} from './app-event-watcher.js'
 import {testAppLinked, testUIExtension} from '../../../models/app/app.test-data.js'
 import {describe, expect, test, vi} from 'vitest'
+import * as fs from '@shopify/cli-kit/node/fs'
 
 vi.mock('@luckycatfactory/esbuild-graphql-loader', () => ({
   default: {
@@ -15,13 +16,14 @@ const extension1 = await testUIExtension({type: 'ui_extension', handle: 'h1', di
 const extension2 = await testUIExtension({type: 'ui_extension', directory: '/extensions/ui_extension_2'})
 
 describe('app-watcher-esbuild', () => {
+  const options: DevAppWatcherOptions = {
+    dotEnvVariables: {key: 'value'},
+    url: 'http://localhost:3000',
+    outputPath: '/path/to/output',
+  }
+
   test('creating contexts', async () => {
     // Given
-    const options: DevAppWatcherOptions = {
-      dotEnvVariables: {key: 'value'},
-      url: 'http://localhost:3000',
-      outputPath: '/path/to/output',
-    }
     const manager = new ESBuildContextManager(options)
     const extensions = [extension1, extension2]
 
@@ -35,11 +37,6 @@ describe('app-watcher-esbuild', () => {
 
   test('deleting contexts', async () => {
     // Given
-    const options: DevAppWatcherOptions = {
-      dotEnvVariables: {key: 'value'},
-      url: 'http://localhost:3000',
-      outputPath: '/path/to/output',
-    }
     const manager = new ESBuildContextManager(options)
     const extensions = [extension1, extension2]
     await manager.createContexts(extensions)
@@ -54,11 +51,6 @@ describe('app-watcher-esbuild', () => {
 
   test('updating contexts with an app event', async () => {
     // Given
-    const options: DevAppWatcherOptions = {
-      dotEnvVariables: {key: 'value'},
-      url: 'http://localhost:3000',
-      outputPath: '/path/to/output',
-    }
     const manager = new ESBuildContextManager(options)
     await manager.createContexts([extension2])
 
@@ -79,4 +71,19 @@ describe('app-watcher-esbuild', () => {
     expect(manager.contexts).toHaveProperty('h1')
     expect(manager.contexts).not.toHaveProperty('test-ui-extension')
   })
+
+  test('rebuilding contexts', async () => {
+    // Given
+    const manager = new ESBuildContextManager(options)
+    await manager.createContexts([extension1])
+    const spyContext = vi.spyOn(manager.contexts.h1!, 'rebuild').mockResolvedValue({} as any)
+    const spyCopy = vi.spyOn(fs, 'copyFile').mockResolvedValue()
+
+    // When
+    await manager.rebuildContext(extension1)
+
+    // Then
+    expect(spyContext).toHaveBeenCalled()
+    expect(spyCopy).toHaveBeenCalledWith('/path/to/output/h1/dist', '/extensions/ui_extension_1/dist')
+  })
 })