Wrap children result in an array if not already in one (#844)

* wrap children result in an array if not already in one

* add test for wrapped children

Author: tomdye

src/testing/renderer.ts

@@ -280,6 +280,9 @@ export function assertion(renderFunc: () => DNode | DNode[]) {
 			const node = findNode(render, wrapped);
 			node.children = node.children || [];
 			let childrenResult = children();
+			if (!Array.isArray(childrenResult)) {
+				childrenResult = [childrenResult];
+			}
 			switch (type) {
 				case 'prepend':
 					node.children = [...childrenResult, ...node.children];

tests/testing/unit/renderer.tsx

@@ -758,5 +758,19 @@ describe('test renderer', () => {
 				))
 			);
 		});
+
+		it('Should wrap single children in an array when calling setChildren', () => {
+			const factory = create().children<string>();
+			const TestWidget = factory(() => 'foo');
+			const App = create()(() => {
+				return <TestWidget>bar</TestWidget>;
+			});
+			const WrappedTestWudget = wrap(TestWidget);
+
+			const testAssertion = assertion(() => <WrappedTestWudget>bar</WrappedTestWudget>);
+			const r = renderer(() => <App />);
+
+			r.expect(testAssertion.setChildren(WrappedTestWudget, () => 'bar'));
+		});
 	});
 });