✅ Allow mocked serialize to accept options

percy · Oct 19, 2020 · 14a522d · 14a522d
1 parent aa1e71e
commit 14a522d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/packages/sdk-utils/test/helper.js b/packages/sdk-utils/test/helper.js
@@ -1,11 +1,18 @@
 const createTestServer = require('@percy/core/test/helpers/server');
 const sdk = {};
 
+// mock serialization script
+const serializeDOM = (options) => {
+  let doc = (options.dom || document).documentElement;
+  if (options.domTransformation) options.domTransformation(doc);
+  return doc.outerHTML;
+};
+
 sdk.setup = async function setup() {
   // mock percy server
   sdk.server = await createTestServer({
     '/percy/dom.js': () => [200, 'application/javascript', (
-      `window.PercyDOM = { serialize: ${sdk.serialize.toString()} }`)],
+      `window.PercyDOM = { serialize: ${sdk.serializeDOM.toString()} }`)],
     '/percy/healthcheck': () => [200, 'application/json', (
       { success: true, config: { snapshot: { widths: [1280] } } })],
     '/percy/snapshot': () => [200, 'application/json', { success: true }]
@@ -14,7 +21,7 @@ sdk.setup = async function setup() {
   // reset things
   delete process.env.PERCY_CLI_API;
   delete process.env.PERCY_LOGLEVEL;
-  sdk.serialize = () => document.documentElement.outerHTML;
+  sdk.serializeDOM = serializeDOM;
   sdk.stdio[1] = []; sdk.stdio[2] = [];
 
   let utils = require('..');

diff --git a/packages/sdk-utils/test/index.test.js b/packages/sdk-utils/test/index.test.js
@@ -205,7 +205,7 @@ describe('SDK Utils', () => {
     it('fetches @percy/dom from the CLI API and caches the result', async () => {
       ({ fetchPercyDOM } = sdk.rerequire('..'));
       await expect(fetchPercyDOM()).resolves.toEqual(
-        'window.PercyDOM = { serialize: () => document.documentElement.outerHTML }');
+        `window.PercyDOM = { serialize: ${sdk.serializeDOM.toString()} }`);
       await expect(fetchPercyDOM()).resolves.toBeDefined();
       expect(sdk.server.requests).toEqual([['/percy/dom.js']]);
     });