Infobar system refactor.

chrome/test/ppapi/ppapi_test.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 "chrome/test/ppapi/ppapi_test.h"
 
 #include "base/command_line.h"
 #include "base/file_util.h"
-#include "base/logging.h"
 #include "base/path_service.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
-#include "base/test/test_timeouts.h"
-#include "build/build_config.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/infobars/confirm_infobar_delegate.h"
 #include "chrome/browser/infobars/infobar.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
-#include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/test_switches.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/dom_operation_notification_details.h"
 #include "content/public/browser/notification_service.h"
-#include "content/public/browser/notification_types.h"
 #include "content/public/browser/web_contents.h"
-#include "content/public/common/content_paths.h"
-#include "content/public/common/content_switches.h"
-#include "content/public/test/browser_test_utils.h"
-#include "content/public/test/test_renderer_host.h"
 #include "net/base/net_util.h"
 #include "net/base/test_data_directory.h"
 #include "ppapi/shared_impl/ppapi_switches.h"
 #include "ui/gl/gl_switches.h"
 
 using content::DomOperationNotificationDetails;
 using content::RenderViewHost;
 
 namespace {
 
@@ skipping 21 matching lines @@
    message_ = trimmed;
    return DONE;
  }
 }
 
 void PPAPITestMessageHandler::Reset() {
   TestMessageHandler::Reset();
   message_.clear();
 }
 
-PPAPITestBase::InfoBarObserver::InfoBarObserver() {
+PPAPITestBase::InfoBarObserver::InfoBarObserver(PPAPITestBase* test_base)
+    : test_base_(test_base),
+      expecting_infobar_(false),
+      should_accept_(false) {
   registrar_.Add(this, chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED,
                  content::NotificationService::AllSources());
 }
 
 PPAPITestBase::InfoBarObserver::~InfoBarObserver() {
-  EXPECT_EQ(0u, expected_infobars_.size()) << "Missing an expected infobar";
+  EXPECT_FALSE(expecting_infobar_) << "Missing an expected infobar";
+}
+
+void PPAPITestBase::InfoBarObserver::ExpectInfoBarAndAccept(
+    bool should_accept) {
+  ASSERT_FALSE(expecting_infobar_);
+  expecting_infobar_ = true;
+  should_accept_ = should_accept;
 }
 
 void PPAPITestBase::InfoBarObserver::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   ASSERT_EQ(chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED, type);
-  InfoBarDelegate* infobar =
-      content::Details<InfoBarAddedDetails>(details).ptr();
-  ConfirmInfoBarDelegate* confirm_infobar_delegate =
-      infobar->AsConfirmInfoBarDelegate();
-  ASSERT_TRUE(confirm_infobar_delegate);
-
-  ASSERT_FALSE(expected_infobars_.empty()) << "Unexpected infobar";
-  if (expected_infobars_.front())
-    confirm_infobar_delegate->Accept();
-  else
-    confirm_infobar_delegate->Cancel();
-  expected_infobars_.pop_front();
-
-  // TODO(bauerb): We should close the infobar.
+  // It's not safe to remove the infobar here, since other observers (e.g. the
+  // InfoBarContainer) may still need to access it.  Instead, post a task to
+  // do all necessary infobar manipulation as soon as this call stack returns.
+  base::MessageLoop::current()->PostTask(
+      FROM_HERE, base::Bind(&InfoBarObserver::VerifyInfoBarState,
+                            base::Unretained(this)));
 }
 
-void PPAPITestBase::InfoBarObserver::ExpectInfoBarAndAccept(
-    bool should_accept) {
-  expected_infobars_.push_back(should_accept);
+void PPAPITestBase::InfoBarObserver::VerifyInfoBarState() {
+  content::WebContents* web_contents =
+      test_base_->browser()->tab_strip_model()->GetActiveWebContents();
+  ASSERT_TRUE(web_contents != NULL);
+  InfoBarService* infobar_service =
+      InfoBarService::FromWebContents(web_contents);
+  ASSERT_TRUE(infobar_service != NULL);
+
+  EXPECT_EQ(expecting_infobar_ ? 1U : 0U, infobar_service->infobar_count());
+  if (!expecting_infobar_)
+    return;
+  expecting_infobar_ = false;
+
+  InfoBar* infobar = infobar_service->infobar_at(0);
+  ConfirmInfoBarDelegate* delegate =
+      infobar->delegate()->AsConfirmInfoBarDelegate();
+  ASSERT_TRUE(delegate != NULL);
+  if (should_accept_)
+    delegate->Accept();
+  else
+    delegate->Cancel();
+
+  infobar_service->RemoveInfoBar(infobar);
 }
 
 PPAPITestBase::PPAPITestBase() {
 }
 
 void PPAPITestBase::SetUpCommandLine(CommandLine* command_line) {
   // The test sends us the result via a cookie.
   command_line->AppendSwitch(switches::kEnableFileCookies);
 
   // Some stuff is hung off of the testing interface which is not enabled
@@ skipping 251 matching lines @@
     const std::string& base,
     const std::string& test_case) {
   return base::StringPrintf("%smode=nacl_newlib&testcase=%s", base.c_str(),
                             test_case.c_str());
 }
 
 void PPAPIBrokerInfoBarTest::SetUpOnMainThread() {
   // The default content setting for the PPAPI broker is ASK. We purposefully
   // don't call PPAPITestBase::SetUpOnMainThread() to keep it that way.
 }