test: update tests for refresh global

pmmmwh · Jun 1, 2021 · aab7ccc · aab7ccc
1 parent 8d2cb76
commit aab7ccc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/unit/getRefreshGlobal.test.js b/test/unit/getRefreshGlobal.test.js
@@ -1,3 +1,4 @@
+const { RuntimeGlobals, Template } = require('webpack');
 const getRefreshGlobal = require('../../lib/utils/getRefreshGlobal');
 
 describe('getRefreshGlobal', () => {
@@ -10,7 +11,7 @@ describe('getRefreshGlobal', () => {
   });
 
   it('should return working refresh global without providing runtime template', () => {
-    const refreshGlobalTemplate = getRefreshGlobal();
+    const refreshGlobalTemplate = getRefreshGlobal(Template);
     expect(refreshGlobalTemplate).toMatchInlineSnapshot(`
       "__webpack_require__.$Refresh$ = {
       	register: function() { return undefined; },
@@ -67,10 +68,12 @@ describe('getRefreshGlobal', () => {
 
   it.skipIf(
     WEBPACK_VERSION !== 5,
-    'should return working refresh global with provided runtime template',
+    'should return working refresh global with provided runtime globals and runtime template',
     () => {
       const RuntimeTemplate = require('webpack/lib/RuntimeTemplate');
       const refreshGlobalTemplate = getRefreshGlobal(
+        Template,
+        RuntimeGlobals,
         new RuntimeTemplate({}, { environment: { arrowFunction: true, const: true } }, (i) => i)
       );
       expect(refreshGlobalTemplate).toMatchInlineSnapshot(`