diff --git a/src/compose.js b/src/compose.js index 8cce3fa..fe5ff45 100644 --- a/src/compose.js +++ b/src/compose.js @@ -133,7 +133,7 @@ export default function compose(dataLoader, options = {}) { dataLoader, options, }; - inheritStatics(Container, Child); + inheritStatics(Container, Child, options.displayName); return mayBeStubbed(Container); }; } diff --git a/src/utils.js b/src/utils.js index 6da0d04..19c6135 100644 --- a/src/utils.js +++ b/src/utils.js @@ -2,7 +2,7 @@ import hoistStatics from 'hoist-non-react-statics'; -export function inheritStatics(Container, ChildComponent) { +export function inheritStatics(Container, ChildComponent, displayName = 'Container') { const childDisplayName = // Get the display name if it's set. ChildComponent.displayName || @@ -11,6 +11,6 @@ export function inheritStatics(Container, ChildComponent) { // If not, just add a default one. 'ChildComponent'; - Container.displayName = `Container(${childDisplayName})`; // eslint-disable-line + Container.displayName = `${displayName}(${childDisplayName})`; // eslint-disable-line return hoistStatics(Container, ChildComponent); }