shill/wifi/mock_p2p_device.h

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

#ifndef SHILL_WIFI_MOCK_P2P_DEVICE_H_
#define SHILL_WIFI_MOCK_P2P_DEVICE_H_

#include "shill/wifi/p2p_device.h"

#include <memory>
#include <string>

#include <gmock/gmock.h>

namespace shill {

class MockP2PDevice : public P2PDevice {
 public:
  MockP2PDevice(Manager* manager,
                LocalDevice::IfaceType iface_type,
                const std::string& primary_link_name,
                uint32_t phy_index,
                int32_t shill_id,
                WiFiPhy::Priority priority,
                const EventCallback& callback)
      : P2PDevice(manager,
                  iface_type,
                  primary_link_name,
                  phy_index,
                  shill_id,
                  priority,
                  callback) {}

  ~MockP2PDevice() = default;

  bool Start() override { return true; }

  bool Stop() override { return true; }

  MOCK_METHOD(KeyValueStore, GetGroupInfo, (), (const));
  MOCK_METHOD(KeyValueStore, GetClientInfo, (), (const));
  MOCK_METHOD(bool, CreateGroup, (std::unique_ptr<P2PService>), ());
  MOCK_METHOD(bool, Connect, (std::unique_ptr<P2PService>), ());
  MOCK_METHOD(bool, RemoveGroup, (), ());
  MOCK_METHOD(bool, Disconnect, (), ());
  MOCK_METHOD(P2PDevice::P2PDeviceState, state, (), (const));
  MOCK_METHOD(void, GroupStarted, (const KeyValueStore&), (override));
  MOCK_METHOD(void, GroupFinished, (const KeyValueStore&), (override));
  MOCK_METHOD(void, GroupFormationFailure, (const std::string&), (override));
};

}  // namespace shill

#endif  // SHILL_WIFI_MOCK_P2P_DEVICE_H_