chrome/browser/undo/undo_manager_test.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/auto_reset.h"
   6 #include "chrome/browser/undo/undo_manager.h"
   7 #include "chrome/browser/undo/undo_operation.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9
  10 namespace {
  11
  12 class TestUndoOperation;
  13
  14 // TestUndoService -------------------------------------------------------------
  15
  16 class TestUndoService {
  17  public:
  18   TestUndoService();
  19   ~TestUndoService();
  20
  21   void Redo();
  22   void TriggerOperation();
  23   void RecordUndoCall();
  24
  25   UndoManager undo_manager_;
  26
  27   bool performing_redo_;
  28
  29   int undo_operation_count_;
  30   int redo_operation_count_;
  31 };
  32
  33 // TestUndoOperation -----------------------------------------------------------
  34
  35 class TestUndoOperation : public UndoOperation {
  36  public:
  37   explicit TestUndoOperation(TestUndoService* undo_service);
  38   virtual ~TestUndoOperation();
  39
  40   // UndoOperation:
  41   virtual void Undo() OVERRIDE;
  42   virtual int GetUndoLabelId() const OVERRIDE;
  43   virtual int GetRedoLabelId() const OVERRIDE;
  44
  45  private:
  46   TestUndoService* undo_service_;
  47
  48   DISALLOW_COPY_AND_ASSIGN(TestUndoOperation);
  49 };
  50
  51 TestUndoOperation::TestUndoOperation(TestUndoService* undo_service)
  52       : undo_service_(undo_service) {
  53 }
  54
  55 TestUndoOperation::~TestUndoOperation() {
  56 }
  57
  58 void TestUndoOperation::Undo() {
  59   undo_service_->TriggerOperation();
  60   undo_service_->RecordUndoCall();
  61 }
  62
  63 int TestUndoOperation::GetUndoLabelId() const {
  64   return 0;
  65 }
  66
  67 int TestUndoOperation::GetRedoLabelId() const {
  68   return 0;
  69 }
  70
  71 // TestUndoService -------------------------------------------------------------
  72
  73 TestUndoService::TestUndoService() : performing_redo_(false),
  74                                      undo_operation_count_(0),
  75                                      redo_operation_count_(0) {
  76 }
  77
  78 TestUndoService::~TestUndoService() {
  79 }
  80
  81 void TestUndoService::Redo() {
  82   base::AutoReset<bool> incoming_changes(&performing_redo_, true);
  83   undo_manager_.Redo();
  84 }
  85
  86 void TestUndoService::TriggerOperation() {
  87   scoped_ptr<TestUndoOperation> op(new TestUndoOperation(this));
  88   undo_manager_.AddUndoOperation(op.PassAs<UndoOperation>());
  89 }
  90
  91 void TestUndoService::RecordUndoCall() {
  92   if (performing_redo_)
  93     ++redo_operation_count_;
  94   else
  95     ++undo_operation_count_;
  96 }
  97
  98 // Tests -----------------------------------------------------------------------
  99
 100 TEST(UndoServiceTest, AddUndoActions) {
 101   TestUndoService undo_service;
 102
 103   undo_service.TriggerOperation();
 104   undo_service.TriggerOperation();
 105   EXPECT_EQ(2U, undo_service.undo_manager_.undo_count());
 106   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 107 }
 108
 109 TEST(UndoServiceTest, UndoMultipleActions) {
 110   TestUndoService undo_service;
 111
 112   undo_service.TriggerOperation();
 113   undo_service.TriggerOperation();
 114
 115   undo_service.undo_manager_.Undo();
 116   EXPECT_EQ(1U, undo_service.undo_manager_.undo_count());
 117   EXPECT_EQ(1U, undo_service.undo_manager_.redo_count());
 118
 119   undo_service.undo_manager_.Undo();
 120   EXPECT_EQ(0U, undo_service.undo_manager_.undo_count());
 121   EXPECT_EQ(2U, undo_service.undo_manager_.redo_count());
 122
 123   EXPECT_EQ(2, undo_service.undo_operation_count_);
 124   EXPECT_EQ(0, undo_service.redo_operation_count_);
 125 }
 126
 127 TEST(UndoServiceTest, RedoAction) {
 128   TestUndoService undo_service;
 129
 130   undo_service.TriggerOperation();
 131
 132   undo_service.undo_manager_.Undo();
 133   EXPECT_EQ(0U, undo_service.undo_manager_.undo_count());
 134   EXPECT_EQ(1U, undo_service.undo_manager_.redo_count());
 135
 136   undo_service.Redo();
 137   EXPECT_EQ(1U, undo_service.undo_manager_.undo_count());
 138   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 139
 140   EXPECT_EQ(1, undo_service.undo_operation_count_);
 141   EXPECT_EQ(1, undo_service.redo_operation_count_);
 142 }
 143
 144 TEST(UndoServiceTest, GroupActions) {
 145   TestUndoService undo_service;
 146
 147   // Add two operations in a single action.
 148   undo_service.undo_manager_.StartGroupingActions();
 149   undo_service.TriggerOperation();
 150   undo_service.TriggerOperation();
 151   undo_service.undo_manager_.EndGroupingActions();
 152
 153   // Check that only one action is created.
 154   EXPECT_EQ(1U, undo_service.undo_manager_.undo_count());
 155   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 156
 157   undo_service.undo_manager_.Undo();
 158   EXPECT_EQ(0U, undo_service.undo_manager_.undo_count());
 159   EXPECT_EQ(1U, undo_service.undo_manager_.redo_count());
 160
 161   undo_service.Redo();
 162   EXPECT_EQ(1U, undo_service.undo_manager_.undo_count());
 163   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 164
 165   // Check that both operations were called in Undo and Redo.
 166   EXPECT_EQ(2, undo_service.undo_operation_count_);
 167   EXPECT_EQ(2, undo_service.redo_operation_count_);
 168 }
 169
 170 TEST(UndoServiceTest, SuspendUndoTracking) {
 171   TestUndoService undo_service;
 172
 173   undo_service.undo_manager_.SuspendUndoTracking();
 174   EXPECT_TRUE(undo_service.undo_manager_.IsUndoTrakingSuspended());
 175
 176   undo_service.TriggerOperation();
 177
 178   undo_service.undo_manager_.ResumeUndoTracking();
 179   EXPECT_FALSE(undo_service.undo_manager_.IsUndoTrakingSuspended());
 180
 181   EXPECT_EQ(0U, undo_service.undo_manager_.undo_count());
 182   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 183 }
 184
 185 TEST(UndoServiceTest, RedoEmptyAfterNewAction) {
 186   TestUndoService undo_service;
 187
 188   undo_service.TriggerOperation();
 189   undo_service.undo_manager_.Undo();
 190   EXPECT_EQ(0U, undo_service.undo_manager_.undo_count());
 191   EXPECT_EQ(1U, undo_service.undo_manager_.redo_count());
 192
 193   undo_service.TriggerOperation();
 194   EXPECT_EQ(1U, undo_service.undo_manager_.undo_count());
 195   EXPECT_EQ(0U, undo_service.undo_manager_.redo_count());
 196 }
 197
 198 TEST(UndoServiceTest, GetAllUndoOperations) {
 199   TestUndoService undo_service;
 200
 201   undo_service.TriggerOperation();
 202
 203   undo_service.undo_manager_.StartGroupingActions();
 204   undo_service.TriggerOperation();
 205   undo_service.TriggerOperation();
 206   undo_service.undo_manager_.EndGroupingActions();
 207
 208   undo_service.TriggerOperation();
 209
 210   undo_service.undo_manager_.Undo();
 211   ASSERT_EQ(2U, undo_service.undo_manager_.undo_count());
 212   ASSERT_EQ(1U, undo_service.undo_manager_.redo_count());
 213
 214   std::vector<UndoOperation*> all_operations =
 215       undo_service.undo_manager_.GetAllUndoOperations();
 216   EXPECT_EQ(4U, all_operations.size());
 217 }
 218
 219 } // namespace