feat: add waitForAssets method

lib/assets.js

@@ -3,6 +3,8 @@ import { EventEmitter } from 'node:events';
 export class Assets extends EventEmitter {
     #expectedAssets = new Set();
     #receivedAssets = new Map();
+    #js;
+    #css;
 
     // pointless constructor with super to appease TS which cryptically errors otherwise.
     constructor() {
@@ -29,12 +31,20 @@ export class Assets extends EventEmitter {
         this.#receivedAssets.set(name, assets);
         if (this.allAssetsReceived) {
             const assets = Array.from(this.#receivedAssets.values());
-            const js = assets.flatMap(({ js = [] }) => js).filter(Boolean);
-            const css = assets.flatMap(({ css = [] }) => css).filter(Boolean);
-            this.emit('received', { js, css });
+            this.#js = assets.flatMap(({ js = [] }) => js).filter(Boolean);
+            this.#css = assets.flatMap(({ css = [] }) => css).filter(Boolean);
+            this.emit('received', { js: this.#js, css: this.#css });
         }
     }
 
+    get js() {
+        return this.#js;
+    }
+
+    get css() {
+        return this.#css;
+    }
+
     /**
      * Value will be true if all expected assets have been received
      */

lib/http-incoming.js

@@ -222,6 +222,25 @@ export default class HttpIncoming {
         return this.#js;
     }
 
+    /**
+     * Wait for assets to be received from 
+     * @returns {Promise<{js: Array<import('./asset-js.js').JavaScriptAsset | string>, css: Array<import('./asset-css.js').CssAsset | string>}>}
+     */
+    waitForAssets() {
+        if (this.assets.allAssetsReceived) {
+            return Promise.resolve({
+                js: this.assets.js,
+                css: this.assets.css,
+            });
+        }
+
+        return new Promise((resolve) => {
+            this.assets.on('received', (assets) => {
+                resolve(assets);
+            });
+        });
+    }
+
     /**
      * @returns {PodiumHttpIncoming<T>}
      */

tests/http-incoming.test.js

@@ -359,10 +359,8 @@ tap.test('can read values from view with default types', (t) => {
     t.end();
 });
 
-tap.test('can read values from view with default types', (t) => {
+tap.test('can set asset expectations and receive assets', (t) => {
     t.plan(1);
-    // really only here for tsc
-
     /**
      * @type {HttpIncoming}
      */
@@ -384,3 +382,53 @@ tap.test('can read values from view with default types', (t) => {
         css: [{ value: 'foo.css' }],
     });
 });
+
+tap.test('can wait for expected assets', (t) => {
+    t.plan(1);
+    /**
+     * @type {HttpIncoming}
+     */
+    const incoming = new HttpIncoming(SIMPLE_REQ, SIMPLE_RES);
+    incoming.assets.addExpectedAsset('foo');
+    incoming.assets.addExpectedAsset('bar');
+
+    incoming.waitForAssets().then(() => {
+        t.ok(incoming.assets.allAssetsReceived);
+        t.end();
+    });
+
+    incoming.assets.addReceivedAsset('foo', {
+        js: [{ value: 'foo.js' }],
+        css: [{ value: 'foo.css' }],
+    });
+    incoming.assets.addReceivedAsset('bar', {
+        js: [{ value: 'foo.js' }],
+        css: [{ value: 'foo.css' }],
+    });
+});
+
+tap.test('waitForAssets will resolve even if all assets have already been received', async (t) => {
+    t.plan(3);
+    /**
+     * @type {HttpIncoming}
+     */
+    const incoming = new HttpIncoming(SIMPLE_REQ, SIMPLE_RES);
+    incoming.assets.addExpectedAsset('foo');
+    incoming.assets.addExpectedAsset('bar');
+
+    incoming.assets.addReceivedAsset('foo', {
+        js: [{ value: 'foo.js' }],
+        css: [{ value: 'foo.css' }],
+    });
+    incoming.assets.addReceivedAsset('bar', {
+        js: [{ value: 'foo.js' }],
+        css: [{ value: 'foo.css' }],
+    });
+
+    const { js, css } = await incoming.waitForAssets();
+
+    t.ok(incoming.assets.allAssetsReceived);
+    t.same(js, [{ value: 'foo.js' }, { value: 'foo.js' }]);
+    t.same(css, [{ value: 'foo.css' }, { value: 'foo.css' }]);
+    t.end();
+});