// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SHILL_MOCK_DEVICE_INFO_H_
#define SHILL_MOCK_DEVICE_INFO_H_

#include <string>
#include <vector>

#include <base/macros.h>
#include <gmock/gmock.h>

#include "shill/device_info.h"

namespace shill {

class ByteString;
class IPAddress;
class Manager;

class MockDeviceInfo : public DeviceInfo {
 public:
  explicit MockDeviceInfo(Manager* manager);
  ~MockDeviceInfo() override;

  MOCK_METHOD(bool, IsDeviceBlackListed, (const std::string&), (override));
  MOCK_METHOD(void, AddDeviceToBlackList, (const std::string&), (override));
  MOCK_METHOD(void,
              RemoveDeviceFromBlackList,
              (const std::string&),
              (override));
  MOCK_METHOD(DeviceRefPtr, GetDevice, (int), (const, override));
  MOCK_METHOD(int, GetIndex, (const std::string&), (const, override));
  MOCK_METHOD(bool, GetMacAddress, (int, ByteString*), (const, override));
  MOCK_METHOD(ByteString, GetMacAddressFromKernel, (int), (const, override));
  MOCK_METHOD(bool,
              GetMacAddressOfPeer,
              (int, const IPAddress&, ByteString*),
              (const, override));
  MOCK_METHOD(bool,
              GetByteCounts,
              (int, uint64_t*, uint64_t*),
              (const, override));
  MOCK_METHOD(bool, GetFlags, (int, unsigned int*), (const, override));
  MOCK_METHOD(std::vector<IPAddress>, GetAddresses, (int), (const, override));
  MOCK_METHOD(void, FlushAddresses, (int), (const, override));
  MOCK_METHOD(bool,
              HasOtherAddress,
              (int, const IPAddress&),
              (const, override));
  MOCK_METHOD(bool, GetPrimaryIPv6Address, (int, IPAddress*), (override));
  MOCK_METHOD(bool,
              GetIPv6DnsServerAddresses,
              (int, std::vector<IPAddress>*, uint32_t*),
              (override));
  MOCK_METHOD(bool, CreateTunnelInterface, (std::string*), (const, override));
  MOCK_METHOD(int,
              OpenTunnelInterface,
              (const std::string&),
              (const, override));
  MOCK_METHOD(bool, DeleteInterface, (int), (const, override));
  MOCK_METHOD(void, RegisterDevice, (const DeviceRefPtr&), (override));
  MOCK_METHOD(void, DeregisterDevice, (const DeviceRefPtr&), (override));
  MOCK_METHOD(bool, SetHostname, (const std::string&), (const, override));

 private:
  DISALLOW_COPY_AND_ASSIGN(MockDeviceInfo);
};

}  // namespace shill

#endif  // SHILL_MOCK_DEVICE_INFO_H_