1 files changed, 6 insertions, 2 deletions
diff --git a/test/integration/helpers.cjs b/test/integration/helpers.cjs
index d38f1b6..10b0dc6 100644
--- a/test/integration/helpers.cjs
+++ b/test/integration/helpers.cjs
@@ -51,10 +51,10 @@ global.page = {
   }
 };
 
-global.mockResponse = async function mockResponse(driver, urlPath, template, createFn) {
+global.mockResponse = async function mockResponse(driver, urlPath, template, buildFn) {
   const connection = driver.createCDPConnection('page');
   const dom = new JSDOM(template);
-  const contents = createFn(dom.window.document);
+  const contents = buildFn(dom.window.document);
 
   const httpResponse = new HttpResponse(url(urlPath));
   httpResponse.body = contents;
@@ -86,3 +86,7 @@ global.selectCounter = function (counter) {
   counter.querySelector('.counter').classList.add('selected');
   return counter;
 }
+
+global.svgDocument = function (document) {
+  return `<?xml version="1.0" standalone="yes"?>\n` + document.querySelector('svg').outerHTML;
+}