Alex Deymo | aea4c1c | 2015-08-19 20:24:43 -0700 | [diff] [blame] | 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 | // |
Alex Deymo | 3053450 | 2015-07-20 15:06:33 -0700 | [diff] [blame] | 16 | |
| 17 | #include "update_engine/fake_shill_proxy.h" |
| 18 | |
Amin Hassani | 7cc8bb0 | 2019-01-14 16:29:47 -0800 | [diff] [blame] | 19 | #include <utility> |
| 20 | |
Alex Deymo | 3053450 | 2015-07-20 15:06:33 -0700 | [diff] [blame] | 21 | using org::chromium::flimflam::ManagerProxyMock; |
| 22 | using org::chromium::flimflam::ServiceProxyInterface; |
| 23 | |
| 24 | namespace chromeos_update_engine { |
| 25 | |
| 26 | FakeShillProxy::FakeShillProxy() |
| 27 | : manager_proxy_mock_(new ManagerProxyMock()) {} |
| 28 | |
| 29 | ManagerProxyMock* FakeShillProxy::GetManagerProxy() { |
| 30 | return manager_proxy_mock_.get(); |
| 31 | } |
| 32 | |
| 33 | std::unique_ptr<ServiceProxyInterface> FakeShillProxy::GetServiceForPath( |
Alex Deymo | 758dd53 | 2015-09-09 15:21:22 -0700 | [diff] [blame] | 34 | const dbus::ObjectPath& path) { |
| 35 | auto it = service_proxy_mocks_.find(path.value()); |
Amin Hassani | 7cc8bb0 | 2019-01-14 16:29:47 -0800 | [diff] [blame] | 36 | CHECK(it != service_proxy_mocks_.end()) |
| 37 | << "No ServiceProxyMock set for " << path.value(); |
Alex Deymo | 3053450 | 2015-07-20 15:06:33 -0700 | [diff] [blame] | 38 | std::unique_ptr<ServiceProxyInterface> result = std::move(it->second); |
| 39 | service_proxy_mocks_.erase(it); |
| 40 | return result; |
| 41 | } |
| 42 | |
| 43 | void FakeShillProxy::SetServiceForPath( |
Alex Deymo | 758dd53 | 2015-09-09 15:21:22 -0700 | [diff] [blame] | 44 | const dbus::ObjectPath& path, |
Alex Deymo | 3053450 | 2015-07-20 15:06:33 -0700 | [diff] [blame] | 45 | std::unique_ptr<ServiceProxyInterface> service_proxy) { |
Alex Deymo | 758dd53 | 2015-09-09 15:21:22 -0700 | [diff] [blame] | 46 | service_proxy_mocks_[path.value()] = std::move(service_proxy); |
Alex Deymo | 3053450 | 2015-07-20 15:06:33 -0700 | [diff] [blame] | 47 | } |
| 48 | |
| 49 | } // namespace chromeos_update_engine |