Index: ash/system/overview/overview_button_tray_unittest.cc |
diff --git a/ash/system/overview/overview_button_tray_unittest.cc b/ash/system/overview/overview_button_tray_unittest.cc |
index 45cd98af79c4b5eddf5d9d69e075c705e98d6271..33a7ae787134b15873ccd68d59dc36e2a3391ac9 100644 |
--- a/ash/system/overview/overview_button_tray_unittest.cc |
+++ b/ash/system/overview/overview_button_tray_unittest.cc |
@@ -5,6 +5,7 @@ |
#include "ash/system/overview/overview_button_tray.h" |
#include "ash/login_status.h" |
+#include "ash/public/cpp/config.h" |
#include "ash/public/cpp/shelf_types.h" |
#include "ash/root_window_controller.h" |
#include "ash/rotator/screen_rotation_animator.h" |
@@ -112,7 +113,7 @@ TEST_F(OverviewButtonTrayTest, PerformAction) { |
// Tests that tapping on the control will record the user action Tray_Overview. |
TEST_F(OverviewButtonTrayTest, TrayOverviewUserAction) { |
// TODO: investigate failure in mash, http://crbug.com/698129. |
- if (ShellPort::Get()->IsRunningInMash()) |
+ if (Shell::GetAshConfig() == Config::MASH) |
return; |
ASSERT_FALSE(Shell::Get()->window_selector_controller()->IsSelecting()); |
@@ -213,7 +214,7 @@ TEST_F(OverviewButtonTrayTest, ActiveStateOnlyDuringOverviewMode) { |
TEST_F(OverviewButtonTrayTest, HideAnimationAlwaysCompletes) { |
// TODO: disabled as ScreenRotationAnimator does not work in mash, |
// http://crbug.com/696754. |
- if (ShellPort::Get()->IsRunningInMash()) |
+ if (Shell::GetAshConfig() == Config::MASH) |
return; |
Shell::Get()->maximize_mode_controller()->EnableMaximizeModeWindowManager( |