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 #include "update_engine/fake_shill_proxy.h"
18 
19 using org::chromium::flimflam::ManagerProxyMock;
20 using org::chromium::flimflam::ServiceProxyInterface;
21 
22 namespace chromeos_update_engine {
23 
FakeShillProxy()24 FakeShillProxy::FakeShillProxy()
25     : manager_proxy_mock_(new ManagerProxyMock()) {}
26 
GetManagerProxy()27 ManagerProxyMock* FakeShillProxy::GetManagerProxy() {
28   return manager_proxy_mock_.get();
29 }
30 
GetServiceForPath(const dbus::ObjectPath & path)31 std::unique_ptr<ServiceProxyInterface> FakeShillProxy::GetServiceForPath(
32     const dbus::ObjectPath& path) {
33   auto it = service_proxy_mocks_.find(path.value());
34   CHECK(it != service_proxy_mocks_.end()) << "No ServiceProxyMock set for "
35                                           << path.value();
36   std::unique_ptr<ServiceProxyInterface> result = std::move(it->second);
37   service_proxy_mocks_.erase(it);
38   return result;
39 }
40 
SetServiceForPath(const dbus::ObjectPath & path,std::unique_ptr<ServiceProxyInterface> service_proxy)41 void FakeShillProxy::SetServiceForPath(
42     const dbus::ObjectPath& path,
43     std::unique_ptr<ServiceProxyInterface> service_proxy) {
44   service_proxy_mocks_[path.value()] = std::move(service_proxy);
45 }
46 
47 }  // namespace chromeos_update_engine
48