1 // 2 // Copyright (C) 2013 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 UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_ 18 #define UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_ 19 20 #include "update_engine/p2p_manager.h" 21 22 #include <string> 23 #include <vector> 24 25 #include <base/files/scoped_temp_dir.h> 26 #include <base/strings/string_util.h> 27 #include <gtest/gtest.h> 28 29 namespace chromeos_update_engine { 30 31 // Configuration for P2PManager for use in unit tests. Instead of 32 // /var/cache/p2p, a temporary directory is used. 33 class FakeP2PManagerConfiguration : public P2PManager::Configuration { 34 public: FakeP2PManagerConfiguration()35 FakeP2PManagerConfiguration() { 36 EXPECT_TRUE(p2p_dir_.CreateUniqueTempDir()); 37 } 38 39 // P2PManager::Configuration override GetP2PDir()40 base::FilePath GetP2PDir() override { 41 return p2p_dir_.path(); 42 } 43 44 // P2PManager::Configuration override GetInitctlArgs(bool is_start)45 std::vector<std::string> GetInitctlArgs(bool is_start) override { 46 return is_start ? initctl_start_args_ : initctl_stop_args_; 47 } 48 49 // P2PManager::Configuration override GetP2PClientArgs(const std::string & file_id,size_t minimum_size)50 std::vector<std::string> GetP2PClientArgs(const std::string &file_id, 51 size_t minimum_size) override { 52 std::vector<std::string> formatted_command = p2p_client_cmd_format_; 53 // Replace {variable} on the passed string. 54 std::string str_minimum_size = std::to_string(minimum_size); 55 for (std::string& arg : formatted_command) { 56 base::ReplaceSubstringsAfterOffset(&arg, 0, "{file_id}", file_id); 57 base::ReplaceSubstringsAfterOffset(&arg, 0, "{minsize}", 58 str_minimum_size); 59 } 60 return formatted_command; 61 } 62 63 // Use |command_line| instead of "initctl start p2p" when attempting 64 // to start the p2p service. SetInitctlStartCommand(const std::vector<std::string> & command)65 void SetInitctlStartCommand(const std::vector<std::string>& command) { 66 initctl_start_args_ = command; 67 } 68 69 // Use |command_line| instead of "initctl stop p2p" when attempting 70 // to stop the p2p service. SetInitctlStopCommand(const std::vector<std::string> & command)71 void SetInitctlStopCommand(const std::vector<std::string>& command) { 72 initctl_stop_args_ = command; 73 } 74 75 // Use |command_format| instead of "p2p-client --get-url={file_id} 76 // --minimum-size={minsize}" when attempting to look up a file using 77 // p2p-client(1). 78 // 79 // The passed |command_format| argument can have "{file_id}" and "{minsize}" 80 // as substrings of any of its elements, that will be replaced by the 81 // corresponding values passed to GetP2PClientArgs(). SetP2PClientCommand(const std::vector<std::string> & command_format)82 void SetP2PClientCommand(const std::vector<std::string>& command_format) { 83 p2p_client_cmd_format_ = command_format; 84 } 85 86 private: 87 // The temporary directory used for p2p. 88 base::ScopedTempDir p2p_dir_; 89 90 // Argument vector for starting p2p. 91 std::vector<std::string> initctl_start_args_{"initctl", "start", "p2p"}; 92 93 // Argument vector for stopping p2p. 94 std::vector<std::string> initctl_stop_args_{"initctl", "stop", "p2p"}; 95 96 // A string for generating the p2p-client command. See the 97 // SetP2PClientCommandLine() for details. 98 std::vector<std::string> p2p_client_cmd_format_{ 99 "p2p-client", "--get-url={file_id}", "--minimum-size={minsize}"}; 100 101 DISALLOW_COPY_AND_ASSIGN(FakeP2PManagerConfiguration); 102 }; 103 104 } // namespace chromeos_update_engine 105 106 #endif // UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_ 107