test: update events tests

maizzle · Jul 18, 2024 · 9227365 · 9227365
1 parent 8175e1b
commit 9227365
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 5 deletions.
diff --git a/test/build.test.js b/test/build.test.js
@@ -64,11 +64,11 @@ describe.concurrent('Build', () => {
             content: ['test/fixtures/build/**/*.html']
           }
         },
-        async afterBuild({ files, config, render }) {
+        async afterBuild({ files, config, transform }) {
           ctx.afterBuild = files
           expect(config).toBeInstanceOf(Object)
           expect(files).toBeInstanceOf(Array)
-          expect(render).toBeInstanceOf(Function)
+          expect(transform.inlineCSS).toBeInstanceOf(Function)
         }
       }
     )

diff --git a/test/render.test.js b/test/render.test.js
@@ -26,8 +26,12 @@ describe.concurrent('Render', () => {
 
   test('Runs the `beforeRender` event', async () => {
     const { html } = await render('<div class="inline">{{ page.foo }}</div>', {
-      beforeRender({ config }) {
+      beforeRender({ config, posthtml, transform }) {
         config.foo = 'bar'
+
+        expect(config).toBeInstanceOf(Object)
+        expect(posthtml).toBeInstanceOf(Function)
+        expect(transform.inlineCSS).toBeInstanceOf(Function)
       }
     })
 
@@ -36,10 +40,14 @@ describe.concurrent('Render', () => {
 
   test('Runs the `afterRender` event', async () => {
     const { html } = await render('<div class="inline">foo</div>', {
-      afterRender({ config }) {
+      afterRender({ config, posthtml, transform }) {
         config.replaceStrings = {
           foo: 'bar'
         }
+
+        expect(config).toBeInstanceOf(Object)
+        expect(posthtml).toBeInstanceOf(Function)
+        expect(transform.inlineCSS).toBeInstanceOf(Function)
       }
     })
 
@@ -51,7 +59,11 @@ describe.concurrent('Render', () => {
       replaceStrings: {
         foo: 'bar'
       },
-      afterTransformers({ html }) {
+      afterTransformers({ html, config, posthtml, transform }) {
+        expect(config).toBeInstanceOf(Object)
+        expect(posthtml).toBeInstanceOf(Function)
+        expect(transform.inlineCSS).toBeInstanceOf(Function)
+
         return html.replace('bar', 'baz')
       }
     })