Test cases for FolderHeaderView.

ui/app_list/test/app_list_test_model.cc

 // Copyright 2013 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 "ui/app_list/test/app_list_test_model.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/stringprintf.h"
 
 namespace app_list {
@@ skipping 51 matching lines @@
 std::string AppListTestModel::GetItemName(int id) {
   return base::StringPrintf("Item %d", id);
 }
 
 void AppListTestModel::PopulateApps(int n) {
   int start_index = static_cast<int>(top_level_item_list()->item_count());
   for (int i = 0; i < n; ++i)
     CreateAndAddItem(GetItemName(start_index + i));
 }
 
-void AppListTestModel::CreateAndPopulateFolderWithApps(int n) {
+AppListFolderItem* AppListTestModel::CreateAndPopulateFolderWithApps(int n) {
   DCHECK_GT(n, 1);
   int start_index = static_cast<int>(top_level_item_list()->item_count());
   AppListTestItem* item = CreateAndAddItem(GetItemName(start_index));
   std::string merged_item_id = item->id();
   for (int i = 1; i < n; ++i) {
     AppListTestItem* new_item = CreateAndAddItem(GetItemName(start_index + i));
     merged_item_id = AppListModel::MergeItems(merged_item_id, new_item->id());
   }
+  AppListItem* merged_item = FindItem(merged_item_id);
+  DCHECK(merged_item->GetItemType() == AppListFolderItem::kItemType);
+  return static_cast<AppListFolderItem*>(merged_item);
+}
+
+AppListFolderItem* AppListTestModel::CreateAndAddOemFolder(
+    const std::string& id) {
+  AppListFolderItem* folder =
+      new AppListFolderItem(id, AppListFolderItem::FOLDER_TYPE_OEM);
+  return static_cast<AppListFolderItem*>(AddItem(folder));
 }
 
 void AppListTestModel::PopulateAppWithId(int id) {
   CreateAndAddItem(GetItemName(id));
 }
 
 std::string AppListTestModel::GetModelContent() {
   std::string content;
   for (size_t i = 0; i < top_level_item_list()->item_count(); ++i) {
     if (i > 0)
@@ skipping 30 matching lines @@
   item->SetHighlighted(true);
 }
 
 void AppListTestModel::ItemActivated(AppListTestItem* item) {
   last_activated_ = item;
   ++activate_count_;
 }
 
 }  // namespace test
 }  // namespace app_list