Third patch in getting rid of caching MessageLoop pointers and always using C...

chrome/browser/bookmarks/bookmark_drag_data_unittest.cc

 // Copyright (c) 2006-2008 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 "app/os_exchange_data.h"
 #include "app/os_exchange_data_provider_win.h"
+#include "base/message_loop.h"
 #include "base/scoped_ptr.h"
 #include "chrome/browser/bookmarks/bookmark_drag_data.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
+#include "chrome/browser/chrome_thread.h"
 #include "chrome/test/testing_profile.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
-typedef testing::Test BookmarkDragDataTest;
+class BookmarkDragDataTest : public testing::Test {
+ public:
+  BookmarkDragDataTest()
+      : ui_thread_(ChromeThread::UI, &loop_),
+        file_thread_(ChromeThread::FILE, &loop_) { }
+
+ private:
+  MessageLoop loop_;
+  ChromeThread ui_thread_;
+  ChromeThread file_thread_;
+};
 
 namespace {
 
 OSExchangeData::Provider* CloneProvider(const OSExchangeData& data) {
   return new OSExchangeDataProviderWin(
       OSExchangeDataProviderWin::GetIDataObject(data));
 }
 
 }  // namespace
 
@@ skipping 27 matching lines @@
   EXPECT_TRUE(drag_data.elements[0].is_url);
   EXPECT_TRUE(drag_data.elements[0].url == url);
   EXPECT_TRUE(drag_data.elements[0].title == title);
   EXPECT_EQ(0, drag_data.elements[0].children.size());
 }
 
 TEST_F(BookmarkDragDataTest, URL) {
   // Write a single node representing a URL to the clipboard.
   TestingProfile profile;
   profile.CreateBookmarkModel(false);
+  profile.BlockUntilBookmarkModelLoaded();
   profile.SetID(L"id");
   BookmarkModel* model = profile.GetBookmarkModel();
   const BookmarkNode* root = model->GetBookmarkBarNode();
   GURL url(GURL("http://foo.com"));
   const std::wstring title(L"blah");
   const BookmarkNode* node = model->AddURL(root, 0, title, url);
   BookmarkDragData drag_data(node);
   EXPECT_TRUE(drag_data.is_valid());
   ASSERT_EQ(1, drag_data.elements.size());
   EXPECT_TRUE(drag_data.elements[0].is_url);
@@ skipping 22 matching lines @@
   std::wstring read_title;
   EXPECT_TRUE(data2.GetURLAndTitle(&read_url, &read_title));
   EXPECT_TRUE(read_url == url);
   EXPECT_EQ(title, read_title);
 }
 
 // Tests writing a group to the clipboard.
 TEST_F(BookmarkDragDataTest, Group) {
   TestingProfile profile;
   profile.CreateBookmarkModel(false);
+  profile.BlockUntilBookmarkModelLoaded();
   profile.SetID(L"id");
   BookmarkModel* model = profile.GetBookmarkModel();
   const BookmarkNode* root = model->GetBookmarkBarNode();
   const BookmarkNode* g1 = model->AddGroup(root, 0, L"g1");
   const BookmarkNode* g11 = model->AddGroup(g1, 0, L"g11");
   const BookmarkNode* g12 = model->AddGroup(g1, 0, L"g12");
 
   BookmarkDragData drag_data(g12);
   EXPECT_TRUE(drag_data.is_valid());
   ASSERT_EQ(1, drag_data.elements.size());
@@ skipping 19 matching lines @@
   // A different profile should return NULL for the node.
   TestingProfile profile2(1);
   EXPECT_TRUE(read_data.GetFirstNode(&profile2) == NULL);
 }
 
 // Tests reading/writing a folder with children.
 TEST_F(BookmarkDragDataTest, GroupWithChild) {
   TestingProfile profile;
   profile.SetID(L"id");
   profile.CreateBookmarkModel(false);
+  profile.BlockUntilBookmarkModelLoaded();
   BookmarkModel* model = profile.GetBookmarkModel();
   const BookmarkNode* root = model->GetBookmarkBarNode();
   const BookmarkNode* group = model->AddGroup(root, 0, L"g1");
 
   GURL url(GURL("http://foo.com"));
   const std::wstring title(L"blah2");
 
   model->AddURL(group, 0, title, url);
 
   BookmarkDragData drag_data(group);
@@ skipping 18 matching lines @@
   // And make sure we get the node back.
   const BookmarkNode* r_group = read_data.GetFirstNode(&profile);
   EXPECT_TRUE(group == r_group);
 }
 
 // Tests reading/writing of multiple nodes.
 TEST_F(BookmarkDragDataTest, MultipleNodes) {
   TestingProfile profile;
   profile.SetID(L"id");
   profile.CreateBookmarkModel(false);
+  profile.BlockUntilBookmarkModelLoaded();
   BookmarkModel* model = profile.GetBookmarkModel();
   const BookmarkNode* root = model->GetBookmarkBarNode();
   const BookmarkNode* group = model->AddGroup(root, 0, L"g1");
 
   GURL url(GURL("http://foo.com"));
   const std::wstring title(L"blah2");
 
   const BookmarkNode* url_node = model->AddURL(group, 0, title, url);
 
   // Write the nodes to the clipboard.
@@ skipping 25 matching lines @@
   // And make sure we get the node back.
   std::vector<const BookmarkNode*> read_nodes = read_data.GetNodes(&profile);
   ASSERT_EQ(2, read_nodes.size());
   EXPECT_TRUE(read_nodes[0] == group);
   EXPECT_TRUE(read_nodes[1] == url_node);
 
   // Asking for the first node should return NULL with more than one element
   // present.
   EXPECT_TRUE(read_data.GetFirstNode(&profile) == NULL);
 }