Skip to content

Commit 036fbc0

Browse files
committed
minor cleanup
1 parent 35b08a8 commit 036fbc0

File tree

2 files changed

+8
-11
lines changed

2 files changed

+8
-11
lines changed

src/renderHookToSnapshotStream.ts renamed to src/renderHookToSnapshotStream.tsx

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import { createRenderStream } from "./renderStream/createRenderStream.js";
33
import type { NextRenderOptions } from "./renderStream/createRenderStream.js";
44

55
import { Render } from "./renderStream/Render.js";
6-
import { createElement } from "react";
6+
import { createElement } from "rehackt";
77
import { Assertable, assertableSymbol, markAssertable } from "./assertable.js";
88

99
export interface SnapshotStream<Snapshot, Props> extends Assertable {

src/renderStream/createRenderStream.tsx

+7-10
Original file line numberDiff line numberDiff line change
@@ -221,15 +221,12 @@ export function createRenderStream<Snapshot extends ValidSnapshot = void>({
221221
return baseRender(ui, {
222222
...options,
223223
wrapper: (props) => {
224-
let elem: React.ReactNode = React.createElement(
225-
Wrapper,
226-
undefined,
227-
props.children
224+
const ParentWrapper = options?.wrapper || React.Fragment;
225+
return (
226+
<ParentWrapper>
227+
<Wrapper>{props.children}</Wrapper>
228+
</ParentWrapper>
228229
);
229-
if (options?.wrapper) {
230-
elem = React.createElement(options.wrapper, undefined, elem);
231-
}
232-
return elem;
233230
},
234231
});
235232
}) as typeof baseRender;
@@ -358,15 +355,15 @@ export function useTrackRenders({ name }: { name?: string } = {}) {
358355

359356
if (!component) {
360357
throw new Error(
361-
"useTrackRender: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option."
358+
"useTrackRenders: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option."
362359
);
363360
}
364361

365362
const ctx = useRenderStreamContext();
366363

367364
if (!ctx) {
368365
throw new Error(
369-
"useTrackComponentRender: A Render Stream must be created and rendered to track component renders"
366+
"useTrackRenders: A Render Stream must be created and rendered to track component renders"
370367
);
371368
}
372369

0 commit comments

Comments
 (0)