Add some browser tests for net-internals

chrome/test/base/ui_test_utils.cc

 // Copyright (c) 2011 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/base/ui_test_utils.h"
 
 #include <vector>
 
 #include "base/callback.h"
 #include "base/command_line.h"
@@ skipping 882 matching lines @@
       (running_ && waiting_for_ == NotificationService::AllSources())) {
     seen_ = true;
     if (running_)
       MessageLoopForUI::current()->Quit();
   } else {
     sources_seen_.insert(source.map_key());
   }
 }
 
 TitleWatcher::TitleWatcher(TabContents* tab_contents,
-                           const string16& expected_title)
+                           const string16& expected_title,
+                           const string16& failure_title)
     : expected_tab_(tab_contents),
       expected_title_(expected_title),
-      title_observed_(false),
+      failure_title_(failure_title),
+      expected_title_observed_(false),
+      failure_title_observed_(false),
       quit_loop_on_observation_(false) {
   EXPECT_TRUE(tab_contents != NULL);
   notification_registrar_.Add(this,
                               content::NOTIFICATION_TAB_CONTENTS_TITLE_UPDATED,
                               Source<TabContents>(tab_contents));
 }
 
 TitleWatcher::~TitleWatcher() {
 }
 
 bool TitleWatcher::Wait() {
-  if (title_observed_)
+  if (expected_title_observed_)
     return true;
+  if (failure_title_observed_)
+    return false;
   quit_loop_on_observation_ = true;
   ui_test_utils::RunMessageLoop();
-  return title_observed_;
+  return expected_title_observed_ && !failure_title_observed_;
 }
 
 void TitleWatcher::Observe(int type,
                            const NotificationSource& source,
                            const NotificationDetails& details) {
   if (type != content::NOTIFICATION_TAB_CONTENTS_TITLE_UPDATED)
     return;
 
   TabContents* source_contents = Source<TabContents>(source).ptr();
   ASSERT_EQ(expected_tab_, source_contents);
-  if (source_contents->GetTitle() != expected_title_)
+  if (source_contents->GetTitle() == expected_title_) {
+    expected_title_observed_ = true;
+  } else if (failure_title_.length() > 0 &&
+             source_contents->GetTitle() == failure_title_) {
+    failure_title_observed_ = true;
+  } else {
     return;
+  }
 
-  title_observed_ = true;
   if (quit_loop_on_observation_)
     MessageLoopForUI::current()->Quit();
 }
 
 DOMMessageQueue::DOMMessageQueue() {
   registrar_.Add(this, chrome::NOTIFICATION_DOM_OPERATION_RESPONSE,
                  NotificationService::AllSources());
 }
 
 DOMMessageQueue::~DOMMessageQueue() {}
@@ skipping 96 matching lines @@
   return taker.TakeRenderWidgetSnapshot(rwh, page_size, page_size, bitmap);
 }
 
 bool TakeEntirePageSnapshot(RenderViewHost* rvh, SkBitmap* bitmap) {
   DCHECK(bitmap);
   SnapshotTaker taker;
   return taker.TakeEntirePageSnapshot(rvh, bitmap);
 }
 
 }  // namespace ui_test_utils