1 // 2 // Copyright (C) 2012 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 SHILL_MOCK_DEVICE_INFO_H_ 18 #define SHILL_MOCK_DEVICE_INFO_H_ 19 20 #include <string> 21 #include <vector> 22 23 #include <base/macros.h> 24 #include <gmock/gmock.h> 25 26 #include "shill/device_info.h" 27 28 namespace shill { 29 30 class ByteString; 31 class ControlInterface; 32 class EventDispatcher; 33 class IPAddress; 34 class Manager; 35 class Metrics; 36 37 class MockDeviceInfo : public DeviceInfo { 38 public: 39 MockDeviceInfo(ControlInterface* control_interface, 40 EventDispatcher* dispatcher, 41 Metrics* metrics, 42 Manager* manager); 43 ~MockDeviceInfo() override; 44 45 MOCK_METHOD1(IsDeviceBlackListed, bool(const std::string& device_name)); 46 MOCK_METHOD1(AddDeviceToBlackList, void(const std::string& device_name)); 47 MOCK_METHOD1(RemoveDeviceFromBlackList, void(const std::string& device_name)); 48 MOCK_CONST_METHOD1(GetDevice, DeviceRefPtr(int interface_index)); 49 MOCK_CONST_METHOD1(GetIndex, int(const std::string& interface_name)); 50 MOCK_CONST_METHOD2(GetMACAddress, bool(int interface_index, 51 ByteString* address)); 52 MOCK_CONST_METHOD1(GetMACAddressFromKernel, ByteString(int interface_index)); 53 MOCK_CONST_METHOD3(GetMACAddressOfPeer, 54 bool(int interface_index, 55 const IPAddress& peer, 56 ByteString* address)); 57 MOCK_CONST_METHOD3(GetByteCounts, bool(int interface_index, 58 uint64_t* rx_bytes, 59 uint64_t* tx_bytes)); 60 MOCK_CONST_METHOD2(GetFlags, bool(int interface_index, 61 unsigned int* flags)); 62 MOCK_CONST_METHOD2(GetAddresses, bool(int interface_index, 63 std::vector<AddressData>* addresses)); 64 MOCK_CONST_METHOD1(FlushAddresses, void(int interface_index)); 65 MOCK_CONST_METHOD2(HasOtherAddress, 66 bool(int interface_index, 67 const IPAddress& excluded_address)); 68 MOCK_CONST_METHOD2(HasDirectConnectivityTo, 69 bool(int interface_index, 70 const IPAddress& address)); 71 MOCK_METHOD2(GetPrimaryIPv6Address, 72 bool(int interface_index, IPAddress* address)); 73 MOCK_METHOD3(GetIPv6DnsServerAddresses, 74 bool(int interface_index, 75 std::vector<IPAddress>* address_list, 76 uint32_t* life_time)); 77 MOCK_CONST_METHOD1(CreateTunnelInterface, bool(std::string* interface_name)); 78 MOCK_CONST_METHOD1(OpenTunnelInterface, 79 int(const std::string& interface_name)); 80 MOCK_CONST_METHOD1(DeleteInterface, bool(int interface_index)); 81 MOCK_METHOD1(RegisterDevice, void(const DeviceRefPtr&)); 82 MOCK_METHOD1(DeregisterDevice, void(const DeviceRefPtr&)); 83 MOCK_CONST_METHOD1(SetHostname, bool(const std::string& hostname)); 84 85 private: 86 DISALLOW_COPY_AND_ASSIGN(MockDeviceInfo); 87 }; 88 89 } // namespace shill 90 91 #endif // SHILL_MOCK_DEVICE_INFO_H_ 92