Change TaskManager test to be in line with new TabContents lifetime uncertainty when Prerendering.

chrome/browser/prerender/prerender_browsertest.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 <deque>
 
 #include "base/command_line.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "chrome/browser/prerender/prerender_contents.h"
@@ skipping 692 matching lines @@
   // Early out if we're not using TabContents for Prerendering.
   if (!prerender::PrerenderContents::UseTabContents()) {
     SUCCEED();
     return;
   }
 
   // Show the task manager. This populates the model.
   browser()->window()->ShowTaskManager();
 
   // Start with two resources.
-  EXPECT_EQ(2, model()->ResourceCount());
   PrerenderTestURL("files/prerender/prerender_page.html", FINAL_STATUS_USED, 1);
 
-  // The prerender makes three.
-  EXPECT_EQ(3, model()->ResourceCount());
+  // One of the resources that has a TabContents associated with it should have
+  // the Prerender prefix.
+  const string16 prefix =
+      l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_PRERENDER_PREFIX, string16());
+  string16 prerender_title;
+  int num_prerender_tabs = 0;
 
-  // It should have a TabContents associated with it.
-  ASSERT_TRUE(model()->GetResourceTabContents(1) != NULL);
-
-  // The prefix should be "Prerender:"
-  string16 prefix =
-      l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_PRERENDER_PREFIX,
-                                 string16());
-  ASSERT_TRUE(StartsWith(model()->GetResourceTitle(2), prefix, true));
+  VLOG(1) << "After prerender:";

[cbentzel, 2011/05/06 21:21:03]

Do you still need these statements? 

+  for (int i = 0; i < model()->ResourceCount(); ++i) {
+    VLOG(1) << "  " << i << ": " << model()->GetResourceTitle(i);
+    if (model()->GetResourceTabContents(i)) {
+      prerender_title = model()->GetResourceTitle(i);
+      if (StartsWith(prerender_title, prefix, true))
+        ++num_prerender_tabs;
+    }
+  }
+  ASSERT_EQ(1, num_prerender_tabs);
+  const string16 prerender_page_title = prerender_title.substr(prefix.length());
 
   NavigateToDestURL();
 
-  // Prerender task should become a normal tab.
-  EXPECT_EQ(3, model()->ResourceCount());
+  // There should be no tabs with the Prerender prefix.
+  const string16 tab_prefix =
+      l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_TAB_PREFIX, string16());
+  num_prerender_tabs = 0;
+  bool found_tab_with_prerender_page_title = false;
+  VLOG(1) << "After navigate:";
+  for (int i = 0; i < model()->ResourceCount(); ++i) {
+    VLOG(1) << "  " << i << ": " << model()->GetResourceTitle(i);
+    if (model()->GetResourceTabContents(i)) {
+      string16 tab_title = model()->GetResourceTitle(i);
+      if (StartsWith(tab_title, prefix, true)) {
+        ++num_prerender_tabs;
+      } else {
+        ASSERT_TRUE(StartsWith(tab_title, tab_prefix, true));

[cbentzel, 2011/05/06 21:21:03]

Nit: EXPECT_TRUE is probably appropriate here. ASSERT is usually used when the
test will crash, EXPECT otherwise. 

 
-  // It should have a TabContents associated with it.
-  ASSERT_TRUE(model()->GetResourceTabContents(2) != NULL);
-
-  // The prefix should now be "Tab:"
-  prefix = l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_TAB_PREFIX,
-                                      string16());
-  ASSERT_TRUE(StartsWith(model()->GetResourceTitle(2), prefix, true));
+        // The prerender tab should now be a normal tab but the title should be
+        // the same.
+        const string16 tab_page_title = tab_title.substr(tab_prefix.length());
+        if (prerender_page_title.compare(tab_page_title) == 0) {
+          ASSERT_FALSE(found_tab_with_prerender_page_title);
+          found_tab_with_prerender_page_title = true;
+        }
+      }
+    }
+  }
+  ASSERT_EQ(0, num_prerender_tabs);
+  ASSERT_TRUE(found_tab_with_prerender_page_title);
 }
 
 // Checks that prerenderers will terminate when an audio tag is encountered.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderHTML5Audio) {
   PrerenderTestURL("files/prerender/prerender_html5_audio.html",
                    FINAL_STATUS_HTML5_MEDIA,
                    1);
 }
 
 // Checks that prerenderers will terminate when a video tag is encountered.
@@ skipping 168 matching lines @@
   std::string replacement_path;
   ASSERT_TRUE(net::TestServer::GetFilePathWithReplacements(
       "files/prerender/prerender_with_image.html",
       replacement_text,
       &replacement_path));
   PrerenderTestURL(replacement_path, FINAL_STATUS_USED, 1);
   NavigateToDestURL();
 }
 
 }  // namespace prerender