1 files changed, 64 insertions, 0 deletions

diff --git a/toxav/rtp_test.cc b/toxav/rtp_test.cc
new file mode 100644
index 00000000..b41ff905
--- /dev/null
+++ b/toxav/rtp_test.cc
@@ -0,0 +1,64 @@
+#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((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