[Web Payments] Add PopMany and Size functions to ViewStack

chrome/browser/ui/views/payments/view_stack_unittest.cc

 // Copyright 2016 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 "base/memory/ptr_util.h"
 #include "base/observer_list.h"
 #include "base/run_loop.h"
 #include "chrome/browser/ui/views/payments/view_stack.h"
 #include "ui/gfx/animation/test_animation_delegate.h"
 #include "ui/views/test/views_test_base.h"
@@ skipping 46 matching lines @@
     target.set_origin(gfx::Point(0, 0));
     EXPECT_EQ(target, view->bounds());
   }
 
   void AssertViewCompletelyNextToStack(views::View* view) {
     gfx::Rect target = view_stack_->bounds();
     target.set_origin(gfx::Point(view_stack_->width(), 0));
     EXPECT_EQ(target, view->bounds());
   }
 
+  // Pushes a view on the stack, waits for its animation to be over, then
+  // returns a pointer to the pushed view.
+  views::View* PushViewOnStackAndWait() {
+    std::unique_ptr<TestStackView> view = base::MakeUnique<TestStackView>();
+    views::View* view_ptr = view.get();
+
+    view_stack_->Push(std::move(view), true);
+    EXPECT_TRUE(view_stack_->slide_in_animator_->IsAnimating());
+    view_stack_->slide_in_animator_->SetAnimationDelegate(
+        view_ptr, std::unique_ptr<gfx::AnimationDelegate>(
+                      new gfx::TestAnimationDelegate()));
+
+    base::RunLoop().Run();
+    EXPECT_FALSE(view_stack_->slide_in_animator_->IsAnimating());
+    return view_ptr;
+  }
+
+  // Pops |n| views from the stack, then waits for |top_view_ptr|'s animation to
+  // be over.
+  void PopManyAndWait(int n, views::View* top_view_ptr) {
+    view_stack_->PopMany(n);
+
+    EXPECT_TRUE(view_stack_->slide_out_animator_->IsAnimating());
+    view_stack_->slide_out_animator_->SetAnimationDelegate(
+        top_view_ptr, std::unique_ptr<gfx::AnimationDelegate>(
+                          new gfx::TestAnimationDelegate()));
+
+    base::RunLoop().Run();
+    EXPECT_FALSE(view_stack_->slide_out_animator_->IsAnimating());
+  }
+
   std::unique_ptr<ViewStack> view_stack_;
 
   DISALLOW_COPY_AND_ASSIGN(ViewStackTest);
 };
 
 TEST_F(ViewStackTest, TestInitialStateAddedAsChildView) {
   EXPECT_EQ(1, view_stack_->child_count());
   // This child was added without any animation so it's on top of its parent
   // already.
   AssertViewOnTopOfStack(view_stack_->top());
@@ skipping 89 matching lines @@
   view_stack_->slide_out_animator_->SetAnimationDelegate(
       view_ptr,
       std::unique_ptr<gfx::AnimationDelegate>(
           new gfx::TestAnimationDelegate()));
 
   base::RunLoop().Run();
   EXPECT_FALSE(view_stack_->slide_out_animator_->IsAnimating());
 
   ASSERT_TRUE(observer.view_deleted());
 }
+
+TEST_F(ViewStackTest, TestPopMany) {
+  views::View* top = PushViewOnStackAndWait();
+  EXPECT_EQ(2U, view_stack_->size());
+
+  PopManyAndWait(1, top);
+  EXPECT_EQ(1U, view_stack_->size());
+
+  top = PushViewOnStackAndWait();
+  top = PushViewOnStackAndWait();
+  top = PushViewOnStackAndWait();
+  EXPECT_EQ(4U, view_stack_->size());
+
+  PopManyAndWait(3, top);
+  EXPECT_EQ(1U, view_stack_->size());
+
+  top = PushViewOnStackAndWait();
+  top = PushViewOnStackAndWait();
+  EXPECT_EQ(3U, view_stack_->size());
+}