| // Copyright 2018 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_CHROME_CLEANER_IPC_MOCK_CHROME_PROMPT_IPC_H_ |
| #define CHROME_CHROME_CLEANER_IPC_MOCK_CHROME_PROMPT_IPC_H_ |
| |
| #include <memory> |
| #include <vector> |
| |
| #include "chrome/chrome_cleaner/ipc/chrome_prompt_ipc.h" |
| #include "components/chrome_cleaner/public/interfaces/chrome_prompt.mojom.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| namespace chrome_cleaner { |
| |
| class MockChromePromptIPC : public ChromePromptIPC { |
| public: |
| MockChromePromptIPC(); |
| ~MockChromePromptIPC() override; |
| |
| MOCK_METHOD1(Initialize, void(ErrorHandler* error_handler)); |
| |
| MOCK_METHOD2(TryDeleteExtensions, |
| void(base::OnceClosure delete_allowed_callback, |
| base::OnceClosure delete_not_allowed_callback)); |
| |
| // Workaround for GMock's limitation, in which MOCK_METHOD* doesn't accept |
| // base::OnceCallback parameters. Will forward any calls to |
| // MockPostPromptUserTask() and pass along a raw pointer for |callback|. |
| void PostPromptUserTask( |
| const std::vector<base::FilePath>& files_to_delete, |
| const std::vector<base::string16>& registry_keys, |
| const std::vector<base::string16>& extension_ids, |
| mojom::ChromePrompt::PromptUserCallback callback) override; |
| MOCK_METHOD4(MockPostPromptUserTask, |
| void(const std::vector<base::FilePath>& files_to_delete, |
| const std::vector<base::string16>& registry_keys, |
| const std::vector<base::string16>& extension_ids, |
| mojom::ChromePrompt::PromptUserCallback* callback)); |
| }; |
| |
| } // namespace chrome_cleaner |
| |
| #endif // CHROME_CHROME_CLEANER_IPC_MOCK_CHROME_PROMPT_IPC_H_ |