#include "chrome/browser/ui/views/bookmarks/bookmark_bar_view.h"
#include "base/prefs/pref_service.h"
-#include "chrome/browser/bookmarks/bookmark_test_helpers.h"
+#include "chrome/browser/bookmarks/bookmark_model_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/search_engines/template_url_service.h"
#include "chrome/browser/search_engines/template_url_service_factory.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "chrome/test/base/scoped_testing_local_state.h"
#include "chrome/test/base/testing_browser_process.h"
+#include "components/bookmarks/core/test/bookmark_test_helpers.h"
#include "ui/views/controls/button/text_button.h"
class BookmarkBarViewInstantExtendedTest : public BrowserWithTestWindowTest {
TEST_F(BookmarkBarViewInstantExtendedTest, AppsShortcutVisibility) {
ScopedTestingLocalState local_state(TestingBrowserProcess::GetGlobal());
profile()->CreateBookmarkModel(true);
- test::WaitForBookmarkModelToLoad(profile());
+ test::WaitForBookmarkModelToLoad(
+ BookmarkModelFactory::GetForProfile(profile()));
BookmarkBarView bookmark_bar_view(browser(), NULL);
bookmark_bar_view.set_owned_by_client();
browser()->profile()->GetPrefs()->SetBoolean(