From 88de62deec589a9d1f221e2b4951c53088743be1 Mon Sep 17 00:00:00 2001
From: Marija Najdova <mnajdova@gmail.com>
Date: Thu, 22 Apr 2021 15:27:35 +0200
Subject: [PATCH] test updates

---
 packages/react/__tests__/rehydration.js                |  9 +++++----
 packages/server/src/index.js                           |  2 +-
 .../extract-critical-to-chunks.test.js.snap            |  4 ++--
 .../server/test/extract-critical-to-chunks.test.js     | 10 +++++++---
 4 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/packages/react/__tests__/rehydration.js b/packages/react/__tests__/rehydration.js
index f6db19c81..60b997602 100644
--- a/packages/react/__tests__/rehydration.js
+++ b/packages/react/__tests__/rehydration.js
@@ -204,9 +204,10 @@ test('global styles can be removed individually after rehydrating SSRed HTML', (
     resetAllModules()
 
     let cache = createCache({ key: 'mui' })
-    let { extractCriticalToChunks, constructStyleTags } = createEmotionServer(
-      cache
-    )
+    let {
+      extractCriticalToChunks,
+      constructStyleTagsFromChunks
+    } = createEmotionServer(cache)
 
     const rendered = ReactDOMServer.renderToString(
       <CacheProvider value={cache}>
@@ -220,7 +221,7 @@ test('global styles can be removed individually after rehydrating SSRed HTML', (
     const extracted = extractCriticalToChunks(rendered)
     return {
       app: extracted.html,
-      styles: constructStyleTags(extracted)
+      styles: constructStyleTagsFromChunks(extracted)
     }
   })
 
diff --git a/packages/server/src/index.js b/packages/server/src/index.js
index 144f862a3..94000e3e6 100644
--- a/packages/server/src/index.js
+++ b/packages/server/src/index.js
@@ -7,5 +7,5 @@ export const {
   extractCriticalToChunks,
   renderStylesToString,
   renderStylesToNodeStream,
-  constructStyleTags
+  constructStyleTagsFromChunks
 } = createEmotionServer(cache)
diff --git a/packages/server/test/__snapshots__/extract-critical-to-chunks.test.js.snap b/packages/server/test/__snapshots__/extract-critical-to-chunks.test.js.snap
index 772423c0d..e37bd760f 100644
--- a/packages/server/test/__snapshots__/extract-critical-to-chunks.test.js.snap
+++ b/packages/server/test/__snapshots__/extract-critical-to-chunks.test.js.snap
@@ -1,6 +1,6 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[`extractCriticalToChunks generates correct style tags using constructStyleTags 1`] = `
+exports[`extractCriticalToChunks generates correct style tags using constructStyleTagsFromChunks 1`] = `
 <style data-emotion="css-global l6h">
   body{color:white;}
 </style>
@@ -12,7 +12,7 @@ exports[`extractCriticalToChunks generates correct style tags using constructSty
 </style>
 `;
 
-exports[`extractCriticalToChunks generates correct style tags using constructStyleTags 2`] = `
+exports[`extractCriticalToChunks generates correct style tags using constructStyleTagsFromChunks 2`] = `
 <style data-emotion="css-global l6h">
   body{color:white;}
 </style>
diff --git a/packages/server/test/extract-critical-to-chunks.test.js b/packages/server/test/extract-critical-to-chunks.test.js
index d48eb96d9..b035592fc 100644
--- a/packages/server/test/extract-critical-to-chunks.test.js
+++ b/packages/server/test/extract-critical-to-chunks.test.js
@@ -71,8 +71,12 @@ describe('extractCriticalToChunks', () => {
     expect(prettifyCriticalChunks(page2Critical)).toMatchSnapshot()
   })
 
-  test('generates correct style tags using constructStyleTags', () => {
-    expect(emotionServer.constructStyleTags(page1Critical)).toMatchSnapshot()
-    expect(emotionServer.constructStyleTags(page2Critical)).toMatchSnapshot()
+  test('generates correct style tags using constructStyleTagsFromChunks', () => {
+    expect(
+      emotionServer.constructStyleTagsFromChunks(page1Critical)
+    ).toMatchSnapshot()
+    expect(
+      emotionServer.constructStyleTagsFromChunks(page2Critical)
+    ).toMatchSnapshot()
   })
 })