#include "rtp.h"

#include "../toxcore/crypto_core.h"

#include <gtest/gtest.h>

namespace {

RTPHeader random_header() {
  return {
      random_u16(), random_u16(), random_u16(), random_u16(), random_u16(),
      random_u16(), random_u16(), random_u32(), random_u32(), random_u64(),
      random_u32(), random_u32(), random_u32(), random_u16(), random_u16(),
  };
}

TEST(Rtp, Deserialisation) {
  RTPHeader const header = random_header();

  uint8_t rdata[RTP_HEADER_SIZE];
  EXPECT_EQ(rtp_header_pack(rdata, &header), RTP_HEADER_SIZE);

  RTPHeader unpacked = {0};
  EXPECT_EQ(rtp_header_unpack(rdata, &unpacked), RTP_HEADER_SIZE);

  EXPECT_EQ(header.ve, unpacked.ve);
  EXPECT_EQ(header.pe, unpacked.pe);
  EXPECT_EQ(header.xe, unpacked.xe);
  EXPECT_EQ(header.cc, unpacked.cc);
  EXPECT_EQ(header.ma, unpacked.ma);
  EXPECT_EQ(header.pt, unpacked.pt);
  EXPECT_EQ(header.sequnum, unpacked.sequnum);
  EXPECT_EQ(header.timestamp, unpacked.timestamp);
  EXPECT_EQ(header.ssrc, unpacked.ssrc);
  EXPECT_EQ(header.flags, unpacked.flags);
  EXPECT_EQ(header.offset_full, unpacked.offset_full);
  EXPECT_EQ(header.data_length_full, unpacked.data_length_full);
  EXPECT_EQ(header.received_length_full, unpacked.received_length_full);
  EXPECT_EQ(header.offset_lower, unpacked.offset_lower);
  EXPECT_EQ(header.data_length_lower, unpacked.data_length_lower);
}

TEST(Rtp, SerialisingAllOnes) {
  RTPHeader header;
  memset(&header, 0xff, sizeof header);

  uint8_t rdata[RTP_HEADER_SIZE];
  rtp_header_pack(rdata, &header);

  EXPECT_EQ(std::string(reinterpret_cast<char const *>(rdata), sizeof rdata),
            std::string("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"
                        "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"
                        "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"
                        "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF"
                        "\x00\x00\x00\x00\x00\x00\x00\x00"
                        "\x00\x00\x00\x00\x00\x00\x00\x00"
                        "\x00\x00\x00\x00\x00\x00\x00\x00"
                        "\x00\x00\x00\x00\x00\x00\x00\x00"
                        "\x00\x00\x00\x00\x00\x00\x00\x00"
                        "\x00\x00\x00\x00\xFF\xFF\xFF\xFF",
                        RTP_HEADER_SIZE));
}

}  // namespace
