| // 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. |
| |
| #ifndef CHROME_INSTALLER_UTIL_WORK_ITEM_MOCKS_H_ |
| #define CHROME_INSTALLER_UTIL_WORK_ITEM_MOCKS_H_ |
| |
| #include "base/macros.h" |
| #include "chrome/installer/util/conditional_work_item_list.h" |
| #include "chrome/installer/util/work_item.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| class MockWorkItem : public WorkItem { |
| public: |
| MockWorkItem(); |
| ~MockWorkItem(); |
| |
| MOCK_METHOD0(DoImpl, bool()); |
| MOCK_METHOD0(RollbackImpl, void()); |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockWorkItem); |
| }; |
| |
| class MockCondition : public WorkItem::Condition { |
| public: |
| MockCondition(); |
| ~MockCondition(); |
| |
| MOCK_CONST_METHOD0(ShouldRun, bool()); |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockCondition); |
| }; |
| |
| using StrictMockWorkItem = testing::StrictMock<MockWorkItem>; |
| using StrictMockCondition = testing::StrictMock<MockCondition>; |
| |
| #endif // CHROME_INSTALLER_UTIL_WORK_ITEM_MOCKS_H_ |