Refactor

mirsujat · Oct 8, 2019 · 8ef57df · 8ef57df
1 parent d4c0191
commit 8ef57df
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/react-context-api/src/components/cart-dropdown/cart-dropdown.component.jsx b/react-context-api/src/components/cart-dropdown/cart-dropdown.component.jsx
@@ -1,5 +1,4 @@
 import React from "react";
-import { withRouter } from "react-router-dom";
 
 import CustomButton from "../custom-button/custom-button.component";
 import CartItem from "../cart-item/cart-item.component";
@@ -28,4 +27,4 @@ const CartDropdown = ({ cartItems, history, toggleCartHidden }) => (
   </div>
 );
 
-export default withRouter(CartDropdown);
+export default CartDropdown;
diff --git a/react-context-api/src/containers/CartDropdownContainer/CartDropdownContainer.js b/react-context-api/src/containers/CartDropdownContainer/CartDropdownContainer.js
@@ -1,4 +1,5 @@
 import React from "react";
+import { withRouter } from "react-router-dom";
 import { connect } from "react-redux";
 import { createStructuredSelector } from "reselect";
 
@@ -7,9 +8,10 @@ import { toggleCartHidden } from "../../redux/cart/cart.actions.js";
 
 import CartDropdownComponent from "../../components/cart-dropdown/cart-dropdown.component";
 
-const CartDropdownContainer = ({ cartItems, toggleCartHidden }) => (
+const CartDropdownContainer = ({ cartItems, history, toggleCartHidden }) => (
   <CartDropdownComponent
     cartItems={cartItems}
+    history={history}
     toggleCartHidden={toggleCartHidden}
   ></CartDropdownComponent>
 );
@@ -22,7 +24,9 @@ const mapDispatchToProps = dispatch => ({
   toggleCartHidden: () => dispatch(toggleCartHidden())
 });
 
-export default connect(
-  mapStateToProps,
-  mapDispatchToProps
-)(CartDropdownContainer);
+export default withRouter(
+  connect(
+    mapStateToProps,
+    mapDispatchToProps
+  )(CartDropdownContainer)
+);