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 UPDATE_ENGINE_CROS_FAKE_SHILL_PROXY_H_ 18 #define UPDATE_ENGINE_CROS_FAKE_SHILL_PROXY_H_ 19 20 #include <map> 21 #include <memory> 22 #include <string> 23 24 #include <base/macros.h> 25 #include <shill/dbus-proxies.h> 26 #include <shill/dbus-proxy-mocks.h> 27 28 #include "update_engine/cros/shill_proxy_interface.h" 29 30 namespace chromeos_update_engine { 31 32 // This class implements the connection to shill using real DBus calls. 33 class FakeShillProxy : public ShillProxyInterface { 34 public: 35 FakeShillProxy(); 36 ~FakeShillProxy() override = default; 37 38 // ShillProxyInterface overrides. 39 40 // GetManagerProxy returns the subclass ManagerProxyMock so tests can easily 41 // use it. Mocks for the return value of GetServiceForPath() can be provided 42 // with SetServiceForPath(). 43 org::chromium::flimflam::ManagerProxyMock* GetManagerProxy() override; 44 std::unique_ptr<org::chromium::flimflam::ServiceProxyInterface> 45 GetServiceForPath(const dbus::ObjectPath& path) override; 46 47 // Sets the service_proxy that will be returned by GetServiceForPath(). 48 void SetServiceForPath( 49 const dbus::ObjectPath& path, 50 std::unique_ptr<org::chromium::flimflam::ServiceProxyInterface> 51 service_proxy); 52 53 private: 54 std::unique_ptr<org::chromium::flimflam::ManagerProxyMock> 55 manager_proxy_mock_; 56 57 std::map<std::string, 58 std::unique_ptr<org::chromium::flimflam::ServiceProxyInterface>> 59 service_proxy_mocks_; 60 61 DISALLOW_COPY_AND_ASSIGN(FakeShillProxy); 62 }; 63 64 } // namespace chromeos_update_engine 65 66 #endif // UPDATE_ENGINE_CROS_FAKE_SHILL_PROXY_H_ 67