shill/vpn/vpn_connection_under_test.h

// Copyright 2021 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_VPN_VPN_CONNECTION_UNDER_TEST_H_
#define SHILL_VPN_VPN_CONNECTION_UNDER_TEST_H_

#include "shill/vpn/vpn_connection.h"

#include <memory>
#include <string>

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include "shill/event_dispatcher.h"
#include "shill/ipconfig.h"
#include "shill/service.h"

namespace shill {

// A simple VPNConnection implementation which can be used in tests.
class VPNConnectionUnderTest : public VPNConnection {
 public:
  VPNConnectionUnderTest(std::unique_ptr<Callbacks> callbacks,
                         EventDispatcher* dispatcher);

  VPNConnectionUnderTest(const VPNConnectionUnderTest&) = delete;
  VPNConnectionUnderTest& operator=(const VPNConnectionUnderTest&) = delete;

  // Make these two functions public to be accessible by EXPECT_CALL.
  MOCK_METHOD(void, OnConnect, (), (override));
  MOCK_METHOD(void, OnDisconnect, (), (override));

  void TriggerConnected(const std::string& link_name,
                        int interface_index,
                        const IPConfig::Properties& ip_properties);
  void TriggerFailure(Service::ConnectFailure reason,
                      const std::string& detail);
  void TriggerStopped();

  void set_state(State state) { state_ = state; }
};

}  // namespace shill

#endif  // SHILL_VPN_VPN_CONNECTION_UNDER_TEST_H_