chore: fix flaky test on CI

sanity-io · Feb 24, 2025 · f94f661 · f94f661
1 parent 9b4886f
commit f94f661
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/sanity/src/core/studio/Studio.test.tsx b/packages/sanity/src/core/studio/Studio.test.tsx
@@ -52,7 +52,7 @@ describe('Studio', () => {
     spy.mockReset()
     spy.mockRestore()
   })
-  it(`SSR to markup for hydration doesn't throw`, () => {
+  it(`SSR to markup for hydration doesn't throw`, async () => {
     const spy = vi.spyOn(console, 'error')
     const node = document.createElement('div')
     document.body.appendChild(node)
@@ -63,7 +63,8 @@ describe('Studio', () => {
       node.innerHTML = html
 
       document.head.innerHTML += sheet.getStyleTags()
-      act(() => hydrateRoot(node, <Studio config={config} />))
+      const root = await act(() => hydrateRoot(node, <Studio config={config} />))
+      await act(() => root.unmount())
     } finally {
       sheet.seal()
     }