show watch asset image from hide token modal

2 files changed, 14 insertions, 12 deletions

diff --git a/ui/app/components/identicon.js b/ui/app/components/identicon.js
index eadd6d5a2..6b632352f 100644
--- a/ui/app/components/identicon.js
+++ b/ui/app/components/identicon.js
@@ -28,14 +28,17 @@ IdenticonComponent.prototype.render = function () {
   var props = this.props
   const { className = '', address, imageUrl } = props
   var diameter = props.diameter || this.defaultDiameter
-  // for tokens added with `watchToken` we need to render the given image
+  const style = {
+    height: diameter,
+    width: diameter,
+    borderRadius: diameter / 2,
+  }
   if (imageUrl) {
-    return h('img.balance-icon', {
+    return h('img', {
+      className: `${className} identicon`,
       src: imageUrl,
       style: {
-        height: diameter,
-        width: diameter,
-        borderRadius: diameter / 2,
+        ...style,
       },
     })
   } else if (address) {
@@ -47,9 +50,7 @@ IdenticonComponent.prototype.render = function () {
         flexShrink: 0,
         alignItems: 'center',
         justifyContent: 'center',
-        height: diameter,
-        width: diameter,
-        borderRadius: diameter / 2,
+        ...style,
         overflow: 'hidden',
       },
     })
@@ -57,9 +58,7 @@ IdenticonComponent.prototype.render = function () {
     return h('img.balance-icon', {
       src: './images/eth_logo.svg',
       style: {
-        height: diameter,
-        width: diameter,
-        borderRadius: diameter / 2,
+        ...style,
       },
     })
   }
diff --git a/ui/app/components/modals/hide-token-confirmation-modal.js b/ui/app/components/modals/hide-token-confirmation-modal.js
index 1518fa9a0..b5f396e6f 100644
--- a/ui/app/components/modals/hide-token-confirmation-modal.js
+++ b/ui/app/components/modals/hide-token-confirmation-modal.js
@@ -10,6 +10,7 @@ function mapStateToProps (state) {
   return {
     network: state.metamask.network,
     token: state.appState.modal.modalState.props.token,
+    tokenImagesHashes: state.metamask.objects,    
   }
 }
 
@@ -40,8 +41,9 @@ module.exports = connect(mapStateToProps, mapDispatchToProps)(HideTokenConfirmat
 
 
 HideTokenConfirmationModal.prototype.render = function () {
-  const { token, network, hideToken, hideModal } = this.props
+  const { token, network, hideToken, hideModal, tokenImagesHashes } = this.props
   const { symbol, address } = token
+  const imageUrl = tokenImagesHashes[address]
 
   return h('div.hide-token-confirmation', {}, [
     h('div.hide-token-confirmation__container', {
@@ -55,6 +57,7 @@ HideTokenConfirmationModal.prototype.render = function () {
         diameter: 45,
         address,
         network,
+        imageUrl,
       }),
 
       h('div.hide-token-confirmation__symbol', {}, symbol),