Add a test for forcing the subtree to be static via a special flag

Author: jviide

test/statics-caching.test.mjs

@@ -91,16 +91,20 @@ describe('htm', () => {
 	});
 
 	describe('the h function should be able to modify `this[0]`', () => {
-		test('should be able to force subtrees to be static', () => {
-			function wrapH(h) {
-				return function(type, props, ...children) {
-					if (props['@static']) {
-						this[0] &= ~3;
-					}
-					return h(type, props, ...children);
-				};
-			}
-
+		function wrapH(h) {
+			return function(type, props, ...children) {
+				if (type === '@static') {
+					this[0] &= ~3;
+					return children;
+				}
+				if (props['@static']) {
+					this[0] &= ~3;
+				}
+				return h(type, props, ...children);
+			};
+		}
+
+		test('should be able to force subtrees to be static via a prop', () => {
 			const html = htm.bind(wrapH(h));
 			const x = () => html`<div @static>${'a'}</div>`;
 			const a = x();
@@ -109,5 +113,15 @@ describe('htm', () => {
 			expect(b).toEqual({ tag: 'div', props: { '@static': true }, children: ['a'] });
 			expect(a).toBe(b);
 		});
+
+		test('should be able to force subtrees to be static via a special tag', () => {
+			const html = htm.bind(wrapH(h));
+			const x = () => html`<@static>${'a'}<//>`;
+			const a = x();
+			const b = x();
+			expect(a).toEqual(['a']);
+			expect(b).toEqual(['a']);
+			expect(a).toBe(b);
+		});
 	});
 });