Iterate over listeners when sending process died notification.

chrome/browser/renderer_host/render_process_host_chrome_browsertest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/command_line.h"
+#include "base/process_util.h"
 #include "chrome/browser/devtools/devtools_window.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/singleton_tabs.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/browser/web_contents_observer.h"
 
 using content::RenderViewHost;
 using content::RenderWidgetHost;
 using content::WebContents;
 
 namespace {
 
 int RenderProcessHostCount() {
   content::RenderProcessHost::iterator hosts =
       content::RenderProcessHost::AllHostsIterator();
@@ skipping 316 matching lines @@
   RenderViewHost* devtools = FindFirstDevToolsHost();
   DCHECK(devtools);
 
   // DevTools start in a separate process.
   DevToolsWindow::ToggleDevToolsWindow(
       devtools, true, DEVTOOLS_TOGGLE_ACTION_INSPECT);
   host_count++;
   EXPECT_EQ(tab_count, browser()->tab_strip_model()->count());
   EXPECT_EQ(host_count, RenderProcessHostCount());
 }
+

[Charlie Reis, 2013/07/01 17:16:42]

nit: Remove extra blank line.

[nasko, 2013/07/01 23:08:42]

Done.

+
+// This class's goal is to close the browser window when a renderer process has
+// crashed. It does so by monitoring WebContents for RenderViewGone event and
+// closing the passed in TabStripModel. This is used in the following test case.
+class WindowDestroyer : public content::WebContentsObserver {
+ public:
+  WindowDestroyer(content::WebContents* web_contents, TabStripModel* model)
+      : content::WebContentsObserver(web_contents),
+        tab_strip_model_(model) {
+  }
+
+  virtual void RenderViewGone(base::TerminationStatus status) OVERRIDE {
+    LOG(ERROR) << "WD::RenderViewGone";

[Charlie Reis, 2013/07/01 17:16:42]

Remove log statements?

[nasko, 2013/07/01 23:08:42]

Done.

+    content::WindowedNotificationObserver observer(
+        chrome::NOTIFICATION_WINDOW_CLOSED,
+        content::NotificationService::AllSources());
+    LOG(ERROR) << "WD::RenderViewGone: observer created.";
+    tab_strip_model_->CloseAllTabs();
+    LOG(ERROR) << "WD::RenderViewGone: Closed all tabs, waiting for window.";
+    observer.Wait();
+    LOG(ERROR) << "WD::RenderViewGone: Closed window.";
+  }
+
+ private:
+  TabStripModel* tab_strip_model_;
+};
+
+// Test to ensure that while iterating through all listeners in
+// RenderProcessHost and they are invalidated, we remove them properly and don't

[Charlie Reis, 2013/07/01 17:16:42]

nit: and they are invalidated -> and invalidating them
(I think?)

[nasko, 2013/07/01 23:08:42]

Done.

+// access already freed objects. See http://crbug.com/255524

[Charlie Reis, 2013/07/01 17:16:42]

nit: End with a period.

[nasko, 2013/07/01 23:08:42]

Done.

+IN_PROC_BROWSER_TEST_F(ChromeRenderProcessHostTest,
+                       CloseAllTabsDuringProcessDied) {
+  GURL url(chrome::kChromeUINewTabURL);
+
+  ui_test_utils::NavigateToURL(browser(), url);
+  ui_test_utils::NavigateToURLWithDisposition(
+      browser(), url, NEW_BACKGROUND_TAB,
+      ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
+
+  EXPECT_EQ(2, browser()->tab_strip_model()->count());
+
+  WebContents* wc1 = browser()->tab_strip_model()->GetWebContentsAt(0);
+  WebContents* wc2 = browser()->tab_strip_model()->GetWebContentsAt(1);
+  EXPECT_EQ(wc1->GetRenderProcessHost(), wc2->GetRenderProcessHost());
+
+  // Create an object that will close the window on a process crash.
+  WindowDestroyer destroyer(wc1, browser()->tab_strip_model());
+
+  content::WindowedNotificationObserver observer(
+      chrome::NOTIFICATION_WINDOW_CLOSED,
+      content::NotificationService::AllSources());
+
+  // Kill the renderer process, simulating a crash. This should the ProcessDied
+  // method to be called. Alternatively, RenderProcessHost::OnChannelError can
+  // be called to directly force a call to ProcessDied.
+  base::KillProcess(wc1->GetRenderProcessHost()->GetHandle(), -1, true);
+
+  observer.Wait();
+}