Extract temp variables for wrapped components in their tests (#4279)

firefox-devtools · Oct 4, 2017 · 0004098 · 0004098
1 parent 6563978
commit 0004098
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/components/Editor/tests/Breakpoints.spec.js b/src/components/Editor/tests/Breakpoints.spec.js
@@ -3,6 +3,8 @@ import { shallow } from "enzyme";
 import Breakpoints from "../Breakpoints";
 import * as I from "immutable";
 
+const BreakpointsComponent = Breakpoints.WrappedComponent;
+
 function generateDefaults(overrides) {
   const sourceId = "server1.conn1.child1/source1";
   const matchingBreakpoints = { id1: { location: { sourceId } } };
@@ -21,7 +23,7 @@ function generateDefaults(overrides) {
 
 function render(overrides = {}) {
   const props = generateDefaults(overrides);
-  const component = shallow(<Breakpoints.WrappedComponent {...props} />);
+  const component = shallow(<BreakpointsComponent {...props} />);
   return { component, props };
 }
 

diff --git a/src/components/Editor/tests/SearchBar.spec.js b/src/components/Editor/tests/SearchBar.spec.js
@@ -4,6 +4,8 @@ import SearchBar from "../SearchBar";
 import "../../../workers/search";
 import "../../../utils/editor";
 
+const SearchBarComponent = SearchBar.WrappedComponent;
+
 jest.mock("../../../workers/search", () => ({
   getMatches: () => Promise.resolve(["result"])
 }));
@@ -37,7 +39,7 @@ function generateDefaults() {
 function render(overrides = {}) {
   const defaults = generateDefaults();
   const props = { ...defaults, ...overrides };
-  const component = shallow(<SearchBar.WrappedComponent {...props} />);
+  const component = shallow(<SearchBarComponent {...props} />);
   return { component, props };
 }