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 
30 namespace chromeos_update_engine {
31 
32 // This class implements the connection to shill using real DBus calls.
33 class 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