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_FAKE_SHILL_PROXY_H_
18#define UPDATE_ENGINE_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/shill_proxy_interface.h"
29
30namespace chromeos_update_engine {
31
32// This class implements the connection to shill using real DBus calls.
33class 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_FAKE_SHILL_PROXY_H_
67