feat: Added support for document.images (#219)

Co-authored-by: Cameron Wheeler <cwheeler@yelp.com>
BuilderIO · Jul 23, 2022 · 10ba9db · 10ba9db · vercel · Jul 23, 2022
1 parent d648520
commit 10ba9db
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/src/lib/web-worker/worker-document.ts b/src/lib/web-worker/worker-document.ts
@@ -169,6 +169,12 @@ export const patchDocument = (
       },
     },
 
+    images: {
+      get() {
+        return getter(this, ['images']);
+      },
+    },
+
     implementation: {
       get() {
         return {

diff --git a/tests/platform/document/document.spec.ts b/tests/platform/document/document.spec.ts
@@ -96,4 +96,8 @@ test('document', async ({ page }) => {
 
   const testVisibilityState = page.locator('#testVisibilityState');
   await expect(testVisibilityState).toHaveText('visible');
+
+  const testImages = page.locator('#testDocumentImages');
+  const pageUrl = new URL(page.url());
+  await expect(testImages).toHaveText(`${pageUrl.origin}/fake.jpg`);
 });
diff --git a/tests/platform/document/index.html b/tests/platform/document/index.html
@@ -452,6 +452,18 @@ <h1 class="title">Document</h1>
         </script>
       </li>
 
+      <li>
+        <strong>document.images</strong>
+        <code id="testDocumentImages"></code>
+        <img src="/fake.jpg" />
+        <script type="text/partytown">
+          (function () {
+            const elm = document.getElementById('testDocumentImages');
+            elm.textContent = document.images[0].src;
+          })();
+        </script>
+      </li>
+
       <script type="text/partytown">
         (function () {
           document.body.classList.add('completed');