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
19using org::chromium::flimflam::ManagerProxyMock;
20using org::chromium::flimflam::ServiceProxyInterface;
21
22namespace chromeos_update_engine {
23
24FakeShillProxy::FakeShillProxy()
25    : manager_proxy_mock_(new ManagerProxyMock()) {}
26
27ManagerProxyMock* FakeShillProxy::GetManagerProxy() {
28  return manager_proxy_mock_.get();
29}
30
31std::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
41void 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