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