Wait for 'after_install' from widget before closing SuggestAppsDialog

chrome/test/data/chromeos/file_manager/cws_container_mock/main.js

Index: chrome/test/data/chromeos/file_manager/cws_container_mock/main.js
diff --git a/chrome/test/data/chromeos/file_manager/cws_container_mock/main.js b/chrome/test/data/chromeos/file_manager/cws_container_mock/main.js
index e4dd9f39e7015993d6509ee73acb4ef62ff2e474..4b2dfe1a2fa3397397901b722f7b61d1ba8ad481 100644
--- a/chrome/test/data/chromeos/file_manager/cws_container_mock/main.js
+++ b/chrome/test/data/chromeos/file_manager/cws_container_mock/main.js
@@ -18,7 +18,10 @@ var cwsContainerMock = {};
 
     switch (data['message']) {
       case 'initialize':
-        cwsContainerMock.onInitialize(source);
+        cwsContainerMock.onInitialize();
+        break;
+      case 'install_success':
+        cwsContainerMock.onInstallSuccess();
         break;
     };
   }
@@ -28,6 +31,14 @@ var cwsContainerMock = {};
       source.postMessage({message: 'widget_loaded'}, origin);
   };
 
+  cwsContainerMock.onInstallSuccess = function() {
+    if (source && origin)
+      source.postMessage({
+          message: 'after_install',
+          item_id: DUMMY_ITEM_ID
+      }, origin);
+  };
+
   cwsContainerMock.onInstallButton = function() {
     if (source && origin) {
       source.postMessage(