net-base/mock_rtnl_handler.h

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

#ifndef NET_BASE_MOCK_RTNL_HANDLER_H_
#define NET_BASE_MOCK_RTNL_HANDLER_H_

#include <memory>
#include <string>

#include <gmock/gmock.h>

#include "net-base/mac_address.h"
#include "net-base/rtnl_handler.h"

namespace net_base {

class BRILLO_EXPORT MockRTNLHandler : public RTNLHandler {
 public:
  MockRTNLHandler();
  MockRTNLHandler(const MockRTNLHandler&) = delete;
  MockRTNLHandler& operator=(const MockRTNLHandler&) = delete;

  ~MockRTNLHandler() override;

  MOCK_METHOD(void, Start, (uint32_t), (override));
  MOCK_METHOD(void, AddListener, (RTNLListener*), (override));
  MOCK_METHOD(void, RemoveListener, (RTNLListener*), (override));
  MOCK_METHOD(void,
              SetInterfaceFlags,
              (int, unsigned int, unsigned int),
              (override));
  MOCK_METHOD(void, SetInterfaceMTU, (int, unsigned int), (override));
  MOCK_METHOD(void,
              SetInterfaceMac,
              (int, const MacAddress&, ResponseCallback),
              (override));
  MOCK_METHOD(bool,
              AddInterfaceAddress,
              (int, const IPCIDR&, const std::optional<IPv4Address>&),
              (override));
  MOCK_METHOD(bool, RemoveInterfaceAddress, (int, const IPCIDR&), (override));
  MOCK_METHOD(bool, RemoveInterface, (int), (override));
  MOCK_METHOD(void, RequestDump, (uint32_t), (override));
  MOCK_METHOD(int, GetInterfaceIndex, (const std::string&), (override));
  MOCK_METHOD(bool, DoSendMessage, (RTNLMessage*, uint32_t*));
  MOCK_METHOD(bool,
              AddInterface,
              (const std::string& interface_name,
               const std::string& link_kind,
               base::span<const uint8_t> link_info_data,
               ResponseCallback response_callback),
              (override));
  bool SendMessage(std::unique_ptr<RTNLMessage> message,
                   uint32_t* seq) override {
    return DoSendMessage(message.get(), seq);
  }
};

}  // namespace net_base
#endif  // NET_BASE_MOCK_RTNL_HANDLER_H_