stubRpcTransport.h

Go to the documentation of this file.

 
#pragma once
 
#include <gmock/gmock.h>
 
#include "profinet/rpc.h"
 
namespace profinet
{
constexpr std::uint16_t kInputFrameId = 0x9000;
constexpr std::uint16_t kOutputFrameId = 0x8001;
 
class StubRpcTransport final : public rpc::IRpcTransport
{
 public:
  Bytes SendReceive(const Bytes& request, std::chrono::milliseconds) override
  {
    requests.push_back(request);
 
    const auto parsed = PNRPCHeader::Parse(request);
    PNRPCHeader response = parsed;
    response.packetType = PNRPCHeader::RESPONSE;
 
    Bytes blocks;
    for (const auto [type, frameId] :
         std::array<std::pair<std::uint16_t, std::uint16_t>, 2>{
             {{1, kInputFrameId}, {2, kOutputFrameId}}})
    {
      wire::PutU16(blocks, PNIOCRBlockRes::BLOCK_TYPE);
      wire::PutU16(blocks, PNIOCRBlockRes::kSize - 4);
      wire::PutU8(blocks, rpc::VERSION_HIGH);
      wire::PutU8(blocks, rpc::VERSION_LOW);
      wire::PutU16(blocks, type);
      wire::PutU16(blocks, type);
      wire::PutU16(blocks, frameId);
    }
 
    PNNRDData nrd;
    nrd.argsMaximumStatus = 0;
    nrd.argsLength = static_cast<std::uint32_t>(blocks.size());
    nrd.maximumCount = 1500;
    nrd.offset = 0;
    nrd.actualCount = static_cast<std::uint32_t>(blocks.size());
    nrd.payload = blocks;
    response.payload = nrd.ToBytes();
    response.lengthOfBody = static_cast<std::uint16_t>(response.payload.size());
    return response.ToBytes();
  }
 
  void Send(const Bytes& data) override
  {
    sends.push_back(data);
  }
 
  void Open(const std::string&, std::uint16_t, bool = false) override
  {
  }
 
  Bytes Receive(std::chrono::milliseconds, asio::ip::udp::endpoint&) override
  {
    return {};
  }
 
  void SendTo(const Bytes& data, const asio::ip::udp::endpoint&) override
  {
    sends.push_back(data);
  }
 
  void Close() noexcept override
  {
  }
 
  std::vector<Bytes> requests;
  std::vector<Bytes> sends;
};
} // namespace profinet