mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef SHILL_MOCK_WIFI_ |
| 6 | #define SHILL_MOCK_WIFI_ |
| 7 | |
Hristo Stefanov | ed2c28c | 2011-11-29 15:37:30 -0800 | [diff] [blame] | 8 | #include <map> |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 9 | #include <string> |
| 10 | |
| 11 | #include <base/memory/ref_counted.h> |
| 12 | #include <gmock/gmock.h> |
| 13 | |
| 14 | #include "shill/key_value_store.h" |
| 15 | #include "shill/refptr_types.h" |
| 16 | #include "shill/wifi.h" |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame^] | 17 | #include "shill/wifi_endpoint.h" |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 18 | #include "shill/wifi_service.h" |
| 19 | |
| 20 | namespace shill { |
| 21 | |
| 22 | class ControlInterface; |
| 23 | class Error; |
| 24 | class EventDispatcher; |
| 25 | |
| 26 | class MockWiFi : public WiFi { |
| 27 | public: |
| 28 | MockWiFi(ControlInterface *control_interface, |
| 29 | EventDispatcher *dispatcher, |
Thieu Le | 3426c8f | 2012-01-11 17:35:11 -0800 | [diff] [blame] | 30 | Metrics *metrics, |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 31 | Manager *manager, |
| 32 | const std::string &link_name, |
| 33 | const std::string &address, |
| 34 | int interface_index); |
| 35 | virtual ~MockWiFi(); |
| 36 | |
| 37 | MOCK_METHOD0(Start, void()); |
| 38 | MOCK_METHOD0(Stop, void()); |
| 39 | MOCK_METHOD1(Scan, void(Error *error)); |
| 40 | MOCK_METHOD2(GetService, |
| 41 | WiFiServiceRefPtr(const KeyValueStore &args, Error *error)); |
Paul Stewart | 66c8600 | 2012-01-30 18:00:52 -0800 | [diff] [blame] | 42 | MOCK_METHOD0(ClearCachedCredentials, void()); |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 43 | MOCK_METHOD2(ConnectTo, |
mukesh agrawal | 6489632 | 2011-12-01 01:13:10 +0000 | [diff] [blame] | 44 | void(WiFiService *service, |
| 45 | std::map<std::string, ::DBus::Variant> service_params)); |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 46 | MOCK_CONST_METHOD0(IsIdle, bool()); |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame^] | 47 | MOCK_METHOD1(NotifyEndpointChanged, void(const WiFiEndpoint &endpoint)); |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 48 | |
| 49 | private: |
| 50 | DISALLOW_COPY_AND_ASSIGN(MockWiFi); |
| 51 | }; |
| 52 | |
| 53 | } // namespace shill |
| 54 | |
| 55 | #endif // SHILL_MOCK_WIFI_ |