remoting/protocol/sdp_message_unittest.cc

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "remoting/protocol/sdp_message.h"

#include "testing/gtest/include/gtest/gtest.h"

namespace remoting {
namespace protocol {

// Verify that SDP is normalized by removing empty lines and normalizing
// line-endings to \r\n.
TEST(SdpMessages, Normalize) {
  SdpMessage sdp_message("a=foo\n\r\nb=bar");
  EXPECT_EQ("a=foo\r\nb=bar\r\n", sdp_message.ToString());
}

// Verify that the normalized form of SDP for signature calculation has
// line-endings of \n, for compatibility with existing clients.
TEST(SdpMessages, NormalizedForSignature) {
  SdpMessage sdp_message("a=foo\nb=bar\r\n");
  EXPECT_EQ("a=foo\nb=bar\n", sdp_message.NormalizedForSignature());
}

TEST(SdpMessages, AddCodecParameter) {
  std::string kSourceSdp =
      "a=group:BUNDLE video\n"
      "m=video 9 UDP/TLS/RTP/SAVPF 96\n"
      "a=rtpmap:96 VP8/90000\n"
      "a=rtcp-fb:96 transport-cc\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_TRUE(sdp_message.AddCodecParameter("VP8", "test_param=1"));
  EXPECT_EQ(
      "a=group:BUNDLE video\r\n"
      "m=video 9 UDP/TLS/RTP/SAVPF 96\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96 test_param=1\r\n"
      "a=rtcp-fb:96 transport-cc\r\n",
      sdp_message.ToString());
}

TEST(SdpMessages, AddCodecParameterMissingCodec) {
  std::string kSourceSdp =
      "a=group:BUNDLE audio video\r\n"
      "m=audio 9 UDP/TLS/RTP/SAVPF 111\r\n"
      "a=extmap:1 urn:ietf:params:rtp-hdrext:ssrc-audio-level\r\n"
      "a=rtpmap:111 opus/48000/2\r\n"
      "a=rtcp-fb:111 transport-cc\r\n"
      "m=video 9 UDP/TLS/RTP/SAVPF 96\r\n"
      "a=rtpmap:96 VP9/90000\r\n"
      "a=rtcp-fb:96 transport-cc\r\n"
      "m=application 9 DTLS/SCTP 5000\r\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_FALSE(sdp_message.AddCodecParameter("VP8", "test_param=1"));
  EXPECT_EQ(kSourceSdp, sdp_message.ToString());
}

TEST(SdpMessages, AddCodecParameter_MultipleTypes) {
  const std::string kSourceSdp =
      "a=group:BUNDLE video\n"
      "m=video 9 UDP/TLS/RTP/SAVPF 96 97 98 99\n"
      "a=rtpmap:96 VP8/90000\n"
      "a=rtcp-fb:96 transport-cc\n"
      "a=rtpmap:97 VP9/90000\n"
      "a=rtcp-fb:97 transport-cc\n"
      "a=rtpmap:98 VP8/90000\n"
      "a=rtcp-fb:98 transport-cc\n"
      "a=rtpmap:99 VP8/90000\n"
      "a=rtcp-fb:99 transport-cc\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_TRUE(sdp_message.AddCodecParameter("VP8", "test_param=1"));
  EXPECT_EQ(
      "a=group:BUNDLE video\r\n"
      "m=video 9 UDP/TLS/RTP/SAVPF 96 97 98 99\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96 test_param=1\r\n"
      "a=rtcp-fb:96 transport-cc\r\n"
      "a=rtpmap:97 VP9/90000\r\n"
      "a=rtcp-fb:97 transport-cc\r\n"
      "a=rtpmap:98 VP8/90000\r\n"
      "a=fmtp:98 test_param=1\r\n"
      "a=rtcp-fb:98 transport-cc\r\n"
      "a=rtpmap:99 VP8/90000\r\n"
      "a=fmtp:99 test_param=1\r\n"
      "a=rtcp-fb:99 transport-cc\r\n",
      sdp_message.ToString());
}

TEST(SdpMessages, PreferVideoCodec_SdpIsUnchanged) {
  static const std::string kSourceSdp =
      "m=video 98 UDP/TLS/RTP/SAVPF 96 97 98 99 100\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("VP8"));
  EXPECT_EQ(kSourceSdp, sdp_message.ToString());
}

TEST(SdpMessages, PreferVideoCodec_MoveToTheStartOfTheList) {
  static const std::string kSourceSdp =
      "m=video 98 UDP/TLS/RTP/SAVPF 96 97 98 99 100\n"
      "a=rtpmap:96 VP8/90000\n"
      "a=fmtp:96\n"
      "a=rtpmap:97 VP8.1/90000\n"
      "a=fmtp:97\n"
      "a=rtpmap:98 VP9/90000\n"
      "a=fmtp:98\n"
      "a=rtpmap:99 VP9.1/90000\n"
      "a=fmtp:99\n"
      "a=rtpmap:100 H264/90000\n"
      "a=fmtp:100\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("VP8.1"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 97 96 98 99 100\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());

  sdp_message = SdpMessage(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("VP9"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 98 96 97 99 100\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());

  sdp_message = SdpMessage(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("VP9.1"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 99 96 97 98 100\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());

  sdp_message = SdpMessage(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("H264"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 100 96 97 98 99\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());
}

TEST(SdpMessages, PreferVideoCodec_UnknownCodec) {
  static const std::string kSourceSdp =
      "m=video 98 UDP/TLS/RTP/SAVPF 96 97 98 99 100\n"
      "a=rtpmap:96 VP8/90000\n"
      "a=fmtp:96\n"
      "a=rtpmap:97 VP8.1/90000\n"
      "a=fmtp:97\n"
      "a=rtpmap:98 VP9/90000\n"
      "a=fmtp:98\n"
      "a=rtpmap:99 VP9.1/90000\n"
      "a=fmtp:99\n"
      "a=rtpmap:100 H264/90000\n"
      "a=fmtp:100\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_FALSE(sdp_message.PreferVideoCodec("VP7"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 96 97 98 99 100\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());
}

TEST(SdpMessages, PreferVideoCodec_MultiplePlayloads) {
  static const std::string kSourceSdp =
      "m=video 98 UDP/TLS/RTP/SAVPF 96 97 100 101 102 98 99\n"
      "a=rtpmap:96 VP8/90000\n"
      "a=fmtp:96\n"
      "a=rtpmap:97 VP8.1/90000\n"
      "a=fmtp:97\n"
      "a=rtpmap:101 VP9/90000\n"
      "a=fmtp:101\n"
      "a=rtpmap:98 VP9/90000\n"
      "a=fmtp:98\n"
      "a=rtpmap:99 VP9.1/90000\n"
      "a=fmtp:99\n"
      "a=rtpmap:102 VP9/90000\n"
      "a=fmtp:102\n"
      "a=rtpmap:103 VP9/90000\n"
      "a=fmtp:103\n"
      "a=rtpmap:100 H264/90000\n"
      "a=fmtp:100\n";
  SdpMessage sdp_message(kSourceSdp);
  EXPECT_TRUE(sdp_message.PreferVideoCodec("VP9"));
  EXPECT_EQ(
      "m=video 98 UDP/TLS/RTP/SAVPF 102 98 101 96 97 100 99\r\n"
      "a=rtpmap:96 VP8/90000\r\n"
      "a=fmtp:96\r\n"
      "a=rtpmap:97 VP8.1/90000\r\n"
      "a=fmtp:97\r\n"
      "a=rtpmap:101 VP9/90000\r\n"
      "a=fmtp:101\r\n"
      "a=rtpmap:98 VP9/90000\r\n"
      "a=fmtp:98\r\n"
      "a=rtpmap:99 VP9.1/90000\r\n"
      "a=fmtp:99\r\n"
      "a=rtpmap:102 VP9/90000\r\n"
      "a=fmtp:102\r\n"
      "a=rtpmap:103 VP9/90000\r\n"
      "a=fmtp:103\r\n"
      "a=rtpmap:100 H264/90000\r\n"
      "a=fmtp:100\r\n",
      sdp_message.ToString());
}

}  // namespace protocol
}  // namespace remoting