Chromium Code Reviews| Index: chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk_unittest.cc |
| diff --git a/chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk_unittest.cc b/chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk_unittest.cc |
| index a48f5878125ff56d69bc0d4e7633611864373dda..765a94fcb6acb34e73f2a4cb940e399b349b8312 100644 |
| --- a/chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk_unittest.cc |
| +++ b/chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk_unittest.cc |
| @@ -4,6 +4,7 @@ |
| #include "chrome/browser/ui/gtk/bookmarks/bookmark_bar_gtk.h" |
| +#include "base/compiler_specific.h" |
| #include "base/task.h" |
| #include "base/utf_string_conversions.h" |
| #include "chrome/browser/bookmarks/bookmark_model.h" |
| @@ -31,18 +32,20 @@ class BookmarkBarGtkUnittest : public testing::Test { |
| file_thread_(BrowserThread::FILE, &message_loop_) { |
| } |
| - virtual void SetUp() { |
| + virtual void SetUp() OVERRIDE { |
| profile_.reset(new TestingProfile()); |
| profile_->CreateBookmarkModel(true); |
| profile_->BlockUntilBookmarkModelLoaded(); |
| browser_.reset(new Browser(Browser::TYPE_TABBED, profile_.get())); |
| + model_ = profile_->GetBookmarkModel(); |
| + |
| origin_provider_.reset(new EmptyTabstripOriginProvider); |
| bookmark_bar_.reset(new BookmarkBarGtk(NULL, browser_.get(), |
| origin_provider_.get())); |
| } |
| - virtual void TearDown() { |
| + virtual void TearDown() OVERRIDE { |
| message_loop_.RunAllPending(); |
| bookmark_bar_.reset(); |
| @@ -51,6 +54,11 @@ class BookmarkBarGtkUnittest : public testing::Test { |
| profile_.reset(); |
| } |
| + protected: |
| + BookmarkModel* model_; |
| + scoped_ptr<BookmarkBarGtk> bookmark_bar_; |
|
Elliot Glaysher
2011/11/15 17:55:59
If you're going to make this protected, violate th
tfarina
2011/11/15 20:56:26
Done.
|
| + |
| + private: |
| MessageLoopForUI message_loop_; |
| content::TestBrowserThread ui_thread_; |
| content::TestBrowserThread file_thread_; |
| @@ -58,12 +66,10 @@ class BookmarkBarGtkUnittest : public testing::Test { |
| scoped_ptr<TestingProfile> profile_; |
| scoped_ptr<Browser> browser_; |
| scoped_ptr<TabstripOriginProvider> origin_provider_; |
| - scoped_ptr<BookmarkBarGtk> bookmark_bar_; |
| }; |
| TEST_F(BookmarkBarGtkUnittest, DisplaysHelpMessageOnEmpty) { |
| - BookmarkModel* model = profile_->GetBookmarkModel(); |
| - bookmark_bar_->Loaded(model, false); |
| + bookmark_bar_->Loaded(model_, false); |
| // There are no bookmarks in the model by default. Expect that the |
| // |instructions_label| is shown. |
| @@ -71,26 +77,22 @@ TEST_F(BookmarkBarGtkUnittest, DisplaysHelpMessageOnEmpty) { |
| } |
| TEST_F(BookmarkBarGtkUnittest, HidesHelpMessageWithBookmark) { |
| - BookmarkModel* model = profile_->GetBookmarkModel(); |
| - |
| - const BookmarkNode* parent = model->bookmark_bar_node(); |
| - model->AddURL(parent, parent->child_count(), |
| - ASCIIToUTF16("title"), GURL("http://one.com")); |
| + const BookmarkNode* parent = model_->bookmark_bar_node(); |
| + model_->AddURL(parent, parent->child_count(), |
| + ASCIIToUTF16("title"), GURL("http://one.com")); |
| - bookmark_bar_->Loaded(model, false); |
| + bookmark_bar_->Loaded(model_, false); |
| EXPECT_FALSE(bookmark_bar_->show_instructions_); |
| } |
| TEST_F(BookmarkBarGtkUnittest, BuildsButtons) { |
| - BookmarkModel* model = profile_->GetBookmarkModel(); |
| - |
| - const BookmarkNode* parent = model->bookmark_bar_node(); |
| - model->AddURL(parent, parent->child_count(), |
| - ASCIIToUTF16("title"), GURL("http://one.com")); |
| - model->AddURL(parent, parent->child_count(), |
| - ASCIIToUTF16("other"), GURL("http://two.com")); |
| + const BookmarkNode* parent = model_->bookmark_bar_node(); |
| + model_->AddURL(parent, parent->child_count(), |
| + ASCIIToUTF16("title"), GURL("http://one.com")); |
| + model_->AddURL(parent, parent->child_count(), |
| + ASCIIToUTF16("other"), GURL("http://two.com")); |
| - bookmark_bar_->Loaded(model, false); |
| + bookmark_bar_->Loaded(model_, false); |
| // We should expect two children to the bookmark bar's toolbar. |
| GList* children = gtk_container_get_children( |