1 files changed, 40 insertions, 9 deletions
diff --git a/ui/views/view_model_utils_unittest.cc b/ui/views/view_model_utils_unittest.cc
index 289feee..92004b7 100644
--- a/ui/views/view_model_utils_unittest.cc
+++ b/ui/views/view_model_utils_unittest.cc
@@ -32,15 +32,46 @@ TEST(ViewModelUtils, DetermineMoveIndex) {
   model.set_ideal_bounds(1, gfx::Rect(10, 0, 1000, 10));
   model.set_ideal_bounds(2, gfx::Rect(1010, 0, 2, 10));
 
-  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(model, &v1, -10));
-  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(model, &v1, 4));
-  EXPECT_EQ(1, ViewModelUtils::DetermineMoveIndex(model, &v1, 506));
-  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(model, &v1, 1010));
-  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(model, &v1, 2000));
-
-  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(model, &v2, -10));
-  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(model, &v2, 4));
-  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(model, &v2, 12));
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::HORIZONTAL, -10, 0));
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::HORIZONTAL, 4, 0));
+  EXPECT_EQ(1, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::HORIZONTAL, 506, 0));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::HORIZONTAL, 1010, 0));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::HORIZONTAL, 2000, 0));
+
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::HORIZONTAL, -10, 0));
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::HORIZONTAL, 4, 0));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::HORIZONTAL, 12, 0));
+
+  // Try the same when vertical.
+  model.set_ideal_bounds(0, gfx::Rect(0, 0, 10, 10));
+  model.set_ideal_bounds(1, gfx::Rect(0, 10, 10, 1000));
+  model.set_ideal_bounds(2, gfx::Rect(0, 1010, 10, 2));
+
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::VERTICAL, 0, -10));
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::VERTICAL, 0, 4));
+  EXPECT_EQ(1, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::VERTICAL, 0, 506));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::VERTICAL, 0, 1010));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v1, ViewModelUtils::VERTICAL, 0, 2000));
+
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::VERTICAL, 0, -10));
+  EXPECT_EQ(0, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::VERTICAL, 0, 4));
+  EXPECT_EQ(2, ViewModelUtils::DetermineMoveIndex(
+                model, &v2, ViewModelUtils::VERTICAL, 0, 12));
 }
 
 }  // namespace views