/proc/self/cwd/test/mocks/network/connection.cc

Source (jump to first uncovered line)
#include "test/mocks/network/connection.h"

using testing::Const;
using testing::Invoke;
using testing::Return;
using testing::ReturnPointee;
using testing::ReturnRef;

namespace Envoy {
namespace Network {

MockConnectionCallbacks::MockConnectionCallbacks() = default;
MockConnectionCallbacks::~MockConnectionCallbacks() = default;

uint64_t MockConnectionBase::next_id_;

void MockConnectionBase::raiseEvent(Network::ConnectionEvent event) {
  if (event == Network::ConnectionEvent::RemoteClose ||
      event == Network::ConnectionEvent::LocalClose) {
    if (state_ == Connection::State::Closed) {
      return;
    }

    state_ = Connection::State::Closed;
  }

  for (Network::ConnectionCallbacks* callbacks : callbacks_) {
    if (callbacks) {
      callbacks->onEvent(event);
    }
  }
}

void MockConnectionBase::raiseBytesSentCallbacks(uint64_t num_bytes) {
  for (Network::Connection::BytesSentCb& cb : bytes_sent_callbacks_) {
    cb(num_bytes);
  }
}

void MockConnectionBase::runHighWatermarkCallbacks() {
  for (auto* callback : callbacks_) {
    if (callback) {
      callback->onAboveWriteBufferHighWatermark();
    }
  }
}

void MockConnectionBase::runLowWatermarkCallbacks() {
  for (auto* callback : callbacks_) {
    if (callback) {
      callback->onBelowWriteBufferLowWatermark();
    }
  }
}

template <class T> static void initializeMockConnection(T& connection) {
  ON_CALL(connection, connectionInfoSetter())
      .WillByDefault(ReturnRef(*connection.stream_info_.downstream_connection_info_provider_));
  ON_CALL(connection, connectionInfoProvider())
      .WillByDefault(ReturnPointee(connection.stream_info_.downstream_connection_info_provider_));
  ON_CALL(connection, connectionInfoProviderSharedPtr())
      .WillByDefault(ReturnPointee(&connection.stream_info_.downstream_connection_info_provider_));
  ON_CALL(connection, dispatcher()).WillByDefault(ReturnRef(connection.dispatcher_));
  ON_CALL(connection, readEnabled()).WillByDefault(ReturnPointee(&connection.read_enabled_));
  ON_CALL(connection, addConnectionCallbacks(_))
      .WillByDefault(Invoke([&connection](Network::ConnectionCallbacks& callbacks) -> void {
        connection.callbacks_.push_back(&callbacks);
      }));
  ON_CALL(connection, removeConnectionCallbacks(_))
      .WillByDefault(Invoke([&connection](Network::ConnectionCallbacks& callbacks) -> void {
        for (auto& callback : connection.callbacks_) {
          if (callback == &callbacks) {
            callback = nullptr;
            return;
          }
        }
      }));
  ON_CALL(connection, addBytesSentCallback(_))
      .WillByDefault(Invoke([&connection](Network::Connection::BytesSentCb cb) {
        connection.bytes_sent_callbacks_.emplace_back(cb);
      }));
  ON_CALL(connection, close(_))
      .WillByDefault(Invoke([&connection](ConnectionCloseType type) -> void {
        if (type == ConnectionCloseType::AbortReset) {
          connection.detected_close_type_ = DetectedCloseType::LocalReset;
        }
        connection.raiseEvent(Network::ConnectionEvent::LocalClose);
      }));
  ON_CALL(connection, detectedCloseType()).WillByDefault(Invoke([&connection]() {
    return connection.detected_close_type_;
  }));
  ON_CALL(connection, close(_, _))
      .WillByDefault(Invoke([&connection](ConnectionCloseType, absl::string_view details) -> void {
        connection.local_close_reason_ = std::string(details);
        connection.raiseEvent(Network::ConnectionEvent::LocalClose);
      }));
  ON_CALL(connection, id()).WillByDefault(Return(connection.next_id_));
  connection.stream_info_.downstream_connection_info_provider_->setConnectionID(connection.id_);
  ON_CALL(connection, state()).WillByDefault(ReturnPointee(&connection.state_));

  // The real implementation will move the buffer data into the socket.
  ON_CALL(connection, write(_, _)).WillByDefault(Invoke([](Buffer::Instance& buffer, bool) -> void {
    buffer.drain(buffer.length());
  }));

  connection.stream_info_.setUpstreamBytesMeter(std::make_shared<StreamInfo::BytesMeter>());
  ON_CALL(connection, streamInfo()).WillByDefault(ReturnRef(connection.stream_info_));
  ON_CALL(Const(connection), streamInfo()).WillByDefault(ReturnRef(connection.stream_info_));
  ON_CALL(connection, localCloseReason()).WillByDefault(Invoke([&connection]() {
    return absl::string_view(connection.local_close_reason_);
  }));
}

MockConnection::MockConnection() {
  stream_info_.downstream_connection_info_provider_->setRemoteAddress(
      *Utility::resolveUrl("tcp://10.0.0.3:50000"));
  stream_info_.filter_state_ =
      std::make_shared<StreamInfo::FilterStateImpl>(StreamInfo::FilterState::LifeSpan::Connection);

  initializeMockConnection(*this);
}
MockConnection::~MockConnection() = default;

MockServerConnection::MockServerConnection() {
  stream_info_.downstream_connection_info_provider_->setRemoteAddress(
      *Utility::resolveUrl("tcp://10.0.0.1:443"));
  stream_info_.downstream_connection_info_provider_->setLocalAddress(
      *Utility::resolveUrl("tcp://10.0.0.2:40000"));
  initializeMockConnection(*this);
}

MockServerConnection::~MockServerConnection() = default;

MockClientConnection::MockClientConnection() {
  stream_info_.downstream_connection_info_provider_->setRemoteAddress(
      *Utility::resolveUrl("tcp://10.0.0.1:443"));
  stream_info_.downstream_connection_info_provider_->setLocalAddress(
      *Utility::resolveUrl("tcp://10.0.0.2:40000"));
  initializeMockConnection(*this);
}

MockClientConnection::~MockClientConnection() = default;

MockFilterManagerConnection::MockFilterManagerConnection() {
  stream_info_.downstream_connection_info_provider_->setRemoteAddress(
      *Utility::resolveUrl("tcp://10.0.0.3:50000"));
  initializeMockConnection(*this);

  // The real implementation will move the buffer data into the socket.
  ON_CALL(*this, rawWrite(_, _)).WillByDefault(Invoke([](Buffer::Instance& buffer, bool) -> void {
    buffer.drain(buffer.length());
  }));
}
MockFilterManagerConnection::~MockFilterManagerConnection() = default;

} // namespace Network
} // namespace Envoy

Coverage Report

Created: 2024-09-19 09:45