Merge pull request #172 from preactjs/bugfix/invoke-diff-hook

Author: marvinhagemeister

package.json

@@ -27,7 +27,8 @@
 		"transpile": "microbundle src/index.js -f es,umd --target web --external preact",
 		"transpile:jsx": "microbundle src/jsx.js -o dist/jsx.js --target web --external none && microbundle dist/jsx.js -o dist/jsx.js -f cjs",
 		"copy-typescript-definition": "copyfiles -f src/*.d.ts dist",
-		"test": "eslint src test && tsc && mocha -r babel-core/register test/**/*.js",
+		"test": "eslint src test && tsc && npm run test:mocha",
+		"test:mocha": "mocha -r babel-core/register test/**/*.js",
 		"format": "prettier src/**/*.{d.ts,js} test/**/*.js --write",
 		"prepublishOnly": "npm run build",
 		"release": "npm run build && git commit -am $npm_package_version && git tag $npm_package_version && git push && git push --tags && npm publish"

src/index.js

@@ -101,7 +101,10 @@ function renderToString(vnode, context, opts, inner, isSvgMode, selectValue) {
 				__h: []
 			});
 
-			// options.render
+			// options._diff
+			if (options.__b) options.__b(vnode);
+
+			// options._render
 			if (options.__r) options.__r(vnode);
 
 			if (

test/render.js

@@ -1,5 +1,5 @@
 import { render, shallowRender } from '../src';
-import { h, Component, createContext, Fragment } from 'preact';
+import { h, Component, createContext, Fragment, options } from 'preact';
 import { useState, useContext, useEffect, useLayoutEffect } from 'preact/hooks';
 import chai, { expect } from 'chai';
 import { spy, stub, match } from 'sinon';
@@ -1049,6 +1049,47 @@ describe('render', () => {
 		});
 	});
 
+	it('should invoke options._render and options._diff', () => {
+		const calls = [];
+		// _diff
+		const oldDiff = options.__b;
+		options.__b = (...args) => {
+			calls.push(['_diff', args]);
+			if (oldDiff) oldDiff(...args);
+		};
+		// _render
+		const oldRender = options.__r;
+		options.__r = (...args) => {
+			calls.push(['_render', args]);
+			if (oldRender) oldRender(...args);
+		};
+
+		function Component1({ children }) {
+			return children;
+		}
+
+		function Component2() {
+			return <div />;
+		}
+
+		const vnode2 = <Component2>1</Component2>;
+		const vnode1 = <Component1>{vnode2}</Component1>;
+
+		render(vnode1);
+
+		expect(calls).to.deep.equal([
+			['_diff', [vnode1]],
+			['_render', [vnode1]],
+			['_diff', [vnode2]],
+			['_render', [vnode2]]
+		]);
+
+		expect(calls.length).to.equal(4);
+
+		options.__b = oldDiff;
+		options.__r = oldRender;
+	});
+
 	it('should render select value on option', () => {
 		let res = render(
 			<select value="B">