1 // 2 // Copyright (C) 2015 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef SHILL_MOCK_PROCESS_MANAGER_H_ 18 #define SHILL_MOCK_PROCESS_MANAGER_H_ 19 20 #include "shill/process_manager.h" 21 22 #include <map> 23 #include <string> 24 #include <vector> 25 26 #include <base/macros.h> 27 #include <gmock/gmock.h> 28 29 namespace shill { 30 31 class MockProcessManager : public ProcessManager { 32 public: 33 MockProcessManager(); 34 ~MockProcessManager() override; 35 36 MOCK_METHOD1(Init, void(EventDispatcher* dispatcher)); 37 MOCK_METHOD0(Stop, void()); 38 MOCK_METHOD6(StartProcess, 39 pid_t(const tracked_objects::Location& spawn_source, 40 const base::FilePath& program, 41 const std::vector<std::string>& arguments, 42 const std::map<std::string, std::string>& env, 43 bool terminate_with_parent, 44 const base::Callback<void(int)>& exit_callback)); 45 MOCK_METHOD7(StartProcessInMinijail, 46 pid_t(const tracked_objects::Location& spawn_source, 47 const base::FilePath& program, 48 const std::vector<std::string>& arguments, 49 const std::string& user, 50 const std::string& group, 51 uint64_t capmask, 52 const base::Callback<void(int)>& exit_callback)); 53 MOCK_METHOD10(StartProcessInMinijailWithPipes, 54 pid_t(const tracked_objects::Location& spawn_source, 55 const base::FilePath& program, 56 const std::vector<std::string>& arguments, 57 const std::string& user, 58 const std::string& group, 59 uint64_t capmask, 60 const base::Callback<void(int)>& exit_callback, 61 int* stdin_fd, 62 int* stdout_fd, 63 int* stderr_fd)); 64 MOCK_METHOD1(StopProcess, bool(pid_t pid)); 65 MOCK_METHOD1(StopProcessAndBlock, bool(pid_t pid)); 66 MOCK_METHOD2(UpdateExitCallback, 67 bool(pid_t pid, const base::Callback<void(int)>& new_callback)); 68 69 private: 70 DISALLOW_COPY_AND_ASSIGN(MockProcessManager); 71 }; 72 73 } // namespace shill 74 75 #endif // SHILL_MOCK_PROCESS_MANAGER_H_ 76