ARC Toast: Prevent onClosed event from being called multiple times

ash/system/toast/toast_manager_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 "ash/display/display_manager.h"
 #include "ash/screen_util.h"
 #include "ash/shelf/shelf.h"
 #include "ash/shelf/shelf_layout_manager.h"
 #include "ash/shell.h"
 #include "ash/system/toast/toast_manager.h"
 #include "ash/test/ash_test_base.h"
 #include "base/run_loop.h"
+#include "ui/compositor/scoped_animation_duration_scale_mode.h"
+#include "ui/compositor/scoped_layer_animation_settings.h"
 
 namespace ash {
 
 // Long duration so the timeout doesn't occur.
 const int64_t kLongLongDuration = INT64_MAX;
 
 class DummyEvent : public ui::Event {
  public:
   DummyEvent() : Event(ui::ET_UNKNOWN, base::TimeDelta(), 0) {}
   ~DummyEvent() override {}
 };
 
 class ToastManagerTest : public test::AshTestBase {
  public:
   ToastManagerTest() {}
   ~ToastManagerTest() override {}
 
  private:
   void SetUp() override {
     test::AshTestBase::SetUp();
 
     manager_ = Shell::GetInstance()->toast_manager();
 
     manager_->ResetToastIdForTesting();
     EXPECT_EQ(0, GetToastId());
+    test_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
+        ui::ScopedAnimationDurationScaleMode::NON_ZERO_DURATION));
   }
 
  protected:
   ToastManager* manager() { return manager_; }
 
   int GetToastId() { return manager_->toast_id_for_testing(); }
 
   ToastOverlay* GetCurrentOverlay() {
     return manager_->GetCurrentOverlayForTesting();
   }
@@ skipping 23 matching lines @@
   }
 
   void SetShelfAutoHideBehavior(ShelfAutoHideBehavior behavior) {
     Shelf::ForPrimaryDisplay()->shelf_layout_manager()->SetAutoHideBehavior(
         behavior);
   }
 
  private:
   ToastManager* manager_ = nullptr;
 
+  scoped_ptr<ui::ScopedAnimationDurationScaleMode> test_duration_mode_;
+
   DISALLOW_COPY_AND_ASSIGN(ToastManagerTest);
 };
 
 TEST_F(ToastManagerTest, ShowAndCloseAutomatically) {
   manager()->Show("DUMMY", 10);
   base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(1, GetToastId());
 
   while (GetCurrentOverlay() != nullptr)
     base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(ToastManagerTest, ShowAndCloseManually) {
   manager()->Show("DUMMY", kLongLongDuration /* prevent timeout */);
   base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(1, GetToastId());
 
+  while (GetCurrentWidget()->GetLayer()->GetAnimator()->is_animating())
+    base::RunLoop().RunUntilIdle();
+
   ClickDismissButton();
 
   while (GetCurrentOverlay() != nullptr)
+    base::RunLoop().RunUntilIdle();
+}
+
+TEST_F(ToastManagerTest, ShowAndCloseManuallyDuringAnimation) {
+  ui::ScopedAnimationDurationScaleMode slow_animation_duration(
+      ui::ScopedAnimationDurationScaleMode::SLOW_DURATION);
+
+  manager()->Show("DUMMY", kLongLongDuration /* prevent timeout */);
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_EQ(1, GetToastId());
+  EXPECT_TRUE(GetCurrentWidget()->GetLayer()->GetAnimator()->is_animating());
+
+  // Close it during animation.
+  ClickDismissButton();
+
+  while (GetCurrentOverlay() != nullptr)
     base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(ToastManagerTest, QueueMessage) {
   manager()->Show("DUMMY1", 10);
   manager()->Show("DUMMY2", 10);
   manager()->Show("DUMMY3", 10);
 
   while (GetToastId() != 1)
     base::RunLoop().RunUntilIdle();
@@ skipping 142 matching lines @@
     // doesn't return correct value.
     gfx::Rect shelf_bounds = shelf->GetIdealBounds();
     EXPECT_FALSE(toast_bounds.Intersects(shelf_bounds));
     EXPECT_EQ(shelf_bounds.y() - 5, toast_bounds.bottom());
     EXPECT_EQ(root_bounds.bottom() - shelf_bounds.height() - 5,
               toast_bounds.bottom());
   }
 }
 
 }  // namespace ash