Index: ash/test/launcher_view_test_api.cc |
diff --git a/ash/test/launcher_view_test_api.cc b/ash/test/launcher_view_test_api.cc |
index f6a418a752e3353a187d2c8b8bcd605ecfe7f177..8d2885bb33e457543e09fe437244946bf20afd4d 100644 |
--- a/ash/test/launcher_view_test_api.cc |
+++ b/ash/test/launcher_view_test_api.cc |
@@ -5,6 +5,7 @@ |
#include "ash/test/launcher_view_test_api.h" |
#include "ash/launcher/launcher_button.h" |
+#include "ash/launcher/launcher_model.h" |
#include "ash/launcher/launcher_view.h" |
#include "ash/launcher/overflow_button.h" |
#include "base/message_loop.h" |
@@ -48,7 +49,7 @@ int LauncherViewTestAPI::GetButtonCount() { |
internal::LauncherButton* LauncherViewTestAPI::GetButton(int index) { |
// App list button is not a LauncherButton. |
- if (index == GetButtonCount() - 1) |
+ if (launcher_view_->model_->items()[index].type == ash::TYPE_APP_LIST) |
return NULL; |
return static_cast<internal::LauncherButton*>( |